Edison commited on
Commit
5234607
·
1 Parent(s): 0982e5b

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -15
README.md CHANGED
@@ -19,12 +19,12 @@ should probably proofread and complete it, then remove this comment. -->
19
 
20
  This model is a fine-tuned version of [neuralsentry/starencoder-git-commits-mlm](https://huggingface.co/neuralsentry/starencoder-git-commits-mlm) on the None dataset.
21
  It achieves the following results on the evaluation set:
22
- - Loss: 0.1191
23
- - Accuracy: 0.9703
24
- - Precision: 0.9769
25
- - Recall: 0.96
26
- - F1: 0.9684
27
- - Roc Auc: 0.9698
28
 
29
  ## Model description
30
 
@@ -55,15 +55,7 @@ The following hyperparameters were used during training:
55
 
56
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1 | Roc Auc |
57
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|:-------:|
58
- | 0.3716 | 0.33 | 66 | 0.2018 | 0.9296 | 0.9368 | 0.9133 | 0.9249 | 0.9288 |
59
- | 0.1745 | 0.67 | 132 | 0.1468 | 0.9533 | 0.9711 | 0.9293 | 0.9498 | 0.9522 |
60
- | 0.1346 | 1.0 | 198 | 0.1091 | 0.9657 | 0.9761 | 0.951 | 0.9634 | 0.9650 |
61
- | 0.0917 | 1.33 | 264 | 0.1294 | 0.9647 | 0.9790 | 0.946 | 0.9622 | 0.9638 |
62
- | 0.0877 | 1.67 | 330 | 0.1090 | 0.9668 | 0.9619 | 0.9683 | 0.9651 | 0.9669 |
63
- | 0.0731 | 2.0 | 396 | 0.1042 | 0.9688 | 0.9746 | 0.9593 | 0.9669 | 0.9684 |
64
- | 0.0342 | 2.33 | 462 | 0.1291 | 0.9692 | 0.9686 | 0.9663 | 0.9675 | 0.9690 |
65
- | 0.0375 | 2.67 | 528 | 0.1202 | 0.9706 | 0.9753 | 0.9623 | 0.9688 | 0.9702 |
66
- | 0.0342 | 3.0 | 594 | 0.1191 | 0.9703 | 0.9769 | 0.96 | 0.9684 | 0.9698 |
67
 
68
 
69
  ### Framework versions
 
19
 
20
  This model is a fine-tuned version of [neuralsentry/starencoder-git-commits-mlm](https://huggingface.co/neuralsentry/starencoder-git-commits-mlm) on the None dataset.
21
  It achieves the following results on the evaluation set:
22
+ - Loss: 0.2327
23
+ - Accuracy: 0.9274
24
+ - Precision: 0.9691
25
+ - Recall: 0.9199
26
+ - F1: 0.9438
27
+ - Roc Auc: 0.9310
28
 
29
  ## Model description
30
 
 
55
 
56
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1 | Roc Auc |
57
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|:-------:|
58
+ | 0.2872 | 1.57 | 66 | 0.2017 | 0.9177 | 0.9439 | 0.9312 | 0.9375 | 0.9111 |
 
 
 
 
 
 
 
 
59
 
60
 
61
  ### Framework versions