Hugging Face's logo

fawern
/
clinical-miniALBERT-312-medical-text-classification

Text Classification
Transformers
Safetensors
bert
Generated from Trainer
Inference Endpoints
Model card Files Community
fawern commited on
Commit
4ed77be
•
1 Parent(s): 8201e3c

End of training

Browse files
Files changed (2) hide show
  1. README.md +15 -17
  2. model.safetensors +1 -1
README.md CHANGED
@@ -20,11 +20,11 @@ should probably proofread and complete it, then remove this comment. -->
20
 
21
  This model is a fine-tuned version of [nlpie/clinical-miniALBERT-312](https://huggingface.co/nlpie/clinical-miniALBERT-312) on an unknown dataset.
22
  It achieves the following results on the evaluation set:
23
- - Loss: 2.4405
24
- - Accuracy: 0.169
25
- - Precision: 0.1511
26
- - Recall: 0.169
27
- - F1: 0.1577
28
 
29
  ## Model description
30
 
@@ -56,18 +56,16 @@ The following hyperparameters were used during training:
56
 
57
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1 |
58
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|
59
- | 2.844 | 1.0 | 250 | 2.8817 | 0.281 | 0.0904 | 0.281 | 0.1366 |
60
- | 2.3646 | 2.0 | 500 | 2.5907 | 0.345 | 0.1361 | 0.345 | 0.1938 |
61
- | 2.4893 | 3.0 | 750 | 2.4661 | 0.366 | 0.1872 | 0.366 | 0.2361 |
62
- | 2.3836 | 4.0 | 1000 | 2.4117 | 0.364 | 0.1864 | 0.364 | 0.2385 |
63
- | 2.1529 | 5.0 | 1250 | 2.4233 | 0.35 | 0.2074 | 0.35 | 0.2480 |
64
- | 1.9937 | 6.0 | 1500 | 2.3446 | 0.324 | 0.2192 | 0.324 | 0.2528 |
65
- | 1.9979 | 7.0 | 1750 | 2.3319 | 0.276 | 0.2039 | 0.276 | 0.2275 |
66
- | 1.7607 | 8.0 | 2000 | 2.2962 | 0.258 | 0.1952 | 0.258 | 0.2169 |
67
- | 1.7754 | 9.0 | 2250 | 2.2418 | 0.244 | 0.1852 | 0.244 | 0.1986 |
68
- | 1.4652 | 10.0 | 2500 | 2.2749 | 0.198 | 0.1720 | 0.198 | 0.1812 |
69
- | 1.2902 | 11.0 | 2750 | 2.3244 | 0.193 | 0.1723 | 0.193 | 0.1782 |
70
- | 1.375 | 12.0 | 3000 | 2.4405 | 0.169 | 0.1511 | 0.169 | 0.1577 |
71
 
72
 
73
  ### Framework versions
 
20
 
21
  This model is a fine-tuned version of [nlpie/clinical-miniALBERT-312](https://huggingface.co/nlpie/clinical-miniALBERT-312) on an unknown dataset.
22
  It achieves the following results on the evaluation set:
23
+ - Loss: 2.3237
24
+ - Accuracy: 0.186
25
+ - Precision: 0.1659
26
+ - Recall: 0.186
27
+ - F1: 0.1736
28
 
29
  ## Model description
30
 
 
56
 
57
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1 |
58
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|
59
+ | 2.8556 | 1.0 | 250 | 2.9403 | 0.221 | 0.0488 | 0.221 | 0.0800 |
60
+ | 2.3771 | 2.0 | 500 | 2.6223 | 0.324 | 0.1226 | 0.324 | 0.1772 |
61
+ | 2.4809 | 3.0 | 750 | 2.4574 | 0.365 | 0.1918 | 0.365 | 0.2409 |
62
+ | 2.3648 | 4.0 | 1000 | 2.4098 | 0.34 | 0.1762 | 0.34 | 0.2241 |
63
+ | 2.1697 | 5.0 | 1250 | 2.4202 | 0.346 | 0.2038 | 0.346 | 0.2457 |
64
+ | 2.0393 | 6.0 | 1500 | 2.3783 | 0.333 | 0.2220 | 0.333 | 0.2564 |
65
+ | 1.8836 | 7.0 | 1750 | 2.3133 | 0.279 | 0.2113 | 0.279 | 0.2289 |
66
+ | 1.6791 | 8.0 | 2000 | 2.2241 | 0.256 | 0.2115 | 0.256 | 0.2260 |
67
+ | 1.6044 | 9.0 | 2250 | 2.2828 | 0.235 | 0.1965 | 0.235 | 0.2055 |
68
+ | 1.4025 | 10.0 | 2500 | 2.3237 | 0.186 | 0.1659 | 0.186 | 0.1736 |
 
 
69
 
70
 
71
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:76cac145539f620b3fbe647e5e7ef2dc40d7f344b7fcd71407bd80b1ab0427b9
3
  size 263267400
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a3aa3c73c765d3fb539b1027c51930c06454372e0c12b340e26e36314d6b0bd1
3
  size 263267400