Hugging Face's logo

CIRCL
/
cwe-parent-vulnerability-classification-roberta-base

Text Classification
Transformers
PyTorch
Safetensors
roberta
Generated from Trainer
classification
nlp
vulnerability
CWE
text-embeddings-inference
Model card Files
xet
Community
elselse commited on
Commit
ffa921e
·
verified ·
1 Parent(s): 2e48069

End of training

Browse files
Files changed (5) hide show
  1. README.md +8 -8
  2. emissions.csv +1 -1
  3. metrics.json +6 -6
  4. model.safetensors +1 -1
  5. pytorch_model.bin +1 -1
README.md CHANGED
@@ -16,9 +16,9 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 0.0437
20
- - F1 Macro: 0.0
21
- - Exact Match: 1.0
22
 
23
  ## Model description
24
 
@@ -49,11 +49,11 @@ The following hyperparameters were used during training:
49
 
50
  | Training Loss | Epoch | Step | Validation Loss | F1 Macro | Exact Match |
51
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:-----------:|
52
- | 0.3234 | 1.0 | 25 | 0.0950 | 0.0 | 1.0 |
53
- | 0.1031 | 2.0 | 50 | 0.0598 | 0.0 | 1.0 |
54
- | 0.0729 | 3.0 | 75 | 0.0493 | 0.0 | 1.0 |
55
- | 0.0557 | 4.0 | 100 | 0.0450 | 0.0 | 1.0 |
56
- | 0.0527 | 5.0 | 125 | 0.0437 | 0.0 | 1.0 |
57
 
58
 
59
  ### Framework versions
 
16
 
17
  This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 2.2966
20
+ - F1 Macro: 0.0453
21
+ - Exact Match: 0.0
22
 
23
  ## Model description
24
 
 
49
 
50
  | Training Loss | Epoch | Step | Validation Loss | F1 Macro | Exact Match |
51
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:-----------:|
52
+ | 1.1881 | 1.0 | 25 | 2.2966 | 0.0453 | 0.0 |
53
+ | 1.1994 | 2.0 | 50 | 2.6744 | 0.0336 | 0.0 |
54
+ | 0.9102 | 3.0 | 75 | 3.0701 | 0.0566 | 0.0 |
55
+ | 1.339 | 4.0 | 100 | 3.0234 | 0.0436 | 0.0 |
56
+ | 1.5817 | 5.0 | 125 | 3.0559 | 0.0428 | 0.0 |
57
 
58
 
59
  ### 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-08-28T13:04:19,codecarbon,bcb16cd7-bb6d-410c-b997-8b2ce64b5319,5b0fa12a-3dd7-45bb-9766-cc326314d9f1,42.758858100976795,0.0005566582427749225,1.3018547910244732e-05,42.5,265.0106673145598,94.34468507766725,0.0005045715434743519,0.0036637176531897353,0.0011199720420895331,0.005288261238753621,Luxembourg,LUX,luxembourg,,,Linux-6.8.0-71-generic-x86_64-with-glibc2.39,3.12.3,2.8.4,64,AMD EPYC 9124 16-Core Processor,2,2 x NVIDIA L40S,6.1294,49.6113,251.5858268737793,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-08-28T13:34:12,codecarbon,7961b0e1-5620-405c-8f72-d9579535c0d6,5b0fa12a-3dd7-45bb-9766-cc326314d9f1,40.892827729927376,0.0005375164639497293,1.314451686979691e-05,42.5,273.5641178292843,94.34468507766725,0.0004821247777324364,0.0035540981210573364,0.0010701911683537544,0.005106414067143528,Luxembourg,LUX,luxembourg,,,Linux-6.8.0-71-generic-x86_64-with-glibc2.39,3.12.3,2.8.4,64,AMD EPYC 9124 16-Core Processor,2,2 x NVIDIA L40S,6.1294,49.6113,251.5858268737793,machine,N,1.0
metrics.json CHANGED
@@ -1,9 +1,9 @@
1
  {
2
- "eval_loss": 0.043746404349803925,
3
- "eval_f1_macro": 0.0,
4
- "eval_exact_match": 1.0,
5
- "eval_runtime": 0.29,
6
- "eval_samples_per_second": 306.863,
7
- "eval_steps_per_second": 10.344,
8
  "epoch": 5.0
9
  }
 
1
  {
2
+ "eval_loss": 2.2966420650482178,
3
+ "eval_f1_macro": 0.04526290158667233,
4
+ "eval_exact_match": 0.0,
5
+ "eval_runtime": 0.2635,
6
+ "eval_samples_per_second": 333.949,
7
+ "eval_steps_per_second": 11.385,
8
  "epoch": 5.0
9
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5385a2dfa707bda92d43664f4826845291c71953a5d2ef6328e80743fff1bcda
3
  size 498686824
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f1d7616b217d79f4034f42ef9b6ef7c03073fd78234bc9cfb5c7b25533f0021d
3
  size 498686824
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5234d3229249456cd8ed6dd8f0c477e22edee962449f6f65fcf7eee2ac051425
3
  size 498747403
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6e8c8385d269758f0f939fe421ef7d4f05d8c7b30b35fc1b818deb58cc93fc39
3
  size 498747403