ChakuChidiya
commited on
Commit
•
b0c0776
1
Parent(s):
404c5ef
Upload TFDistilBertForTokenClassification
Browse files- README.md +5 -5
- config.json +8 -8
- tf_model.h5 +1 -1
README.md
CHANGED
@@ -15,8 +15,8 @@ probably proofread and complete it, then remove this comment. -->
|
|
15 |
|
16 |
This model is a fine-tuned version of [ChakuChidiya/distilbert-base-uncased-G2](https://huggingface.co/ChakuChidiya/distilbert-base-uncased-G2) on an unknown dataset.
|
17 |
It achieves the following results on the evaluation set:
|
18 |
-
- Train Loss: 0.
|
19 |
-
- Validation Loss: 0.
|
20 |
- Epoch: 2
|
21 |
|
22 |
## Model description
|
@@ -43,9 +43,9 @@ The following hyperparameters were used during training:
|
|
43 |
|
44 |
| Train Loss | Validation Loss | Epoch |
|
45 |
|:----------:|:---------------:|:-----:|
|
46 |
-
| 0.
|
47 |
-
| 0.
|
48 |
-
| 0.
|
49 |
|
50 |
|
51 |
### Framework versions
|
|
|
15 |
|
16 |
This model is a fine-tuned version of [ChakuChidiya/distilbert-base-uncased-G2](https://huggingface.co/ChakuChidiya/distilbert-base-uncased-G2) on an unknown dataset.
|
17 |
It achieves the following results on the evaluation set:
|
18 |
+
- Train Loss: 0.2192
|
19 |
+
- Validation Loss: 0.3240
|
20 |
- Epoch: 2
|
21 |
|
22 |
## Model description
|
|
|
43 |
|
44 |
| Train Loss | Validation Loss | Epoch |
|
45 |
|:----------:|:---------------:|:-----:|
|
46 |
+
| 0.3628 | 0.3204 | 0 |
|
47 |
+
| 0.2708 | 0.3328 | 1 |
|
48 |
+
| 0.2192 | 0.3240 | 2 |
|
49 |
|
50 |
|
51 |
### Framework versions
|
config.json
CHANGED
@@ -10,25 +10,25 @@
|
|
10 |
"hidden_dim": 3072,
|
11 |
"id2label": {
|
12 |
"0": "O",
|
13 |
-
"1": "B-
|
14 |
-
"2": "I-
|
15 |
"3": "B-cancer",
|
16 |
"4": "I-cancer",
|
17 |
"5": "B-chronic_disease",
|
18 |
"6": "I-chronic_disease",
|
19 |
-
"7": "B-
|
20 |
-
"8": "I-
|
21 |
},
|
22 |
"initializer_range": 0.02,
|
23 |
"label2id": {
|
24 |
-
"B-allergy_name":
|
25 |
"B-cancer": 3,
|
26 |
"B-chronic_disease": 5,
|
27 |
-
"B-treatment":
|
28 |
-
"I-allergy_name":
|
29 |
"I-cancer": 4,
|
30 |
"I-chronic_disease": 6,
|
31 |
-
"I-treatment":
|
32 |
"O": 0
|
33 |
},
|
34 |
"max_position_embeddings": 512,
|
|
|
10 |
"hidden_dim": 3072,
|
11 |
"id2label": {
|
12 |
"0": "O",
|
13 |
+
"1": "B-allergy_name",
|
14 |
+
"2": "I-allergy_name",
|
15 |
"3": "B-cancer",
|
16 |
"4": "I-cancer",
|
17 |
"5": "B-chronic_disease",
|
18 |
"6": "I-chronic_disease",
|
19 |
+
"7": "B-treatment",
|
20 |
+
"8": "I-treatment"
|
21 |
},
|
22 |
"initializer_range": 0.02,
|
23 |
"label2id": {
|
24 |
+
"B-allergy_name": 1,
|
25 |
"B-cancer": 3,
|
26 |
"B-chronic_disease": 5,
|
27 |
+
"B-treatment": 7,
|
28 |
+
"I-allergy_name": 2,
|
29 |
"I-cancer": 4,
|
30 |
"I-chronic_disease": 6,
|
31 |
+
"I-treatment": 8,
|
32 |
"O": 0
|
33 |
},
|
34 |
"max_position_embeddings": 512,
|
tf_model.h5
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 265606416
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:dd25dec4a93ec2018506026a3d03a60ce7dd8fda684f5cd92479ffb4d9a6a9be
|
3 |
size 265606416
|