LightChen2333 commited on
Commit
e03f735
1 Parent(s): cb35c21

Upload 3 files

Browse files
Files changed (3) hide show
  1. config.json +251 -0
  2. pytorch_model.bin +3 -0
  3. tokenizer.pkl +3 -0
config.json ADDED
@@ -0,0 +1,251 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_id2label": {
3
+ "intent": [
4
+ "PlayMusic",
5
+ "AddToPlaylist",
6
+ "RateBook",
7
+ "SearchScreeningEvent",
8
+ "BookRestaurant",
9
+ "GetWeather",
10
+ "SearchCreativeWork"
11
+ ],
12
+ "slot": [
13
+ "O",
14
+ "B-artist",
15
+ "B-album",
16
+ "B-service",
17
+ "I-service",
18
+ "B-entity_name",
19
+ "I-entity_name",
20
+ "B-playlist",
21
+ "I-playlist",
22
+ "B-object_select",
23
+ "B-object_type",
24
+ "B-rating_value",
25
+ "B-best_rating",
26
+ "B-music_item",
27
+ "B-track",
28
+ "I-track",
29
+ "I-artist",
30
+ "B-playlist_owner",
31
+ "B-year",
32
+ "B-sort",
33
+ "B-movie_name",
34
+ "I-movie_name",
35
+ "B-party_size_number",
36
+ "B-state",
37
+ "B-city",
38
+ "B-timeRange",
39
+ "I-timeRange",
40
+ "B-object_part_of_series_type",
41
+ "I-object_type",
42
+ "B-movie_type",
43
+ "B-spatial_relation",
44
+ "I-spatial_relation",
45
+ "B-geographic_poi",
46
+ "I-geographic_poi",
47
+ "B-restaurant_type",
48
+ "I-city",
49
+ "B-party_size_description",
50
+ "I-party_size_description",
51
+ "B-object_location_type",
52
+ "I-object_location_type",
53
+ "B-object_name",
54
+ "I-object_name",
55
+ "I-movie_type",
56
+ "B-rating_unit",
57
+ "I-sort",
58
+ "B-location_name",
59
+ "I-location_name",
60
+ "B-current_location",
61
+ "I-current_location",
62
+ "I-playlist_owner",
63
+ "B-served_dish",
64
+ "B-country",
65
+ "B-condition_temperature",
66
+ "B-poi",
67
+ "I-poi",
68
+ "B-condition_description",
69
+ "B-genre",
70
+ "B-restaurant_name",
71
+ "I-restaurant_name",
72
+ "I-state",
73
+ "I-served_dish",
74
+ "B-cuisine",
75
+ "I-album",
76
+ "I-country",
77
+ "B-facility",
78
+ "I-facility",
79
+ "I-cuisine",
80
+ "I-music_item",
81
+ "I-object_select",
82
+ "I-restaurant_type",
83
+ "I-object_part_of_series_type",
84
+ "I-genre"
85
+ ]
86
+ },
87
+ "_label2id": {
88
+ "intent": {
89
+ "AddToPlaylist": 1,
90
+ "BookRestaurant": 4,
91
+ "GetWeather": 5,
92
+ "PlayMusic": 0,
93
+ "RateBook": 2,
94
+ "SearchCreativeWork": 6,
95
+ "SearchScreeningEvent": 3
96
+ },
97
+ "slot": {
98
+ "B-album": 2,
99
+ "B-artist": 1,
100
+ "B-best_rating": 12,
101
+ "B-city": 24,
102
+ "B-condition_description": 55,
103
+ "B-condition_temperature": 52,
104
+ "B-country": 51,
105
+ "B-cuisine": 61,
106
+ "B-current_location": 47,
107
+ "B-entity_name": 5,
108
+ "B-facility": 64,
109
+ "B-genre": 56,
110
+ "B-geographic_poi": 32,
111
+ "B-location_name": 45,
112
+ "B-movie_name": 20,
113
+ "B-movie_type": 29,
114
+ "B-music_item": 13,
115
+ "B-object_location_type": 38,
116
+ "B-object_name": 40,
117
+ "B-object_part_of_series_type": 27,
118
+ "B-object_select": 9,
119
+ "B-object_type": 10,
120
+ "B-party_size_description": 36,
121
+ "B-party_size_number": 22,
122
+ "B-playlist": 7,
123
+ "B-playlist_owner": 17,
124
+ "B-poi": 53,
125
+ "B-rating_unit": 43,
126
+ "B-rating_value": 11,
127
+ "B-restaurant_name": 57,
128
+ "B-restaurant_type": 34,
129
+ "B-served_dish": 50,
130
+ "B-service": 3,
131
+ "B-sort": 19,
132
+ "B-spatial_relation": 30,
133
+ "B-state": 23,
134
+ "B-timeRange": 25,
135
+ "B-track": 14,
136
+ "B-year": 18,
137
+ "I-album": 62,
138
+ "I-artist": 16,
139
+ "I-city": 35,
140
+ "I-country": 63,
141
+ "I-cuisine": 66,
142
+ "I-current_location": 48,
143
+ "I-entity_name": 6,
144
+ "I-facility": 65,
145
+ "I-genre": 71,
146
+ "I-geographic_poi": 33,
147
+ "I-location_name": 46,
148
+ "I-movie_name": 21,
149
+ "I-movie_type": 42,
150
+ "I-music_item": 67,
151
+ "I-object_location_type": 39,
152
+ "I-object_name": 41,
153
+ "I-object_part_of_series_type": 70,
154
+ "I-object_select": 68,
155
+ "I-object_type": 28,
156
+ "I-party_size_description": 37,
157
+ "I-playlist": 8,
158
+ "I-playlist_owner": 49,
159
+ "I-poi": 54,
160
+ "I-restaurant_name": 58,
161
+ "I-restaurant_type": 69,
162
+ "I-served_dish": 60,
163
+ "I-service": 4,
164
+ "I-sort": 44,
165
+ "I-spatial_relation": 31,
166
+ "I-state": 59,
167
+ "I-timeRange": 26,
168
+ "I-track": 15,
169
+ "O": 0
170
+ }
171
+ },
172
+ "_name_or_path": "OpenSLUv1",
173
+ "_num_labels": {
174
+ "intent": 7,
175
+ "slot": 72
176
+ },
177
+ "architectures": [
178
+ "PretrainedModelForSLUToSave"
179
+ ],
180
+ "is_decoder": true,
181
+ "model": {
182
+ "_model_target_": "model.OpenSLUModel",
183
+ "decoder": {
184
+ "_model_target_": "model.decoder.StackPropagationDecoder",
185
+ "intent_classifier": {
186
+ "_model_target_": "model.decoder.classifier.AutoregressiveLSTMClassifier",
187
+ "bidirectional": false,
188
+ "dropout_rate": 0.4,
189
+ "embedding_dim": 8,
190
+ "force_ratio": 0.9,
191
+ "hidden_dim": 64,
192
+ "ignore_index": -100,
193
+ "input_dim": 384,
194
+ "intent_label_num": 7,
195
+ "layer_num": 1,
196
+ "mode": "token-level-intent",
197
+ "return_sentence_level": true,
198
+ "use_intent": true,
199
+ "use_multi": false
200
+ },
201
+ "interaction": {
202
+ "_model_target_": "model.decoder.interaction.StackInteraction",
203
+ "differentiable": false,
204
+ "input_dim": 384
205
+ },
206
+ "slot_classifier": {
207
+ "_model_target_": "model.decoder.classifier.AutoregressiveLSTMClassifier",
208
+ "bidirectional": false,
209
+ "dropout_rate": 0.4,
210
+ "embedding_dim": 32,
211
+ "force_ratio": 0.9,
212
+ "hidden_dim": 64,
213
+ "ignore_index": -100,
214
+ "input_dim": 384,
215
+ "layer_num": 1,
216
+ "mode": "slot",
217
+ "return_sentence_level": false,
218
+ "slot_label_num": 72,
219
+ "use_multi": false,
220
+ "use_slot": true
221
+ }
222
+ },
223
+ "encoder": {
224
+ "_model_target_": "model.encoder.AutoEncoder",
225
+ "attention": {
226
+ "dropout_rate": 0.4,
227
+ "hidden_dim": 1024,
228
+ "output_dim": 128
229
+ },
230
+ "embedding": {
231
+ "dropout_rate": 0.4,
232
+ "embedding_dim": 256
233
+ },
234
+ "encoder_name": "self-attention-lstm",
235
+ "lstm": {
236
+ "bidirectional": true,
237
+ "dropout_rate": 0.4,
238
+ "layer_num": 1,
239
+ "output_dim": 256
240
+ },
241
+ "output_dim": 384,
242
+ "return_sentence_level_hidden": false,
243
+ "return_with_input": true
244
+ }
245
+ },
246
+ "return_dict": false,
247
+ "tokenizer_class": "OpenSLUv1",
248
+ "torch_dtype": "float32",
249
+ "transformers_version": "4.25.1",
250
+ "vocab_size": 11767
251
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0e75316bf85f6a71fdb67322f629e6e4b12c5bf6f8a79d9b43bc6b3d8b124c21
3
+ size 16875564
tokenizer.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fee59a56fb6cbbe664361aaf51d6a4e518c78a6af7beeac249f21b01fba4ab2a
3
+ size 205516