SharonTudi commited on
Commit
709a715
·
verified ·
1 Parent(s): a578ed8

End of training

Browse files
README.md CHANGED
@@ -1,6 +1,6 @@
1
  ---
2
  license: apache-2.0
3
- base_model: distilbert-base-cased
4
  tags:
5
  - generated_from_trainer
6
  metrics:
@@ -18,13 +18,13 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  # DIALOGUE_one
20
 
21
- This model is a fine-tuned version of [distilbert-base-cased](https://huggingface.co/distilbert-base-cased) on the None dataset.
22
  It achieves the following results on the evaluation set:
23
- - Loss: 1.1527
24
- - Precision: 0.8207
25
- - Recall: 0.8158
26
- - F1: 0.8103
27
- - Accuracy: 0.8158
28
 
29
  ## Model description
30
 
@@ -55,54 +55,54 @@ The following hyperparameters were used during training:
55
 
56
  | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
57
  |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
58
- | 1.3373 | 0.62 | 30 | 1.2236 | 0.6642 | 0.5263 | 0.4857 | 0.5263 |
59
- | 0.9036 | 1.25 | 60 | 0.8198 | 0.7325 | 0.7237 | 0.7209 | 0.7237 |
60
- | 0.5825 | 1.88 | 90 | 0.6795 | 0.7291 | 0.7237 | 0.7257 | 0.7237 |
61
- | 0.4134 | 2.5 | 120 | 0.5864 | 0.8026 | 0.8026 | 0.8017 | 0.8026 |
62
- | 0.2866 | 3.12 | 150 | 0.6642 | 0.7360 | 0.7368 | 0.7352 | 0.7368 |
63
- | 0.2291 | 3.75 | 180 | 0.7632 | 0.7673 | 0.7632 | 0.7646 | 0.7632 |
64
- | 0.2907 | 4.38 | 210 | 0.7228 | 0.8109 | 0.7632 | 0.7607 | 0.7632 |
65
- | 0.1721 | 5.0 | 240 | 0.7910 | 0.7793 | 0.7763 | 0.7765 | 0.7763 |
66
- | 0.1475 | 5.62 | 270 | 0.7622 | 0.8173 | 0.8158 | 0.8156 | 0.8158 |
67
- | 0.1541 | 6.25 | 300 | 0.6803 | 0.7911 | 0.7895 | 0.7896 | 0.7895 |
68
- | 0.0706 | 6.88 | 330 | 0.8729 | 0.8207 | 0.8158 | 0.8103 | 0.8158 |
69
- | 0.0825 | 7.5 | 360 | 0.8173 | 0.8472 | 0.8421 | 0.8408 | 0.8421 |
70
- | 0.0419 | 8.12 | 390 | 0.6985 | 0.8356 | 0.8289 | 0.8294 | 0.8289 |
71
- | 0.0588 | 8.75 | 420 | 0.7242 | 0.8137 | 0.8026 | 0.8041 | 0.8026 |
72
- | 0.0535 | 9.38 | 450 | 0.9187 | 0.8060 | 0.8026 | 0.8007 | 0.8026 |
73
- | 0.0275 | 10.0 | 480 | 1.3225 | 0.7839 | 0.7763 | 0.7615 | 0.7763 |
74
- | 0.01 | 10.62 | 510 | 1.2285 | 0.7667 | 0.7632 | 0.7533 | 0.7632 |
75
- | 0.0336 | 11.25 | 540 | 1.1652 | 0.8170 | 0.8026 | 0.7976 | 0.8026 |
76
- | 0.0031 | 11.88 | 570 | 0.8795 | 0.7941 | 0.7895 | 0.7854 | 0.7895 |
77
- | 0.0019 | 12.5 | 600 | 1.0385 | 0.8076 | 0.8026 | 0.7971 | 0.8026 |
78
- | 0.0035 | 13.12 | 630 | 1.1230 | 0.8207 | 0.8158 | 0.8103 | 0.8158 |
79
- | 0.0016 | 13.75 | 660 | 1.0793 | 0.8207 | 0.8158 | 0.8103 | 0.8158 |
80
- | 0.0014 | 14.38 | 690 | 1.0877 | 0.8207 | 0.8158 | 0.8103 | 0.8158 |
81
- | 0.0015 | 15.0 | 720 | 1.1482 | 0.8207 | 0.8158 | 0.8103 | 0.8158 |
82
- | 0.0012 | 15.62 | 750 | 1.1424 | 0.7935 | 0.7895 | 0.7834 | 0.7895 |
83
- | 0.0011 | 16.25 | 780 | 1.0405 | 0.8207 | 0.8158 | 0.8103 | 0.8158 |
84
- | 0.0011 | 16.88 | 810 | 1.4101 | 0.7874 | 0.7763 | 0.7630 | 0.7763 |
85
- | 0.0012 | 17.5 | 840 | 1.2321 | 0.8207 | 0.8158 | 0.8103 | 0.8158 |
86
- | 0.001 | 18.12 | 870 | 1.1961 | 0.8207 | 0.8158 | 0.8103 | 0.8158 |
87
- | 0.001 | 18.75 | 900 | 1.1638 | 0.8207 | 0.8158 | 0.8103 | 0.8158 |
88
- | 0.0009 | 19.38 | 930 | 1.1429 | 0.8207 | 0.8158 | 0.8103 | 0.8158 |
89
- | 0.0009 | 20.0 | 960 | 1.3692 | 0.7839 | 0.7763 | 0.7615 | 0.7763 |
90
- | 0.0009 | 20.62 | 990 | 1.3196 | 0.7839 | 0.7763 | 0.7615 | 0.7763 |
91
- | 0.0008 | 21.25 | 1020 | 1.1942 | 0.7839 | 0.7763 | 0.7615 | 0.7763 |
92
- | 0.0008 | 21.88 | 1050 | 1.1439 | 0.8207 | 0.8158 | 0.8103 | 0.8158 |
93
- | 0.0008 | 22.5 | 1080 | 1.1689 | 0.8207 | 0.8158 | 0.8103 | 0.8158 |
94
- | 0.0008 | 23.12 | 1110 | 1.1553 | 0.8207 | 0.8158 | 0.8103 | 0.8158 |
95
- | 0.0007 | 23.75 | 1140 | 1.1560 | 0.8207 | 0.8158 | 0.8103 | 0.8158 |
96
- | 0.0007 | 24.38 | 1170 | 1.1451 | 0.8207 | 0.8158 | 0.8103 | 0.8158 |
97
- | 0.0007 | 25.0 | 1200 | 1.1473 | 0.8207 | 0.8158 | 0.8103 | 0.8158 |
98
- | 0.0007 | 25.62 | 1230 | 1.1412 | 0.8207 | 0.8158 | 0.8103 | 0.8158 |
99
- | 0.0007 | 26.25 | 1260 | 1.1655 | 0.8207 | 0.8158 | 0.8103 | 0.8158 |
100
- | 0.0007 | 26.88 | 1290 | 1.1649 | 0.8207 | 0.8158 | 0.8103 | 0.8158 |
101
- | 0.0006 | 27.5 | 1320 | 1.1556 | 0.8207 | 0.8158 | 0.8103 | 0.8158 |
102
- | 0.0006 | 28.12 | 1350 | 1.1521 | 0.8207 | 0.8158 | 0.8103 | 0.8158 |
103
- | 0.0006 | 28.75 | 1380 | 1.1481 | 0.8207 | 0.8158 | 0.8103 | 0.8158 |
104
- | 0.0008 | 29.38 | 1410 | 1.1511 | 0.8207 | 0.8158 | 0.8103 | 0.8158 |
105
- | 0.0006 | 30.0 | 1440 | 1.1527 | 0.8207 | 0.8158 | 0.8103 | 0.8158 |
106
 
107
 
108
  ### Framework versions
 
1
  ---
2
  license: apache-2.0
3
+ base_model: bert-base-cased
4
  tags:
5
  - generated_from_trainer
6
  metrics:
 
18
 
19
  # DIALOGUE_one
20
 
21
+ This model is a fine-tuned version of [bert-base-cased](https://huggingface.co/bert-base-cased) on the None dataset.
22
  It achieves the following results on the evaluation set:
23
+ - Loss: 1.2331
24
+ - Precision: 0.7635
25
+ - Recall: 0.7632
26
+ - F1: 0.7625
27
+ - Accuracy: 0.7632
28
 
29
  ## Model description
30
 
 
55
 
56
  | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
57
  |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
58
+ | 1.229 | 0.62 | 30 | 1.0350 | 0.5533 | 0.5526 | 0.4885 | 0.5526 |
59
+ | 0.865 | 1.25 | 60 | 0.7887 | 0.8236 | 0.7763 | 0.7723 | 0.7763 |
60
+ | 0.6577 | 1.88 | 90 | 0.6786 | 0.8159 | 0.7632 | 0.7509 | 0.7632 |
61
+ | 0.4934 | 2.5 | 120 | 0.5809 | 0.8732 | 0.8289 | 0.8263 | 0.8289 |
62
+ | 0.4062 | 3.12 | 150 | 0.5829 | 0.7959 | 0.7895 | 0.7904 | 0.7895 |
63
+ | 0.2879 | 3.75 | 180 | 0.6573 | 0.7914 | 0.7763 | 0.7767 | 0.7763 |
64
+ | 0.3208 | 4.38 | 210 | 0.6510 | 0.7684 | 0.7632 | 0.7632 | 0.7632 |
65
+ | 0.2356 | 5.0 | 240 | 0.7371 | 0.8324 | 0.8026 | 0.8035 | 0.8026 |
66
+ | 0.1637 | 5.62 | 270 | 0.6212 | 0.8561 | 0.8553 | 0.8548 | 0.8553 |
67
+ | 0.1117 | 6.25 | 300 | 0.6782 | 0.8318 | 0.8289 | 0.8295 | 0.8289 |
68
+ | 0.0484 | 6.88 | 330 | 0.8347 | 0.8217 | 0.8158 | 0.8159 | 0.8158 |
69
+ | 0.0563 | 7.5 | 360 | 0.8745 | 0.8017 | 0.7895 | 0.7922 | 0.7895 |
70
+ | 0.0385 | 8.12 | 390 | 0.7141 | 0.8390 | 0.8158 | 0.8173 | 0.8158 |
71
+ | 0.0418 | 8.75 | 420 | 1.0020 | 0.8341 | 0.8289 | 0.8293 | 0.8289 |
72
+ | 0.0423 | 9.38 | 450 | 0.8778 | 0.7639 | 0.7632 | 0.7630 | 0.7632 |
73
+ | 0.0193 | 10.0 | 480 | 0.9370 | 0.8255 | 0.8026 | 0.8048 | 0.8026 |
74
+ | 0.0131 | 10.62 | 510 | 1.0443 | 0.8037 | 0.7895 | 0.7908 | 0.7895 |
75
+ | 0.0143 | 11.25 | 540 | 0.9799 | 0.8390 | 0.8158 | 0.8173 | 0.8158 |
76
+ | 0.0111 | 11.88 | 570 | 1.1469 | 0.7656 | 0.7632 | 0.7626 | 0.7632 |
77
+ | 0.0033 | 12.5 | 600 | 0.9717 | 0.7477 | 0.75 | 0.7486 | 0.75 |
78
+ | 0.0016 | 13.12 | 630 | 1.0248 | 0.7809 | 0.7763 | 0.7773 | 0.7763 |
79
+ | 0.0028 | 13.75 | 660 | 1.4541 | 0.7514 | 0.75 | 0.7487 | 0.75 |
80
+ | 0.0045 | 14.38 | 690 | 1.2292 | 0.7656 | 0.7632 | 0.7626 | 0.7632 |
81
+ | 0.0013 | 15.0 | 720 | 1.2528 | 0.7484 | 0.75 | 0.7484 | 0.75 |
82
+ | 0.0011 | 15.62 | 750 | 1.2332 | 0.7484 | 0.75 | 0.7484 | 0.75 |
83
+ | 0.0011 | 16.25 | 780 | 1.2946 | 0.7484 | 0.75 | 0.7484 | 0.75 |
84
+ | 0.001 | 16.88 | 810 | 1.2817 | 0.7484 | 0.75 | 0.7484 | 0.75 |
85
+ | 0.001 | 17.5 | 840 | 1.2400 | 0.7484 | 0.75 | 0.7484 | 0.75 |
86
+ | 0.0011 | 18.12 | 870 | 1.5933 | 0.7385 | 0.7368 | 0.7323 | 0.7368 |
87
+ | 0.0009 | 18.75 | 900 | 1.4288 | 0.7385 | 0.7368 | 0.7323 | 0.7368 |
88
+ | 0.0008 | 19.38 | 930 | 1.2843 | 0.7656 | 0.7632 | 0.7626 | 0.7632 |
89
+ | 0.0008 | 20.0 | 960 | 1.2699 | 0.7656 | 0.7632 | 0.7626 | 0.7632 |
90
+ | 0.0008 | 20.62 | 990 | 1.2718 | 0.7656 | 0.7632 | 0.7626 | 0.7632 |
91
+ | 0.0008 | 21.25 | 1020 | 1.2456 | 0.7656 | 0.7632 | 0.7626 | 0.7632 |
92
+ | 0.0007 | 21.88 | 1050 | 1.2366 | 0.7484 | 0.75 | 0.7484 | 0.75 |
93
+ | 0.0007 | 22.5 | 1080 | 1.2430 | 0.7484 | 0.75 | 0.7484 | 0.75 |
94
+ | 0.0007 | 23.12 | 1110 | 1.2353 | 0.7484 | 0.75 | 0.7484 | 0.75 |
95
+ | 0.0007 | 23.75 | 1140 | 1.2345 | 0.7484 | 0.75 | 0.7484 | 0.75 |
96
+ | 0.0007 | 24.38 | 1170 | 1.2325 | 0.7635 | 0.7632 | 0.7625 | 0.7632 |
97
+ | 0.0007 | 25.0 | 1200 | 1.2283 | 0.7484 | 0.75 | 0.7484 | 0.75 |
98
+ | 0.0007 | 25.62 | 1230 | 1.2236 | 0.7635 | 0.7632 | 0.7625 | 0.7632 |
99
+ | 0.0006 | 26.25 | 1260 | 1.2247 | 0.7635 | 0.7632 | 0.7625 | 0.7632 |
100
+ | 0.0006 | 26.88 | 1290 | 1.2263 | 0.7635 | 0.7632 | 0.7625 | 0.7632 |
101
+ | 0.0006 | 27.5 | 1320 | 1.2286 | 0.7635 | 0.7632 | 0.7625 | 0.7632 |
102
+ | 0.0006 | 28.12 | 1350 | 1.2307 | 0.7635 | 0.7632 | 0.7625 | 0.7632 |
103
+ | 0.0006 | 28.75 | 1380 | 1.2322 | 0.7635 | 0.7632 | 0.7625 | 0.7632 |
104
+ | 0.0006 | 29.38 | 1410 | 1.2330 | 0.7635 | 0.7632 | 0.7625 | 0.7632 |
105
+ | 0.0006 | 30.0 | 1440 | 1.2331 | 0.7635 | 0.7632 | 0.7625 | 0.7632 |
106
 
107
 
108
  ### Framework versions
config.json CHANGED
@@ -1,13 +1,14 @@
1
  {
2
- "_name_or_path": "distilbert-base-cased",
3
- "activation": "gelu",
4
  "architectures": [
5
- "DistilBertForSequenceClassification"
6
  ],
7
- "attention_dropout": 0.1,
8
- "dim": 768,
9
- "dropout": 0.1,
10
- "hidden_dim": 3072,
 
 
11
  "id2label": {
12
  "0": "Hospital-Inform",
13
  "1": "Hospital-Request",
@@ -15,24 +16,24 @@
15
  "3": "general-thank"
16
  },
17
  "initializer_range": 0.02,
 
18
  "label2id": {
19
  "LABEL_0": 0,
20
  "LABEL_1": 1,
21
  "LABEL_2": 2,
22
  "LABEL_3": 3
23
  },
 
24
  "max_position_embeddings": 512,
25
- "model_type": "distilbert",
26
- "n_heads": 12,
27
- "n_layers": 6,
28
- "output_past": true,
29
  "pad_token_id": 0,
 
30
  "problem_type": "single_label_classification",
31
- "qa_dropout": 0.1,
32
- "seq_classif_dropout": 0.2,
33
- "sinusoidal_pos_embds": false,
34
- "tie_weights_": true,
35
  "torch_dtype": "float32",
36
  "transformers_version": "4.36.2",
 
 
37
  "vocab_size": 28996
38
  }
 
1
  {
2
+ "_name_or_path": "bert-base-cased",
 
3
  "architectures": [
4
+ "BertForSequenceClassification"
5
  ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "classifier_dropout": null,
8
+ "gradient_checkpointing": false,
9
+ "hidden_act": "gelu",
10
+ "hidden_dropout_prob": 0.1,
11
+ "hidden_size": 768,
12
  "id2label": {
13
  "0": "Hospital-Inform",
14
  "1": "Hospital-Request",
 
16
  "3": "general-thank"
17
  },
18
  "initializer_range": 0.02,
19
+ "intermediate_size": 3072,
20
  "label2id": {
21
  "LABEL_0": 0,
22
  "LABEL_1": 1,
23
  "LABEL_2": 2,
24
  "LABEL_3": 3
25
  },
26
+ "layer_norm_eps": 1e-12,
27
  "max_position_embeddings": 512,
28
+ "model_type": "bert",
29
+ "num_attention_heads": 12,
30
+ "num_hidden_layers": 12,
 
31
  "pad_token_id": 0,
32
+ "position_embedding_type": "absolute",
33
  "problem_type": "single_label_classification",
 
 
 
 
34
  "torch_dtype": "float32",
35
  "transformers_version": "4.36.2",
36
+ "type_vocab_size": 2,
37
+ "use_cache": true,
38
  "vocab_size": 28996
39
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:09c7427ea012b025bda2f34f64ed40ef4215eb1e9c2117052ed746b6b82c46f6
3
- size 263150840
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f289213bdf7cd4d47ef304341a3eedbae9d979a01cfb3a7f8903dd29ded6293a
3
+ size 433276920
runs/Jan22_10-45-09_e9a77ba5e600/events.out.tfevents.1705920321.e9a77ba5e600.245.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:241690c246e400054e33dac8d6feb885d61c89f060becaab3cdbcc40b4ce39fe
3
+ size 35043
tokenizer_config.json CHANGED
@@ -52,6 +52,6 @@
52
  "sep_token": "[SEP]",
53
  "strip_accents": null,
54
  "tokenize_chinese_chars": true,
55
- "tokenizer_class": "DistilBertTokenizer",
56
  "unk_token": "[UNK]"
57
  }
 
52
  "sep_token": "[SEP]",
53
  "strip_accents": null,
54
  "tokenize_chinese_chars": true,
55
+ "tokenizer_class": "BertTokenizer",
56
  "unk_token": "[UNK]"
57
  }
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:404bfff7d8d97d27a0d247d82159e9cba93f34a3803eae9c6f494c91a96b2cff
3
  size 4664
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:88e2f7149f0d113e95e3b4cb9549c76b64e7d131377e2c6236f5b6794ea70371
3
  size 4664