Hugging Face's logo

fawern
/
distil-clinicalbert-medical-text-classification

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

End of training

Browse files
Files changed (2) hide show
  1. README.md +11 -12
  2. model.safetensors +1 -1
README.md CHANGED
@@ -19,11 +19,11 @@ should probably proofread and complete it, then remove this comment. -->
19
 
20
  This model is a fine-tuned version of [nlpie/distil-clinicalbert](https://huggingface.co/nlpie/distil-clinicalbert) on an unknown dataset.
21
  It achieves the following results on the evaluation set:
22
- - Loss: 1.9496
23
- - Accuracy: 0.226
24
- - Precision: 0.2021
25
- - Recall: 0.226
26
- - F1: 0.2058
27
 
28
  ## Model description
29
 
@@ -55,13 +55,12 @@ The following hyperparameters were used during training:
55
 
56
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1 |
57
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|
58
- | 2.5309 | 1.0 | 250 | 2.6137 | 0.339 | 0.1410 | 0.339 | 0.1943 |
59
- | 2.0359 | 2.0 | 500 | 2.1867 | 0.357 | 0.2377 | 0.357 | 0.2432 |
60
- | 1.9003 | 3.0 | 750 | 1.9129 | 0.324 | 0.2801 | 0.324 | 0.2882 |
61
- | 1.5959 | 4.0 | 1000 | 1.8227 | 0.317 | 0.2995 | 0.317 | 0.2904 |
62
- | 1.3873 | 5.0 | 1250 | 1.8668 | 0.302 | 0.2670 | 0.302 | 0.2740 |
63
- | 1.3281 | 6.0 | 1500 | 1.8773 | 0.26 | 0.2264 | 0.26 | 0.2358 |
64
- | 1.3854 | 7.0 | 1750 | 1.9496 | 0.226 | 0.2021 | 0.226 | 0.2058 |
65
 
66
 
67
  ### Framework versions
 
19
 
20
  This model is a fine-tuned version of [nlpie/distil-clinicalbert](https://huggingface.co/nlpie/distil-clinicalbert) on an unknown dataset.
21
  It achieves the following results on the evaluation set:
22
+ - Loss: 1.8719
23
+ - Accuracy: 0.266
24
+ - Precision: 0.2357
25
+ - Recall: 0.266
26
+ - F1: 0.2427
27
 
28
  ## Model description
29
 
 
55
 
56
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1 |
57
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|
58
+ | 2.5512 | 1.0 | 250 | 2.6302 | 0.335 | 0.1402 | 0.335 | 0.1911 |
59
+ | 2.0609 | 2.0 | 500 | 2.1857 | 0.357 | 0.2240 | 0.357 | 0.2474 |
60
+ | 1.9056 | 3.0 | 750 | 1.8964 | 0.321 | 0.2773 | 0.321 | 0.2812 |
61
+ | 1.5646 | 4.0 | 1000 | 1.8117 | 0.323 | 0.3183 | 0.323 | 0.2949 |
62
+ | 1.3789 | 5.0 | 1250 | 1.8869 | 0.302 | 0.2643 | 0.302 | 0.2701 |
63
+ | 1.3189 | 6.0 | 1500 | 1.8719 | 0.266 | 0.2357 | 0.266 | 0.2427 |
 
64
 
65
 
66
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2dbab43f9ae43f9ad0e716bd9c2225e73e31e3e16b082eb7185af95b9d31ee66
3
  size 263267400
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:da60f767b649e238d57ca70093d8f0988988b17fc349b5888da734a9c5a822b0
3
  size 263267400