mamiksik commited on
Commit
91daf39
1 Parent(s): d7aa311

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -9
README.md CHANGED
@@ -3,6 +3,7 @@ tags:
3
  - generated_from_trainer
4
  metrics:
5
  - accuracy
 
6
  model-index:
7
  - name: CommitPredictor
8
  results: []
@@ -15,8 +16,10 @@ should probably proofread and complete it, then remove this comment. -->
15
 
16
  This model is a fine-tuned version of [microsoft/codebert-base-mlm](https://huggingface.co/microsoft/codebert-base-mlm) on the None dataset.
17
  It achieves the following results on the evaluation set:
18
- - Loss: 0.5999
19
- - Accuracy: 0.8784
 
 
20
 
21
  ## Model description
22
 
@@ -45,13 +48,13 @@ The following hyperparameters were used during training:
45
 
46
  ### Training results
47
 
48
- | Training Loss | Epoch | Step | Validation Loss | Accuracy |
49
- |:-------------:|:-----:|:----:|:---------------:|:--------:|
50
- | 0.8473 | 1.0 | 1373 | 0.6892 | 0.8579 |
51
- | 0.7042 | 2.0 | 2746 | 0.6433 | 0.8665 |
52
- | 0.6607 | 3.0 | 4119 | 0.6184 | 0.8745 |
53
- | 0.6539 | 4.0 | 5492 | 0.6051 | 0.8775 |
54
- | 0.6286 | 5.0 | 6865 | 0.5999 | 0.8784 |
55
 
56
 
57
  ### Framework versions
 
3
  - generated_from_trainer
4
  metrics:
5
  - accuracy
6
+ - f1
7
  model-index:
8
  - name: CommitPredictor
9
  results: []
 
16
 
17
  This model is a fine-tuned version of [microsoft/codebert-base-mlm](https://huggingface.co/microsoft/codebert-base-mlm) on the None dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 2.1076
20
+ - Accuracy: 0.7284
21
+ - F1: 0.7284
22
+ - Bleu4: 0.7262
23
 
24
  ## Model description
25
 
 
48
 
49
  ### Training results
50
 
51
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Bleu4 |
52
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:------:|
53
+ | No log | 1.0 | 1 | 2.4321 | 0.7137 | 0.7137 | 0.5207 |
54
+ | No log | 2.0 | 2 | 2.2808 | 0.7186 | 0.7186 | 0.6834 |
55
+ | No log | 3.0 | 3 | 2.1813 | 0.7273 | 0.7273 | 0.6974 |
56
+ | No log | 4.0 | 4 | 2.2816 | 0.7316 | 0.7316 | 0.6591 |
57
+ | No log | 5.0 | 5 | 2.1076 | 0.7284 | 0.7284 | 0.7262 |
58
 
59
 
60
  ### Framework versions