kearney commited on
Commit
81b36e0
·
1 Parent(s): 0837888

End of training

Browse files
Files changed (6) hide show
  1. README.md +9 -9
  2. added_tokens.json +3 -0
  3. config.json +23 -29
  4. model.safetensors +2 -2
  5. tokenizer.json +9 -0
  6. 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.6870
22
- - Accuracy: 0.1302
23
- - F1: 0.1144
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.771 | 0.16 | 100 | 2.7583 | 0.0859 | 0.0468 |
55
- | 2.7059 | 0.31 | 200 | 2.7404 | 0.1026 | 0.0769 |
56
- | 2.6483 | 0.47 | 300 | 2.7053 | 0.1234 | 0.1098 |
57
- | 2.5993 | 0.62 | 400 | 2.7071 | 0.1328 | 0.1118 |
58
- | 2.532 | 0.78 | 500 | 2.6908 | 0.1240 | 0.1078 |
59
- | 2.5205 | 0.94 | 600 | 2.6870 | 0.1302 | 0.1144 |
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": "Darryl",
15
- "3": "Dwight",
16
- "4": "Erin",
17
- "5": "Jan",
18
- "6": "Jim",
19
- "7": "Kelly",
20
- "8": "Kevin",
21
- "9": "Michael",
22
- "10": "Oscar",
23
- "11": "Pam",
24
- "12": "Phyllis",
25
- "13": "Ryan",
26
- "14": "Stanley",
27
- "15": "Toby"
28
  },
29
  "initializer_range": 0.02,
30
  "label2id": {
31
  "Andy": 0,
32
  "Angela": 1,
33
- "Darryl": 2,
34
- "Dwight": 3,
35
- "Erin": 4,
36
- "Jan": 5,
37
- "Jim": 6,
38
- "Kelly": 7,
39
- "Kevin": 8,
40
- "Michael": 9,
41
- "Oscar": 10,
42
- "Pam": 11,
43
- "Phyllis": 12,
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": 30522
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:ce02e519f88ed18b0cdbf6fbed4697b4fc35b405cf7e46a2fd08391365aad9ed
3
- size 267875632
 
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:2abb06c6cac7cb77bced319ae3c3006924885496c317cd2ad2c9281488f6a6a8
3
  size 4027
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9df876036232d55ce1a32d427f9a437a6a940d2c72a03e2a0bb74b206e76c4e1
3
  size 4027