Hugging Face's logo

skylord
/
pharma_classification

Text Classification
Transformers
Safetensors
distilbert
Generated from Trainer
Inference Endpoints
Model card Files Community
skylord commited on
Commit
956a966
•
1 Parent(s): 042cc40

skylord/pharma_classification

Browse files
Files changed (3) hide show
  1. README.md +12 -10
  2. model.safetensors +1 -1
  3. training_args.bin +1 -1
README.md CHANGED
@@ -5,6 +5,7 @@ tags:
5
  - generated_from_trainer
6
  metrics:
7
  - accuracy
 
8
  model-index:
9
  - name: pharma_classification
10
  results: []
@@ -17,8 +18,9 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the None dataset.
19
  It achieves the following results on the evaluation set:
20
- - Loss: 0.6035
21
- - Accuracy: 0.9664
 
22
 
23
  ## Model description
24
 
@@ -47,14 +49,14 @@ The following hyperparameters were used during training:
47
 
48
  ### Training results
49
 
50
- | Training Loss | Epoch | Step | Validation Loss | Accuracy |
51
- |:-------------:|:-----:|:-----:|:---------------:|:--------:|
52
- | 0.0194 | 5.99 | 5000 | 0.2594 | 0.9635 |
53
- | 0.0 | 11.98 | 10000 | 0.4335 | 0.9641 |
54
- | 0.0 | 17.96 | 15000 | 0.5338 | 0.9641 |
55
- | 0.0 | 23.95 | 20000 | 0.4973 | 0.9664 |
56
- | 0.0 | 29.94 | 25000 | 0.5737 | 0.9664 |
57
- | 0.0 | 35.93 | 30000 | 0.6035 | 0.9664 |
58
 
59
 
60
  ### Framework versions
 
5
  - generated_from_trainer
6
  metrics:
7
  - accuracy
8
+ - f1
9
  model-index:
10
  - name: pharma_classification
11
  results: []
 
18
 
19
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the None dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 0.5315
22
+ - Accuracy: 0.9581
23
+ - F1: 0.9506
24
 
25
  ## Model description
26
 
 
49
 
50
  ### Training results
51
 
52
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
53
+ |:-------------:|:-----:|:-----:|:---------------:|:--------:|:------:|
54
+ | 0.0035 | 5.99 | 5000 | 0.2892 | 0.9539 | 0.9554 |
55
+ | 0.0137 | 11.98 | 10000 | 0.2620 | 0.9641 | 0.9600 |
56
+ | 0.0 | 17.96 | 15000 | 0.4022 | 0.9611 | 0.9586 |
57
+ | 0.0001 | 23.95 | 20000 | 0.3838 | 0.9611 | 0.9552 |
58
+ | 0.0 | 29.94 | 25000 | 0.4363 | 0.9575 | 0.9490 |
59
+ | 0.0 | 35.93 | 30000 | 0.5315 | 0.9581 | 0.9506 |
60
 
61
 
62
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0e695507e37a2a5698ecc8b53af1f86b1052ff0d384f50ecc57e550d3287df2f
3
  size 267832560
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1d0bf5a7bf66dbb4ef0d0393230fb7b2d244eb97c8fa6b6a1d3dd76fd76a004c
3
  size 267832560
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0ca5a2cfe4d31926d2ba1eb360f57ebc70f08a88ec5600be4e5539e4c0228350
3
  size 4920
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d03217c4bd9bcefed5aac74887e8af89205ab97fa69573fb1624a269f263ed02
3
  size 4920