pabRomero commited on
Commit
a90c332
1 Parent(s): 504f6a5

Training complete

Browse files
Files changed (1) hide show
  1. README.md +8 -8
README.md CHANGED
@@ -21,11 +21,11 @@ should probably proofread and complete it, then remove this comment. -->
21
 
22
  This model is a fine-tuned version of [FacebookAI/roberta-base](https://huggingface.co/FacebookAI/roberta-base) on the None dataset.
23
  It achieves the following results on the evaluation set:
24
- - Loss: 0.0616
25
- - Precision: 0.8439
26
- - Recall: 0.8346
27
- - F1: 0.8392
28
- - Accuracy: 0.9811
29
 
30
  ## Model description
31
 
@@ -58,13 +58,13 @@ The following hyperparameters were used during training:
58
 
59
  | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
60
  |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
61
- | No log | 1.0 | 231 | 0.0805 | 0.7890 | 0.8039 | 0.7964 | 0.9755 |
62
- | No log | 2.0 | 462 | 0.0616 | 0.8439 | 0.8346 | 0.8392 | 0.9811 |
63
 
64
 
65
  ### Framework versions
66
 
67
  - Transformers 4.44.2
68
- - Pytorch 2.4.1+cu121
69
  - Datasets 2.21.0
70
  - Tokenizers 0.19.1
 
21
 
22
  This model is a fine-tuned version of [FacebookAI/roberta-base](https://huggingface.co/FacebookAI/roberta-base) on the None dataset.
23
  It achieves the following results on the evaluation set:
24
+ - Loss: 0.0765
25
+ - Precision: 0.7970
26
+ - Recall: 0.7733
27
+ - F1: 0.7850
28
+ - Accuracy: 0.9763
29
 
30
  ## Model description
31
 
 
58
 
59
  | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
60
  |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
61
+ | No log | 1.0 | 231 | 0.0904 | 0.7605 | 0.7562 | 0.7583 | 0.9723 |
62
+ | No log | 2.0 | 462 | 0.0765 | 0.7970 | 0.7733 | 0.7850 | 0.9763 |
63
 
64
 
65
  ### Framework versions
66
 
67
  - Transformers 4.44.2
68
+ - Pytorch 2.4.0+cu121
69
  - Datasets 2.21.0
70
  - Tokenizers 0.19.1