Hugging Face's logo

CIRCL
/
vuln-patch-cwe-guesser-model-huggingface-CodeBERTa-small-v1

Text Classification
Transformers
Safetensors
roberta
Generated from Trainer
Model card Files
xet
Community
elselse commited on
Commit
3c84579
·
verified ·
1 Parent(s): 2f2f3d6

End of training

Browse files
Files changed (3) hide show
  1. README.md +8 -8
  2. metrics.json +6 -6
  3. model.safetensors +1 -1
README.md CHANGED
@@ -18,9 +18,9 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  This model is a fine-tuned version of [huggingface/CodeBERTa-small-v1](https://huggingface.co/huggingface/CodeBERTa-small-v1) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 4.2891
22
- - Accuracy: 0.4
23
- - F1: 0.0519
24
 
25
  ## Model description
26
 
@@ -52,11 +52,11 @@ The following hyperparameters were used during training:
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
54
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
55
- | No log | 1.0 | 1 | 4.8834 | 0.35 | 0.0431 |
56
- | No log | 2.0 | 2 | 4.6211 | 0.4 | 0.0519 |
57
- | No log | 3.0 | 3 | 4.4458 | 0.4 | 0.0519 |
58
- | No log | 4.0 | 4 | 4.3395 | 0.4 | 0.0519 |
59
- | No log | 5.0 | 5 | 4.2891 | 0.4 | 0.0519 |
60
 
61
 
62
  ### Framework versions
 
18
 
19
  This model is a fine-tuned version of [huggingface/CodeBERTa-small-v1](https://huggingface.co/huggingface/CodeBERTa-small-v1) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 4.5509
22
+ - Accuracy: 0.25
23
+ - F1: 0.0333
24
 
25
  ## Model description
26
 
 
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
54
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
55
+ | No log | 1.0 | 1 | 4.9917 | 0.25 | 0.0333 |
56
+ | No log | 2.0 | 2 | 4.7972 | 0.25 | 0.0333 |
57
+ | No log | 3.0 | 3 | 4.6686 | 0.25 | 0.0333 |
58
+ | No log | 4.0 | 4 | 4.5891 | 0.25 | 0.0333 |
59
+ | No log | 5.0 | 5 | 4.5509 | 0.25 | 0.0333 |
60
 
61
 
62
  ### Framework versions
metrics.json CHANGED
@@ -1,9 +1,9 @@
1
  {
2
- "eval_loss": 4.289130687713623,
3
- "eval_accuracy": 0.4,
4
- "eval_f1": 0.051948051948051945,
5
- "eval_runtime": 0.0811,
6
- "eval_samples_per_second": 246.553,
7
- "eval_steps_per_second": 12.328,
8
  "epoch": 5.0
9
  }
 
1
  {
2
+ "eval_loss": 4.550906181335449,
3
+ "eval_accuracy": 0.25,
4
+ "eval_f1": 0.03333333333333333,
5
+ "eval_runtime": 0.1019,
6
+ "eval_samples_per_second": 196.363,
7
+ "eval_steps_per_second": 9.818,
8
  "epoch": 5.0
9
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c0296b7f3b58eb7acf98f8f775d58d610d41673cc2d3a15c6e70b217529ff62b
3
  size 334388184
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6f010fd4836a73f9db92aed022abd1929cc0c6bd9608b85c64268a3876e9c2bc
3
  size 334388184