Ben10x commited on
Commit
3cd5cc8
·
verified ·
1 Parent(s): e072ace

Model save

Browse files
README.md CHANGED
@@ -4,8 +4,6 @@ license: mit
4
  base_model: EleutherAI/gpt-neo-1.3B
5
  tags:
6
  - generated_from_trainer
7
- datasets:
8
- - Ben10x/MedMentions-MTI881-NER
9
  metrics:
10
  - precision
11
  - recall
@@ -13,26 +11,7 @@ metrics:
13
  - accuracy
14
  model-index:
15
  - name: gpt-medmentions
16
- results:
17
- - task:
18
- name: Token Classification
19
- type: token-classification
20
- dataset:
21
- name: Ben10x/MedMentions-MTI881-NER
22
- type: Ben10x/MedMentions-MTI881-NER
23
- metrics:
24
- - name: Precision
25
- type: precision
26
- value: 0.44261028378758077
27
- - name: Recall
28
- type: recall
29
- value: 0.5340735717918291
30
- - name: F1
31
- type: f1
32
- value: 0.48405930705999844
33
- - name: Accuracy
34
- type: accuracy
35
- value: 0.8431182650696939
36
  ---
37
 
38
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -40,13 +19,13 @@ should probably proofread and complete it, then remove this comment. -->
40
 
41
  # gpt-medmentions
42
 
43
- This model is a fine-tuned version of [EleutherAI/gpt-neo-1.3B](https://huggingface.co/EleutherAI/gpt-neo-1.3B) on the Ben10x/MedMentions-MTI881-NER dataset.
44
  It achieves the following results on the evaluation set:
45
- - Loss: 0.5186
46
- - Precision: 0.4426
47
- - Recall: 0.5341
48
- - F1: 0.4841
49
- - Accuracy: 0.8431
50
 
51
  ## Model description
52
 
@@ -78,11 +57,11 @@ The following hyperparameters were used during training:
78
 
79
  | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
80
  |:-------------:|:-----:|:-----:|:---------------:|:---------:|:------:|:------:|:--------:|
81
- | 0.5309 | 1.0 | 5850 | 0.5358 | 0.4123 | 0.4673 | 0.4381 | 0.8330 |
82
- | 0.3521 | 2.0 | 11700 | 0.5186 | 0.4426 | 0.5341 | 0.4841 | 0.8431 |
83
- | 0.1695 | 3.0 | 17550 | 0.6351 | 0.4681 | 0.4846 | 0.4762 | 0.8485 |
84
- | 0.0711 | 4.0 | 23400 | 0.9214 | 0.4716 | 0.5156 | 0.4926 | 0.8498 |
85
- | 0.0384 | 5.0 | 29250 | 1.1011 | 0.4748 | 0.5144 | 0.4938 | 0.8506 |
86
 
87
 
88
  ### Framework versions
 
4
  base_model: EleutherAI/gpt-neo-1.3B
5
  tags:
6
  - generated_from_trainer
 
 
7
  metrics:
8
  - precision
9
  - recall
 
11
  - accuracy
12
  model-index:
13
  - name: gpt-medmentions
14
+ results: []
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
15
  ---
16
 
17
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
19
 
20
  # gpt-medmentions
21
 
22
+ This model is a fine-tuned version of [EleutherAI/gpt-neo-1.3B](https://huggingface.co/EleutherAI/gpt-neo-1.3B) on an unknown dataset.
23
  It achieves the following results on the evaluation set:
24
+ - Loss: 1.0903
25
+ - Precision: 0.4758
26
+ - Recall: 0.5181
27
+ - F1: 0.4960
28
+ - Accuracy: 0.8518
29
 
30
  ## Model description
31
 
 
57
 
58
  | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
59
  |:-------------:|:-----:|:-----:|:---------------:|:---------:|:------:|:------:|:--------:|
60
+ | 0.5307 | 1.0 | 5850 | 0.5369 | 0.4129 | 0.4711 | 0.4401 | 0.8341 |
61
+ | 0.3585 | 2.0 | 11700 | 0.5111 | 0.4453 | 0.5247 | 0.4818 | 0.8454 |
62
+ | 0.1758 | 3.0 | 17550 | 0.6349 | 0.4718 | 0.4900 | 0.4807 | 0.8497 |
63
+ | 0.0751 | 4.0 | 23400 | 0.9264 | 0.4628 | 0.5208 | 0.4901 | 0.8497 |
64
+ | 0.0387 | 5.0 | 29250 | 1.0903 | 0.4758 | 0.5181 | 0.4960 | 0.8518 |
65
 
66
 
67
  ### Framework versions
model-00001-of-00002.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7d4eb0bf2a9c54ad5a94655691ade3b2c25f4b67937a03df0c469e5a243b9489
3
  size 4993794184
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:30717d510e73efac7ea091447f27cb3041934da161fdc8b8a2b5aeb507ea007d
3
  size 4993794184
model-00002-of-00002.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8b87b92a458b9f83199839826d26700d09f14291465a47cbdda29f25ca8619f3
3
  size 269429120
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0b83ae006bf803333da263092a79cf7c0e3d10a895ee55087bbe49ceb45d3a41
3
  size 269429120