Improve model card: Add pipeline tag, paper, project, code links, and descriptive tags (#1)
Browse files- Improve model card: Add pipeline tag, paper, project, code links, and descriptive tags (e8599983e7734e48b8ba5f5176126d4b11055121)
Co-authored-by: Niels Rogge <[email protected]>
README.md
CHANGED
@@ -1,25 +1,32 @@
|
|
1 |
---
|
|
|
|
|
|
|
2 |
library_name: transformers
|
3 |
license: apache-2.0
|
4 |
-
base_model: distilbert-base-uncased
|
5 |
-
tags:
|
6 |
-
- generated_from_trainer
|
7 |
metrics:
|
8 |
- accuracy
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
9 |
model-index:
|
10 |
- name: vulnerability-severity-classification-distilbert-base-uncased
|
11 |
results: []
|
12 |
-
datasets:
|
13 |
-
- CIRCL/vulnerability-scores
|
14 |
---
|
15 |
|
16 |
-
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
17 |
-
should probably proofread and complete it, then remove this comment. -->
|
18 |
-
|
19 |
# vulnerability-severity-classification-distilbert-base-uncased
|
20 |
|
21 |
This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on the dataset [CIRCL/vulnerability-scores](https://huggingface.co/datasets/CIRCL/vulnerability-scores).
|
22 |
|
|
|
|
|
|
|
|
|
|
|
23 |
It achieves the following results on the evaluation set:
|
24 |
- Loss: 0.6447
|
25 |
- Accuracy: 0.7595
|
|
|
1 |
---
|
2 |
+
base_model: distilbert-base-uncased
|
3 |
+
datasets:
|
4 |
+
- CIRCL/vulnerability-scores
|
5 |
library_name: transformers
|
6 |
license: apache-2.0
|
|
|
|
|
|
|
7 |
metrics:
|
8 |
- accuracy
|
9 |
+
pipeline_tag: text-classification
|
10 |
+
tags:
|
11 |
+
- generated_from_trainer
|
12 |
+
- security
|
13 |
+
- vulnerability
|
14 |
+
- classification
|
15 |
+
- distilbert
|
16 |
model-index:
|
17 |
- name: vulnerability-severity-classification-distilbert-base-uncased
|
18 |
results: []
|
|
|
|
|
19 |
---
|
20 |
|
|
|
|
|
|
|
21 |
# vulnerability-severity-classification-distilbert-base-uncased
|
22 |
|
23 |
This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on the dataset [CIRCL/vulnerability-scores](https://huggingface.co/datasets/CIRCL/vulnerability-scores).
|
24 |
|
25 |
+
This model is part of the work presented in the paper [VLAI: A RoBERTa-Based Model for Automated Vulnerability Severity Classification](https://huggingface.co/papers/2507.03607).
|
26 |
+
|
27 |
+
**Project Page**: [https://vulnerability.circl.lu](https://vulnerability.circl.lu)
|
28 |
+
**Code Repository**: [https://github.com/vulnerability-lookup/ML-Gateway](https://github.com/vulnerability-lookup/ML-Gateway)
|
29 |
+
|
30 |
It achieves the following results on the evaluation set:
|
31 |
- Loss: 0.6447
|
32 |
- Accuracy: 0.7595
|