How to use CIRCL/cwe-parent-vulnerability-classification-roberta-base with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="CIRCL/cwe-parent-vulnerability-classification-roberta-base")
# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("CIRCL/cwe-parent-vulnerability-classification-roberta-base") model = AutoModelForSequenceClassification.from_pretrained("CIRCL/cwe-parent-vulnerability-classification-roberta-base")
4e98507 90274d3 9b52994 4e98507
1
2
3
4
5
6
7
8
9
{ "eval_loss": 1.7509655952453613, "eval_accuracy": 0.5454545454545454, "eval_f1_macro": 0.37756565971323075, "eval_runtime": 1.6377, "eval_samples_per_second": 544.058, "eval_steps_per_second": 8.549, "epoch": 40.0 }
