Hugging Face's logo

felipe-nextly
/
text-classification-medical

Text Classification
Transformers
TensorBoard
Safetensors
bert
Generated from Trainer
text-embeddings-inference
Model card Files
xet
 Metrics Community
1
felipe-nextly commited on
Commit
8b97b3a
·
1 Parent(s): bd92449

Model save

Browse files
Files changed (1) hide show
  1. README.md +12 -12
README.md CHANGED
@@ -18,7 +18,7 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on the None dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.0512
22
  - Accuracy: 1.0
23
  - F1: 1.0
24
 
@@ -51,21 +51,21 @@ The following hyperparameters were used during training:
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
53
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
54
- | No log | 1.0 | 13 | 0.6337 | 0.6154 | 0.7235 |
55
- | No log | 2.0 | 26 | 0.5057 | 0.8736 | 0.8730 |
56
- | No log | 3.0 | 39 | 0.3518 | 0.9396 | 0.9402 |
57
- | No log | 4.0 | 52 | 0.2269 | 0.9835 | 0.9836 |
58
- | No log | 5.0 | 65 | 0.1601 | 0.9945 | 0.9945 |
59
- | No log | 6.0 | 78 | 0.1092 | 1.0 | 1.0 |
60
- | No log | 7.0 | 91 | 0.0803 | 1.0 | 1.0 |
61
- | No log | 8.0 | 104 | 0.0637 | 1.0 | 1.0 |
62
- | No log | 9.0 | 117 | 0.0541 | 1.0 | 1.0 |
63
- | No log | 10.0 | 130 | 0.0512 | 1.0 | 1.0 |
64
 
65
 
66
  ### Framework versions
67
 
68
- - Transformers 4.36.1
69
  - Pytorch 2.1.0+cu121
70
  - Datasets 2.15.0
71
  - Tokenizers 0.15.0
 
18
 
19
  This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on the None dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 0.0394
22
  - Accuracy: 1.0
23
  - F1: 1.0
24
 
 
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
53
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
54
+ | No log | 1.0 | 13 | 0.6409 | 0.5714 | 0.7273 |
55
+ | No log | 2.0 | 26 | 0.5385 | 0.8297 | 0.8397 |
56
+ | No log | 3.0 | 39 | 0.3346 | 0.9286 | 0.9293 |
57
+ | No log | 4.0 | 52 | 0.1979 | 0.9780 | 0.9781 |
58
+ | No log | 5.0 | 65 | 0.1321 | 0.9945 | 0.9945 |
59
+ | No log | 6.0 | 78 | 0.0932 | 1.0 | 1.0 |
60
+ | No log | 7.0 | 91 | 0.0654 | 1.0 | 1.0 |
61
+ | No log | 8.0 | 104 | 0.0508 | 1.0 | 1.0 |
62
+ | No log | 9.0 | 117 | 0.0420 | 1.0 | 1.0 |
63
+ | No log | 10.0 | 130 | 0.0394 | 1.0 | 1.0 |
64
 
65
 
66
  ### Framework versions
67
 
68
+ - Transformers 4.36.2
69
  - Pytorch 2.1.0+cu121
70
  - Datasets 2.15.0
71
  - Tokenizers 0.15.0