End of training
Browse files- README.md +18 -12
- config.json +18 -14
- pytorch_model.bin +2 -2
- training_args.bin +1 -1
README.md
CHANGED
@@ -17,8 +17,8 @@ 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: 2.
|
21 |
-
- Accuracy: 0.
|
22 |
|
23 |
## Model description
|
24 |
|
@@ -43,21 +43,27 @@ The following hyperparameters were used during training:
|
|
43 |
- seed: 42
|
44 |
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
45 |
- lr_scheduler_type: linear
|
46 |
-
- num_epochs:
|
47 |
|
48 |
### Training results
|
49 |
|
50 |
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|
51 |
|:-------------:|:-----:|:----:|:---------------:|:--------:|
|
52 |
-
| 2.
|
53 |
-
| 2.
|
54 |
-
| 2.
|
55 |
-
| 2.
|
56 |
-
| 2.
|
57 |
-
| 2.
|
58 |
-
| 2.
|
59 |
-
| 2.
|
60 |
-
| 2.
|
|
|
|
|
|
|
|
|
|
|
|
|
61 |
|
62 |
|
63 |
### Framework versions
|
|
|
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: 2.7752
|
21 |
+
- Accuracy: 0.1429
|
22 |
|
23 |
## Model description
|
24 |
|
|
|
43 |
- seed: 42
|
44 |
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
45 |
- lr_scheduler_type: linear
|
46 |
+
- num_epochs: 3.0
|
47 |
|
48 |
### Training results
|
49 |
|
50 |
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|
51 |
|:-------------:|:-----:|:----:|:---------------:|:--------:|
|
52 |
+
| 2.8784 | 0.2 | 60 | 2.8602 | 0.0807 |
|
53 |
+
| 2.8535 | 0.4 | 120 | 2.8465 | 0.0971 |
|
54 |
+
| 2.8292 | 0.6 | 180 | 2.8197 | 0.1004 |
|
55 |
+
| 2.7971 | 0.8 | 240 | 2.8041 | 0.1037 |
|
56 |
+
| 2.7996 | 1.0 | 300 | 2.7861 | 0.1227 |
|
57 |
+
| 2.668 | 1.2 | 360 | 2.7832 | 0.1288 |
|
58 |
+
| 2.6317 | 1.4 | 420 | 2.7636 | 0.1309 |
|
59 |
+
| 2.6236 | 1.59 | 480 | 2.7676 | 0.1315 |
|
60 |
+
| 2.571 | 1.79 | 540 | 2.7519 | 0.1326 |
|
61 |
+
| 2.5832 | 1.99 | 600 | 2.7572 | 0.1358 |
|
62 |
+
| 2.3685 | 2.19 | 660 | 2.7616 | 0.1484 |
|
63 |
+
| 2.376 | 2.39 | 720 | 2.7727 | 0.1429 |
|
64 |
+
| 2.3499 | 2.59 | 780 | 2.7808 | 0.1391 |
|
65 |
+
| 2.3561 | 2.79 | 840 | 2.7736 | 0.1451 |
|
66 |
+
| 2.3265 | 2.99 | 900 | 2.7752 | 0.1429 |
|
67 |
|
68 |
|
69 |
### Framework versions
|
config.json
CHANGED
@@ -18,13 +18,15 @@
|
|
18 |
"6": "Jim",
|
19 |
"7": "Kelly",
|
20 |
"8": "Kevin",
|
21 |
-
"9": "
|
22 |
-
"10": "
|
23 |
-
"11": "
|
24 |
-
"12": "
|
25 |
-
"13": "
|
26 |
-
"14": "
|
27 |
-
"15": "
|
|
|
|
|
28 |
},
|
29 |
"initializer_range": 0.02,
|
30 |
"label2id": {
|
@@ -37,13 +39,15 @@
|
|
37 |
"Jim": 6,
|
38 |
"Kelly": 7,
|
39 |
"Kevin": 8,
|
40 |
-
"
|
41 |
-
"
|
42 |
-
"
|
43 |
-
"
|
44 |
-
"
|
45 |
-
"
|
46 |
-
"
|
|
|
|
|
47 |
},
|
48 |
"max_position_embeddings": 512,
|
49 |
"model_type": "distilbert",
|
|
|
18 |
"6": "Jim",
|
19 |
"7": "Kelly",
|
20 |
"8": "Kevin",
|
21 |
+
"9": "Meredith",
|
22 |
+
"10": "Michael",
|
23 |
+
"11": "Nellie",
|
24 |
+
"12": "Oscar",
|
25 |
+
"13": "Pam",
|
26 |
+
"14": "Phyllis",
|
27 |
+
"15": "Ryan",
|
28 |
+
"16": "Stanley",
|
29 |
+
"17": "Toby"
|
30 |
},
|
31 |
"initializer_range": 0.02,
|
32 |
"label2id": {
|
|
|
39 |
"Jim": 6,
|
40 |
"Kelly": 7,
|
41 |
"Kevin": 8,
|
42 |
+
"Meredith": 9,
|
43 |
+
"Michael": 10,
|
44 |
+
"Nellie": 11,
|
45 |
+
"Oscar": 12,
|
46 |
+
"Pam": 13,
|
47 |
+
"Phyllis": 14,
|
48 |
+
"Ryan": 15,
|
49 |
+
"Stanley": 16,
|
50 |
+
"Toby": 17
|
51 |
},
|
52 |
"max_position_embeddings": 512,
|
53 |
"model_type": "distilbert",
|
pytorch_model.bin
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:13dd0404e778e890786501e168831d2dc6e6840428f0a5c0476c072091688e06
|
3 |
+
size 267903341
|
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:da0bb00b17e21382cb6997166b8df1e55bb1422ead0ef6d287d5ff30554e69bd
|
3 |
size 4027
|