Hugging Face's logo

fawern
/
Bio_ClinicalBERT-medical-text-classification

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

End of training

Browse files
Files changed (2) hide show
  1. README.md +11 -12
  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 [emilyalsentzer/Bio_ClinicalBERT](https://huggingface.co/emilyalsentzer/Bio_ClinicalBERT) on an unknown dataset.
22
  It achieves the following results on the evaluation set:
23
- - Loss: 1.9963
24
- - Accuracy: 0.221
25
- - Precision: 0.1982
26
- - Recall: 0.221
27
- - F1: 0.2013
28
 
29
  ## Model description
30
 
@@ -56,13 +56,12 @@ The following hyperparameters were used during training:
56
 
57
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1 |
58
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|
59
- | 2.4889 | 1.0 | 250 | 2.5603 | 0.353 | 0.1678 | 0.353 | 0.2178 |
60
- | 1.885 | 2.0 | 500 | 2.0021 | 0.371 | 0.2706 | 0.371 | 0.2725 |
61
- | 1.7769 | 3.0 | 750 | 1.8414 | 0.318 | 0.2762 | 0.318 | 0.2822 |
62
- | 1.4955 | 4.0 | 1000 | 1.7687 | 0.315 | 0.2943 | 0.315 | 0.2870 |
63
- | 1.3151 | 5.0 | 1250 | 1.8651 | 0.315 | 0.2527 | 0.315 | 0.2723 |
64
- | 1.235 | 6.0 | 1500 | 1.9358 | 0.254 | 0.2298 | 0.254 | 0.2354 |
65
- | 1.2697 | 7.0 | 1750 | 1.9963 | 0.221 | 0.1982 | 0.221 | 0.2013 |
66
 
67
 
68
  ### Framework versions
 
20
 
21
  This model is a fine-tuned version of [emilyalsentzer/Bio_ClinicalBERT](https://huggingface.co/emilyalsentzer/Bio_ClinicalBERT) on an unknown dataset.
22
  It achieves the following results on the evaluation set:
23
+ - Loss: 1.8941
24
+ - Accuracy: 0.273
25
+ - Precision: 0.2486
26
+ - Recall: 0.273
27
+ - F1: 0.2532
28
 
29
  ## Model description
30
 
 
56
 
57
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1 |
58
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|
59
+ | 2.4866 | 1.0 | 250 | 2.5436 | 0.355 | 0.1460 | 0.355 | 0.2036 |
60
+ | 1.9145 | 2.0 | 500 | 2.0555 | 0.369 | 0.2437 | 0.369 | 0.2406 |
61
+ | 1.849 | 3.0 | 750 | 1.8421 | 0.321 | 0.2862 | 0.321 | 0.2949 |
62
+ | 1.4025 | 4.0 | 1000 | 1.7678 | 0.325 | 0.2950 | 0.325 | 0.2957 |
63
+ | 1.311 | 5.0 | 1250 | 1.8007 | 0.312 | 0.2654 | 0.312 | 0.2743 |
64
+ | 1.2112 | 6.0 | 1500 | 1.8941 | 0.273 | 0.2486 | 0.273 | 0.2532 |
 
65
 
66
 
67
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:59e42543bed8ab971fc7618e3a2c3e2a7584158766d56128d61c36ba4a7b6226
3
  size 433387656
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:be29cf1f99342feabb6bd3b8800116d3284944ba5bee07522f2bac527d01657a
3
  size 433387656