End of training
Browse files- README.md +9 -9
- added_tokens.json +3 -0
- config.json +23 -29
- model.safetensors +2 -2
- tokenizer.json +9 -0
- training_args.bin +1 -1
README.md
CHANGED
@@ -18,9 +18,9 @@ should probably proofread and complete it, then remove this comment. -->
|
|
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: 2.
|
22 |
-
- Accuracy: 0.
|
23 |
-
- F1: 0.
|
24 |
|
25 |
## Model description
|
26 |
|
@@ -51,12 +51,12 @@ The following hyperparameters were used during training:
|
|
51 |
|
52 |
| Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
|
53 |
|:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
|
54 |
-
| 2.
|
55 |
-
| 2.
|
56 |
-
| 2.
|
57 |
-
| 2.
|
58 |
-
| 2.
|
59 |
-
| 2.
|
60 |
|
61 |
|
62 |
### Framework versions
|
|
|
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: 2.4938
|
22 |
+
- Accuracy: 0.1487
|
23 |
+
- F1: 0.1387
|
24 |
|
25 |
## Model description
|
26 |
|
|
|
51 |
|
52 |
| Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
|
53 |
|:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
|
54 |
+
| 2.557 | 0.15 | 100 | 2.5581 | 0.1078 | 0.0639 |
|
55 |
+
| 2.5185 | 0.31 | 200 | 2.5136 | 0.1223 | 0.0990 |
|
56 |
+
| 2.4336 | 0.46 | 300 | 2.4953 | 0.1394 | 0.1163 |
|
57 |
+
| 2.3705 | 0.62 | 400 | 2.4926 | 0.1373 | 0.1292 |
|
58 |
+
| 2.3217 | 0.77 | 500 | 2.4875 | 0.1539 | 0.1496 |
|
59 |
+
| 2.2475 | 0.92 | 600 | 2.4938 | 0.1487 | 0.1387 |
|
60 |
|
61 |
|
62 |
### Framework versions
|
added_tokens.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"<CHR>": 30522
|
3 |
+
}
|
config.json
CHANGED
@@ -11,39 +11,33 @@
|
|
11 |
"id2label": {
|
12 |
"0": "Andy",
|
13 |
"1": "Angela",
|
14 |
-
"2": "
|
15 |
-
"3": "
|
16 |
-
"4": "
|
17 |
-
"5": "
|
18 |
-
"6": "
|
19 |
-
"7": "
|
20 |
-
"8": "
|
21 |
-
"9": "
|
22 |
-
"10": "
|
23 |
-
"11": "
|
24 |
-
"12": "
|
25 |
-
"13": "Ryan",
|
26 |
-
"14": "Stanley",
|
27 |
-
"15": "Toby"
|
28 |
},
|
29 |
"initializer_range": 0.02,
|
30 |
"label2id": {
|
31 |
"Andy": 0,
|
32 |
"Angela": 1,
|
33 |
-
"
|
34 |
-
"
|
35 |
-
"
|
36 |
-
"
|
37 |
-
"
|
38 |
-
"
|
39 |
-
"
|
40 |
-
"
|
41 |
-
"
|
42 |
-
"
|
43 |
-
"
|
44 |
-
"Ryan": 13,
|
45 |
-
"Stanley": 14,
|
46 |
-
"Toby": 15
|
47 |
},
|
48 |
"max_position_embeddings": 512,
|
49 |
"model_type": "distilbert",
|
@@ -57,5 +51,5 @@
|
|
57 |
"tie_weights_": true,
|
58 |
"torch_dtype": "float32",
|
59 |
"transformers_version": "4.32.1",
|
60 |
-
"vocab_size":
|
61 |
}
|
|
|
11 |
"id2label": {
|
12 |
"0": "Andy",
|
13 |
"1": "Angela",
|
14 |
+
"2": "Creed",
|
15 |
+
"3": "Darryl",
|
16 |
+
"4": "Dwight",
|
17 |
+
"5": "Jim",
|
18 |
+
"6": "Kevin",
|
19 |
+
"7": "Meredith",
|
20 |
+
"8": "Michael",
|
21 |
+
"9": "Oscar",
|
22 |
+
"10": "Pam",
|
23 |
+
"11": "Phyllis",
|
24 |
+
"12": "Stanley"
|
|
|
|
|
|
|
25 |
},
|
26 |
"initializer_range": 0.02,
|
27 |
"label2id": {
|
28 |
"Andy": 0,
|
29 |
"Angela": 1,
|
30 |
+
"Creed": 2,
|
31 |
+
"Darryl": 3,
|
32 |
+
"Dwight": 4,
|
33 |
+
"Jim": 5,
|
34 |
+
"Kevin": 6,
|
35 |
+
"Meredith": 7,
|
36 |
+
"Michael": 8,
|
37 |
+
"Oscar": 9,
|
38 |
+
"Pam": 10,
|
39 |
+
"Phyllis": 11,
|
40 |
+
"Stanley": 12
|
|
|
|
|
|
|
41 |
},
|
42 |
"max_position_embeddings": 512,
|
43 |
"model_type": "distilbert",
|
|
|
51 |
"tie_weights_": true,
|
52 |
"torch_dtype": "float32",
|
53 |
"transformers_version": "4.32.1",
|
54 |
+
"vocab_size": 30523
|
55 |
}
|
model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:20bc5dca3f8bab836af700616ef335d63e9401173360dd470c8acebed8146c48
|
3 |
+
size 267869476
|
tokenizer.json
CHANGED
@@ -52,6 +52,15 @@
|
|
52 |
"rstrip": false,
|
53 |
"normalized": false,
|
54 |
"special": true
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
55 |
}
|
56 |
],
|
57 |
"normalizer": {
|
|
|
52 |
"rstrip": false,
|
53 |
"normalized": false,
|
54 |
"special": true
|
55 |
+
},
|
56 |
+
{
|
57 |
+
"id": 30522,
|
58 |
+
"content": "<CHR>",
|
59 |
+
"single_word": false,
|
60 |
+
"lstrip": false,
|
61 |
+
"rstrip": false,
|
62 |
+
"normalized": true,
|
63 |
+
"special": false
|
64 |
}
|
65 |
],
|
66 |
"normalizer": {
|
training_args.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 4027
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9df876036232d55ce1a32d427f9a437a6a940d2c72a03e2a0bb74b206e76c4e1
|
3 |
size 4027
|