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
|