Hugging Face's logo

andreaceto
/
schedulebot-nlu-engine

Transformers
TensorBoard
Safetensors
English
distilbert
Generated from Trainer
Model card Files
xet
Metrics Community
andreaceto commited on
Commit
fc035e6
·
verified ·
1 Parent(s): 9cd6a68

End of training

Browse files
Files changed (5) hide show
  1. README.md +53 -53
  2. config.json +2 -2
  3. model.safetensors +1 -1
  4. runs/Jul18_11-24-49_83c12e474755/events.out.tfevents.1752837890.83c12e474755.863.1 +0 -0
  5. training_args.bin +0 -0
README.md CHANGED
@@ -16,10 +16,10 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 0.3743
20
  - Intent Accuracy: 0.9132
21
- - Intent F1: 0.9132
22
- - Ner F1: 0.9161
23
 
24
  ## Model description
25
 
@@ -50,56 +50,56 @@ The following hyperparameters were used during training:
50
 
51
  | Training Loss | Epoch | Step | Validation Loss | Intent Accuracy | Intent F1 | Ner F1 |
52
  |:-------------:|:-----:|:----:|:---------------:|:---------------:|:---------:|:------:|
53
- | 1.0569 | 1.0 | 64 | 0.7285 | 0.8082 | 0.8086 | 0.9076 |
54
- | 1.0159 | 2.0 | 128 | 0.6989 | 0.8128 | 0.8145 | 0.9065 |
55
- | 0.9781 | 3.0 | 192 | 0.6679 | 0.8196 | 0.8185 | 0.9044 |
56
- | 0.9258 | 4.0 | 256 | 0.6406 | 0.8219 | 0.8206 | 0.9054 |
57
- | 0.879 | 5.0 | 320 | 0.6164 | 0.8379 | 0.8390 | 0.9060 |
58
- | 0.8516 | 6.0 | 384 | 0.5951 | 0.8425 | 0.8431 | 0.9066 |
59
- | 0.8276 | 7.0 | 448 | 0.5706 | 0.8470 | 0.8468 | 0.9090 |
60
- | 0.7897 | 8.0 | 512 | 0.5699 | 0.8379 | 0.8390 | 0.9102 |
61
- | 0.7477 | 9.0 | 576 | 0.5442 | 0.8539 | 0.8537 | 0.9099 |
62
- | 0.7443 | 10.0 | 640 | 0.5300 | 0.8562 | 0.8563 | 0.9069 |
63
- | 0.709 | 11.0 | 704 | 0.5254 | 0.8584 | 0.8583 | 0.9099 |
64
- | 0.6971 | 12.0 | 768 | 0.5185 | 0.8607 | 0.8605 | 0.9123 |
65
- | 0.6667 | 13.0 | 832 | 0.4957 | 0.8653 | 0.8650 | 0.9117 |
66
- | 0.6643 | 14.0 | 896 | 0.4860 | 0.8813 | 0.8808 | 0.9083 |
67
- | 0.6274 | 15.0 | 960 | 0.4743 | 0.8813 | 0.8810 | 0.9076 |
68
- | 0.6398 | 16.0 | 1024 | 0.4680 | 0.8790 | 0.8791 | 0.9093 |
69
- | 0.615 | 17.0 | 1088 | 0.4644 | 0.8813 | 0.8810 | 0.9142 |
70
- | 0.6102 | 18.0 | 1152 | 0.4575 | 0.8858 | 0.8859 | 0.9100 |
71
- | 0.5579 | 19.0 | 1216 | 0.4548 | 0.8813 | 0.8813 | 0.9103 |
72
- | 0.5779 | 20.0 | 1280 | 0.4436 | 0.8881 | 0.8881 | 0.9092 |
73
- | 0.5659 | 21.0 | 1344 | 0.4395 | 0.8927 | 0.8925 | 0.9120 |
74
- | 0.5458 | 22.0 | 1408 | 0.4295 | 0.8927 | 0.8930 | 0.9126 |
75
- | 0.5375 | 23.0 | 1472 | 0.4294 | 0.8973 | 0.8974 | 0.9134 |
76
- | 0.5259 | 24.0 | 1536 | 0.4225 | 0.8973 | 0.8972 | 0.9150 |
77
- | 0.5064 | 25.0 | 1600 | 0.4223 | 0.8927 | 0.8930 | 0.9138 |
78
- | 0.5109 | 26.0 | 1664 | 0.4118 | 0.8973 | 0.8971 | 0.9133 |
79
- | 0.5001 | 27.0 | 1728 | 0.4161 | 0.8904 | 0.8902 | 0.9134 |
80
- | 0.5052 | 28.0 | 1792 | 0.4085 | 0.8995 | 0.8994 | 0.9138 |
81
- | 0.4863 | 29.0 | 1856 | 0.4052 | 0.9041 | 0.9041 | 0.9132 |
82
- | 0.4842 | 30.0 | 1920 | 0.4123 | 0.8927 | 0.8927 | 0.9132 |
83
- | 0.474 | 31.0 | 1984 | 0.3944 | 0.9018 | 0.9017 | 0.9138 |
84
- | 0.4638 | 32.0 | 2048 | 0.3946 | 0.9041 | 0.9041 | 0.9148 |
85
- | 0.4678 | 33.0 | 2112 | 0.3989 | 0.8904 | 0.8903 | 0.9133 |
86
- | 0.4641 | 34.0 | 2176 | 0.3920 | 0.9041 | 0.9040 | 0.9150 |
87
- | 0.4632 | 35.0 | 2240 | 0.3884 | 0.9064 | 0.9063 | 0.9145 |
88
- | 0.4825 | 36.0 | 2304 | 0.3882 | 0.9064 | 0.9065 | 0.9138 |
89
- | 0.4403 | 37.0 | 2368 | 0.3854 | 0.8995 | 0.8998 | 0.9134 |
90
- | 0.459 | 38.0 | 2432 | 0.3833 | 0.9041 | 0.9041 | 0.9145 |
91
- | 0.4367 | 39.0 | 2496 | 0.3841 | 0.9064 | 0.9063 | 0.9179 |
92
- | 0.4388 | 40.0 | 2560 | 0.3801 | 0.9041 | 0.9040 | 0.9150 |
93
- | 0.4261 | 41.0 | 2624 | 0.3793 | 0.9087 | 0.9087 | 0.9150 |
94
- | 0.4236 | 42.0 | 2688 | 0.3775 | 0.9064 | 0.9062 | 0.9145 |
95
- | 0.4242 | 43.0 | 2752 | 0.3795 | 0.9064 | 0.9063 | 0.9156 |
96
- | 0.4364 | 44.0 | 2816 | 0.3785 | 0.9041 | 0.9040 | 0.9150 |
97
- | 0.4263 | 45.0 | 2880 | 0.3772 | 0.9041 | 0.9041 | 0.9149 |
98
- | 0.4112 | 46.0 | 2944 | 0.3769 | 0.9041 | 0.9040 | 0.9149 |
99
- | 0.428 | 47.0 | 3008 | 0.3749 | 0.9087 | 0.9085 | 0.9155 |
100
- | 0.4274 | 48.0 | 3072 | 0.3743 | 0.9064 | 0.9062 | 0.9144 |
101
- | 0.4243 | 49.0 | 3136 | 0.3748 | 0.9110 | 0.9109 | 0.9155 |
102
- | 0.4215 | 50.0 | 3200 | 0.3743 | 0.9132 | 0.9132 | 0.9161 |
103
 
104
 
105
  ### Framework versions
 
16
 
17
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 0.3685
20
  - Intent Accuracy: 0.9132
21
+ - Intent F1: 0.9128
22
+ - Ner F1: 0.9255
23
 
24
  ## Model description
25
 
 
50
 
51
  | Training Loss | Epoch | Step | Validation Loss | Intent Accuracy | Intent F1 | Ner F1 |
52
  |:-------------:|:-----:|:----:|:---------------:|:---------------:|:---------:|:------:|
53
+ | 1.0518 | 1.0 | 64 | 0.6730 | 0.8151 | 0.8147 | 0.9153 |
54
+ | 0.9965 | 2.0 | 128 | 0.6458 | 0.8219 | 0.8217 | 0.9176 |
55
+ | 0.9344 | 3.0 | 192 | 0.6251 | 0.8265 | 0.8259 | 0.9181 |
56
+ | 0.8837 | 4.0 | 256 | 0.5905 | 0.8288 | 0.8283 | 0.9192 |
57
+ | 0.8614 | 5.0 | 320 | 0.5697 | 0.8470 | 0.8464 | 0.9184 |
58
+ | 0.7958 | 6.0 | 384 | 0.5637 | 0.8333 | 0.8335 | 0.9178 |
59
+ | 0.7729 | 7.0 | 448 | 0.5383 | 0.8379 | 0.8363 | 0.9190 |
60
+ | 0.7564 | 8.0 | 512 | 0.5250 | 0.8607 | 0.8606 | 0.9195 |
61
+ | 0.7342 | 9.0 | 576 | 0.5082 | 0.8493 | 0.8488 | 0.9201 |
62
+ | 0.7095 | 10.0 | 640 | 0.5058 | 0.8562 | 0.8553 | 0.9181 |
63
+ | 0.69 | 11.0 | 704 | 0.4996 | 0.8653 | 0.8646 | 0.9181 |
64
+ | 0.6732 | 12.0 | 768 | 0.4954 | 0.8516 | 0.8501 | 0.9195 |
65
+ | 0.6615 | 13.0 | 832 | 0.4703 | 0.8562 | 0.8558 | 0.9182 |
66
+ | 0.6669 | 14.0 | 896 | 0.4638 | 0.8721 | 0.8724 | 0.9192 |
67
+ | 0.6227 | 15.0 | 960 | 0.4543 | 0.8790 | 0.8783 | 0.9204 |
68
+ | 0.6185 | 16.0 | 1024 | 0.4453 | 0.8767 | 0.8758 | 0.9217 |
69
+ | 0.5853 | 17.0 | 1088 | 0.4440 | 0.8676 | 0.8671 | 0.9222 |
70
+ | 0.5685 | 18.0 | 1152 | 0.4395 | 0.8858 | 0.8851 | 0.9233 |
71
+ | 0.5691 | 19.0 | 1216 | 0.4351 | 0.8767 | 0.8764 | 0.9217 |
72
+ | 0.5561 | 20.0 | 1280 | 0.4266 | 0.8813 | 0.8802 | 0.9215 |
73
+ | 0.5721 | 21.0 | 1344 | 0.4178 | 0.8790 | 0.8788 | 0.9209 |
74
+ | 0.5091 | 22.0 | 1408 | 0.4136 | 0.8836 | 0.8833 | 0.9209 |
75
+ | 0.5356 | 23.0 | 1472 | 0.4113 | 0.8858 | 0.8850 | 0.9211 |
76
+ | 0.526 | 24.0 | 1536 | 0.4083 | 0.8836 | 0.8826 | 0.9206 |
77
+ | 0.5132 | 25.0 | 1600 | 0.4022 | 0.8995 | 0.8997 | 0.9222 |
78
+ | 0.5017 | 26.0 | 1664 | 0.3996 | 0.8927 | 0.8925 | 0.9232 |
79
+ | 0.4926 | 27.0 | 1728 | 0.4006 | 0.8904 | 0.8905 | 0.9220 |
80
+ | 0.4878 | 28.0 | 1792 | 0.3973 | 0.9018 | 0.9018 | 0.9237 |
81
+ | 0.5028 | 29.0 | 1856 | 0.3968 | 0.8950 | 0.8944 | 0.9243 |
82
+ | 0.4519 | 30.0 | 1920 | 0.3921 | 0.8950 | 0.8946 | 0.9232 |
83
+ | 0.4718 | 31.0 | 1984 | 0.3879 | 0.9041 | 0.9039 | 0.9232 |
84
+ | 0.4551 | 32.0 | 2048 | 0.3873 | 0.8995 | 0.8991 | 0.9238 |
85
+ | 0.4696 | 33.0 | 2112 | 0.3819 | 0.9064 | 0.9064 | 0.9244 |
86
+ | 0.4655 | 34.0 | 2176 | 0.3830 | 0.9018 | 0.9011 | 0.925 |
87
+ | 0.4421 | 35.0 | 2240 | 0.3814 | 0.9018 | 0.9013 | 0.9249 |
88
+ | 0.4455 | 36.0 | 2304 | 0.3790 | 0.9041 | 0.9034 | 0.9256 |
89
+ | 0.4521 | 37.0 | 2368 | 0.3795 | 0.9064 | 0.9058 | 0.9245 |
90
+ | 0.451 | 38.0 | 2432 | 0.3772 | 0.9132 | 0.9129 | 0.9245 |
91
+ | 0.4581 | 39.0 | 2496 | 0.3781 | 0.9064 | 0.9060 | 0.9245 |
92
+ | 0.44 | 40.0 | 2560 | 0.3735 | 0.9132 | 0.9129 | 0.9245 |
93
+ | 0.4259 | 41.0 | 2624 | 0.3729 | 0.9110 | 0.9105 | 0.9251 |
94
+ | 0.4273 | 42.0 | 2688 | 0.3724 | 0.9132 | 0.9129 | 0.9255 |
95
+ | 0.4484 | 43.0 | 2752 | 0.3730 | 0.9110 | 0.9104 | 0.925 |
96
+ | 0.418 | 44.0 | 2816 | 0.3705 | 0.9132 | 0.9128 | 0.925 |
97
+ | 0.4197 | 45.0 | 2880 | 0.3688 | 0.9132 | 0.9131 | 0.925 |
98
+ | 0.4411 | 46.0 | 2944 | 0.3696 | 0.9132 | 0.9129 | 0.9245 |
99
+ | 0.4251 | 47.0 | 3008 | 0.3695 | 0.9132 | 0.9128 | 0.9245 |
100
+ | 0.4313 | 48.0 | 3072 | 0.3685 | 0.9110 | 0.9106 | 0.9245 |
101
+ | 0.4305 | 49.0 | 3136 | 0.3685 | 0.9132 | 0.9128 | 0.9255 |
102
+ | 0.4158 | 50.0 | 3200 | 0.3685 | 0.9132 | 0.9128 | 0.9255 |
103
 
104
 
105
  ### Framework versions
config.json CHANGED
@@ -7,7 +7,7 @@
7
  "dim": 768,
8
  "dropout": 0.1,
9
  "hidden_dim": 3072,
10
- "id2label": {
11
  "0": "bye",
12
  "1": "cancel",
13
  "2": "greeting",
@@ -28,7 +28,7 @@
28
  "6": "I-practitioner_name"
29
  },
30
  "initializer_range": 0.02,
31
- "label2id": {
32
  "bye": 0,
33
  "cancel": 1,
34
  "greeting": 2,
 
7
  "dim": 768,
8
  "dropout": 0.1,
9
  "hidden_dim": 3072,
10
+ "id2label_intent": {
11
  "0": "bye",
12
  "1": "cancel",
13
  "2": "greeting",
 
28
  "6": "I-practitioner_name"
29
  },
30
  "initializer_range": 0.02,
31
+ "label2id_intent": {
32
  "bye": 0,
33
  "cancel": 1,
34
  "greeting": 2,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:980d1ae83c64a61efe21afa1e57bc15394ee4ac110914a2437f09573fe4db2cf
3
  size 267851552
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:21d0702e9c2f5e2b496e278a68f340662a5352afcf5375f9f90bb2830e69d5a4
3
  size 267851552
runs/Jul18_11-24-49_83c12e474755/events.out.tfevents.1752837890.83c12e474755.863.1 ADDED
Binary file (38.4 kB). View file
 
training_args.bin CHANGED
Binary files a/training_args.bin and b/training_args.bin differ