cedricbonhomme nielsr HF Staff commited on
Commit
bfc1c12
·
verified ·
1 Parent(s): f5168c8

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]>

Files changed (1) hide show
  1. README.md +15 -8
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