Text Classification
Transformers
Safetensors
roberta
Generated from Trainer
classification
nlp
vulnerability
text-embeddings-inference
Instructions to use CIRCL/vulnerability-severity-classification-roberta-base with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use CIRCL/vulnerability-severity-classification-roberta-base with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="CIRCL/vulnerability-severity-classification-roberta-base")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("CIRCL/vulnerability-severity-classification-roberta-base") model = AutoModelForSequenceClassification.from_pretrained("CIRCL/vulnerability-severity-classification-roberta-base") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -63,10 +63,6 @@ predicted_class = torch.argmax(predictions, dim=-1).item()
|
|
| 63 |
print("Predicted severity:", labels[predicted_class])
|
| 64 |
```
|
| 65 |
|
| 66 |
-
## Training and evaluation data
|
| 67 |
-
|
| 68 |
-
More information needed
|
| 69 |
-
|
| 70 |
## Training procedure
|
| 71 |
|
| 72 |
### Training hyperparameters
|
|
|
|
| 63 |
print("Predicted severity:", labels[predicted_class])
|
| 64 |
```
|
| 65 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 66 |
## Training procedure
|
| 67 |
|
| 68 |
### Training hyperparameters
|