Hugging Face's logo

CIRCL
/
vulnerability-severity-classification-roberta-base

Text Classification
Transformers
Safetensors
roberta
Generated from Trainer
classification
nlp
vulnerability
Model card Files
xet
Community
cedricbonhomme commited on
Commit
cae98c2
·
verified ·
1 Parent(s): 2e0fe28

End of training

Browse files
Files changed (3) hide show
  1. README.md +7 -7
  2. emissions.csv +1 -1
  3. model.safetensors +1 -1
README.md CHANGED
@@ -18,8 +18,8 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.5095
22
- - Accuracy: 0.8245
23
 
24
  ## Model description
25
 
@@ -50,11 +50,11 @@ The following hyperparameters were used during training:
50
 
51
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
52
  |:-------------:|:-----:|:------:|:---------------:|:--------:|
53
- | 0.5087 | 1.0 | 26400 | 0.6427 | 0.7397 |
54
- | 0.4433 | 2.0 | 52800 | 0.5840 | 0.7693 |
55
- | 0.5995 | 3.0 | 79200 | 0.5373 | 0.7951 |
56
- | 0.4051 | 4.0 | 105600 | 0.5059 | 0.8154 |
57
- | 0.38 | 5.0 | 132000 | 0.5095 | 0.8245 |
58
 
59
 
60
  ### Framework versions
 
18
 
19
  This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 0.5043
22
+ - Accuracy: 0.8266
23
 
24
  ## Model description
25
 
 
50
 
51
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
52
  |:-------------:|:-----:|:------:|:---------------:|:--------:|
53
+ | 0.6699 | 1.0 | 26473 | 0.6352 | 0.7442 |
54
+ | 0.5909 | 2.0 | 52946 | 0.5688 | 0.7736 |
55
+ | 0.4787 | 3.0 | 79419 | 0.5192 | 0.7981 |
56
+ | 0.4534 | 4.0 | 105892 | 0.5025 | 0.8174 |
57
+ | 0.3843 | 5.0 | 132365 | 0.5043 | 0.8266 |
58
 
59
 
60
  ### Framework versions
emissions.csv CHANGED
@@ -1,2 +1,2 @@
1
  timestamp,project_name,run_id,experiment_id,duration,emissions,emissions_rate,cpu_power,gpu_power,ram_power,cpu_energy,gpu_energy,ram_energy,energy_consumed,country_name,country_iso_code,region,cloud_provider,cloud_region,os,python_version,codecarbon_version,cpu_count,cpu_model,gpu_count,gpu_model,longitude,latitude,ram_total_size,tracking_mode,on_cloud,pue
2
- 2025-04-01T14:52:08,codecarbon,5ccddff8-ac04-4987-a99e-6c9ccda53bd9,5b0fa12a-3dd7-45bb-9766-cc326314d9f1,21771.618839884177,0.39048864465868255,1.7935673388849385e-05,42.5,183.91220637717836,94.34470081329346,0.2568590272682563,2.8826165966470114,0.5701720648137456,3.709647688729017,Luxembourg,LUX,luxembourg,,,Linux-6.8.0-48-generic-x86_64-with-glibc2.39,3.12.3,2.8.3,64,AMD EPYC 9124 16-Core Processor,2,2 x NVIDIA L40S,6.1294,49.6113,251.58586883544922,machine,N,1.0
 
1
  timestamp,project_name,run_id,experiment_id,duration,emissions,emissions_rate,cpu_power,gpu_power,ram_power,cpu_energy,gpu_energy,ram_energy,energy_consumed,country_name,country_iso_code,region,cloud_provider,cloud_region,os,python_version,codecarbon_version,cpu_count,cpu_model,gpu_count,gpu_model,longitude,latitude,ram_total_size,tracking_mode,on_cloud,pue
2
+ 2025-04-04T11:15:16,codecarbon,444803ec-c9b0-400e-aeaf-53be6a2fa5e6,5b0fa12a-3dd7-45bb-9766-cc326314d9f1,21771.339703876525,0.3901343951688573,1.791963197833854e-05,42.5,182.52661961740924,94.34470081329346,0.25685810986615504,2.879254088679204,0.5701701149822527,3.7062823135276144,Luxembourg,LUX,luxembourg,,,Linux-6.8.0-48-generic-x86_64-with-glibc2.39,3.12.3,2.8.3,64,AMD EPYC 9124 16-Core Processor,2,2 x NVIDIA L40S,6.1294,49.6113,251.58586883544922,machine,N,1.0
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9a02a9b9f295142afebc0e1d9dfb871f3edea2a061fa4ce825e8f2ea6457541a
3
  size 498618976
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:71ce549b41b244ebb15cbef663a87d1e06ba12bc475b95268ffa3cd5f80d2bcb
3
  size 498618976