LightChen2333 commited on
Commit
67622b0
1 Parent(s): f199679

Upload 3 files

Browse files
Files changed (3) hide show
  1. config.json +377 -0
  2. pytorch_model.bin +3 -0
  3. tokenizer.pkl +3 -0
config.json ADDED
@@ -0,0 +1,377 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_id2label": {
3
+ "intent": [
4
+ "atis_abbreviation",
5
+ "atis_airport",
6
+ "atis_city",
7
+ "atis_capacity",
8
+ "atis_aircraft",
9
+ "atis_ground_service",
10
+ "atis_meal",
11
+ "atis_quantity",
12
+ "atis_flight_no",
13
+ "atis_flight_time",
14
+ "atis_ground_fare",
15
+ "atis_cheapest",
16
+ "atis_flight",
17
+ "atis_distance",
18
+ "atis_restriction",
19
+ "atis_airfare",
20
+ "atis_airline"
21
+ ],
22
+ "slot": [
23
+ "O",
24
+ "B-airline_code",
25
+ "B-airline_name",
26
+ "I-airline_name",
27
+ "B-aircraft_code",
28
+ "B-city_name",
29
+ "I-city_name",
30
+ "B-meal",
31
+ "B-flight_number",
32
+ "B-fromloc.city_name",
33
+ "I-fromloc.city_name",
34
+ "B-toloc.city_name",
35
+ "I-toloc.city_name",
36
+ "B-depart_time.time_relative",
37
+ "B-depart_time.time",
38
+ "I-depart_time.time",
39
+ "B-depart_time.period_of_day",
40
+ "B-class_type",
41
+ "I-class_type",
42
+ "B-flight_time",
43
+ "B-cost_relative",
44
+ "I-cost_relative",
45
+ "B-depart_date.day_name",
46
+ "B-toloc.airport_name",
47
+ "I-toloc.airport_name",
48
+ "B-transport_type",
49
+ "I-transport_type",
50
+ "B-arrive_time.time_relative",
51
+ "B-arrive_time.time",
52
+ "I-arrive_time.time",
53
+ "B-arrive_date.day_name",
54
+ "B-or",
55
+ "B-economy",
56
+ "I-economy",
57
+ "B-depart_date.month_name",
58
+ "B-depart_date.day_number",
59
+ "B-fromloc.airport_name",
60
+ "I-fromloc.airport_name",
61
+ "B-airport_name",
62
+ "I-airport_name",
63
+ "B-fare_amount",
64
+ "I-fare_amount",
65
+ "B-round_trip",
66
+ "I-round_trip",
67
+ "B-fare_basis_code",
68
+ "B-toloc.state_code",
69
+ "B-restriction_code",
70
+ "B-stoploc.city_name",
71
+ "I-stoploc.city_name",
72
+ "I-depart_time.period_of_day",
73
+ "B-fromloc.state_name",
74
+ "B-toloc.state_name",
75
+ "B-flight_mod",
76
+ "B-toloc.airport_code",
77
+ "B-mod",
78
+ "B-depart_date.year",
79
+ "I-flight_time",
80
+ "B-airport_code",
81
+ "B-fromloc.airport_code",
82
+ "B-arrive_date.month_name",
83
+ "B-arrive_date.day_number",
84
+ "B-depart_date.date_relative",
85
+ "I-depart_date.day_number",
86
+ "B-state_code",
87
+ "B-flight_stop",
88
+ "B-depart_time.start_time",
89
+ "B-depart_time.end_time",
90
+ "B-depart_date.today_relative",
91
+ "B-day_name",
92
+ "B-month_name",
93
+ "B-day_number",
94
+ "B-meal_description",
95
+ "I-meal_description",
96
+ "I-restriction_code",
97
+ "B-fromloc.state_code",
98
+ "I-arrive_date.day_number",
99
+ "B-arrive_time.start_time",
100
+ "I-arrive_time.start_time",
101
+ "B-arrive_time.end_time",
102
+ "I-arrive_time.end_time",
103
+ "B-meal_code",
104
+ "I-flight_mod",
105
+ "B-days_code",
106
+ "B-today_relative",
107
+ "I-today_relative",
108
+ "B-time",
109
+ "B-flight_days",
110
+ "B-connect",
111
+ "B-period_of_day",
112
+ "I-meal_code",
113
+ "B-time_relative",
114
+ "I-time",
115
+ "B-arrive_time.period_of_day",
116
+ "B-return_date.month_name",
117
+ "B-return_date.day_number",
118
+ "I-toloc.state_name",
119
+ "B-state_name",
120
+ "B-depart_time.period_mod",
121
+ "I-arrive_time.period_of_day",
122
+ "I-flight_stop",
123
+ "I-depart_date.today_relative",
124
+ "I-fromloc.state_name",
125
+ "I-depart_time.end_time",
126
+ "B-arrive_date.date_relative",
127
+ "B-return_date.date_relative",
128
+ "I-depart_time.start_time",
129
+ "B-stoploc.state_code",
130
+ "I-depart_time.time_relative",
131
+ "B-stoploc.airport_name",
132
+ "B-arrive_time.period_mod",
133
+ "B-toloc.country_name"
134
+ ]
135
+ },
136
+ "_label2id": {
137
+ "intent": {
138
+ "atis_abbreviation": 0,
139
+ "atis_aircraft": 4,
140
+ "atis_airfare": 15,
141
+ "atis_airline": 16,
142
+ "atis_airport": 1,
143
+ "atis_capacity": 3,
144
+ "atis_cheapest": 11,
145
+ "atis_city": 2,
146
+ "atis_distance": 13,
147
+ "atis_flight": 12,
148
+ "atis_flight_no": 8,
149
+ "atis_flight_time": 9,
150
+ "atis_ground_fare": 10,
151
+ "atis_ground_service": 5,
152
+ "atis_meal": 6,
153
+ "atis_quantity": 7,
154
+ "atis_restriction": 14
155
+ },
156
+ "slot": {
157
+ "B-aircraft_code": 4,
158
+ "B-airline_code": 1,
159
+ "B-airline_name": 2,
160
+ "B-airport_code": 57,
161
+ "B-airport_name": 38,
162
+ "B-arrive_date.date_relative": 103,
163
+ "B-arrive_date.day_name": 30,
164
+ "B-arrive_date.day_number": 60,
165
+ "B-arrive_date.month_name": 59,
166
+ "B-arrive_time.end_time": 78,
167
+ "B-arrive_time.period_mod": 109,
168
+ "B-arrive_time.period_of_day": 92,
169
+ "B-arrive_time.start_time": 76,
170
+ "B-arrive_time.time": 28,
171
+ "B-arrive_time.time_relative": 27,
172
+ "B-city_name": 5,
173
+ "B-class_type": 17,
174
+ "B-connect": 87,
175
+ "B-cost_relative": 20,
176
+ "B-day_name": 68,
177
+ "B-day_number": 70,
178
+ "B-days_code": 82,
179
+ "B-depart_date.date_relative": 61,
180
+ "B-depart_date.day_name": 22,
181
+ "B-depart_date.day_number": 35,
182
+ "B-depart_date.month_name": 34,
183
+ "B-depart_date.today_relative": 67,
184
+ "B-depart_date.year": 55,
185
+ "B-depart_time.end_time": 66,
186
+ "B-depart_time.period_mod": 97,
187
+ "B-depart_time.period_of_day": 16,
188
+ "B-depart_time.start_time": 65,
189
+ "B-depart_time.time": 14,
190
+ "B-depart_time.time_relative": 13,
191
+ "B-economy": 32,
192
+ "B-fare_amount": 40,
193
+ "B-fare_basis_code": 44,
194
+ "B-flight_days": 86,
195
+ "B-flight_mod": 52,
196
+ "B-flight_number": 8,
197
+ "B-flight_stop": 64,
198
+ "B-flight_time": 19,
199
+ "B-fromloc.airport_code": 58,
200
+ "B-fromloc.airport_name": 36,
201
+ "B-fromloc.city_name": 9,
202
+ "B-fromloc.state_code": 74,
203
+ "B-fromloc.state_name": 50,
204
+ "B-meal": 7,
205
+ "B-meal_code": 80,
206
+ "B-meal_description": 71,
207
+ "B-mod": 54,
208
+ "B-month_name": 69,
209
+ "B-or": 31,
210
+ "B-period_of_day": 88,
211
+ "B-restriction_code": 46,
212
+ "B-return_date.date_relative": 104,
213
+ "B-return_date.day_number": 94,
214
+ "B-return_date.month_name": 93,
215
+ "B-round_trip": 42,
216
+ "B-state_code": 63,
217
+ "B-state_name": 96,
218
+ "B-stoploc.airport_name": 108,
219
+ "B-stoploc.city_name": 47,
220
+ "B-stoploc.state_code": 106,
221
+ "B-time": 85,
222
+ "B-time_relative": 90,
223
+ "B-today_relative": 83,
224
+ "B-toloc.airport_code": 53,
225
+ "B-toloc.airport_name": 23,
226
+ "B-toloc.city_name": 11,
227
+ "B-toloc.country_name": 110,
228
+ "B-toloc.state_code": 45,
229
+ "B-toloc.state_name": 51,
230
+ "B-transport_type": 25,
231
+ "I-airline_name": 3,
232
+ "I-airport_name": 39,
233
+ "I-arrive_date.day_number": 75,
234
+ "I-arrive_time.end_time": 79,
235
+ "I-arrive_time.period_of_day": 98,
236
+ "I-arrive_time.start_time": 77,
237
+ "I-arrive_time.time": 29,
238
+ "I-city_name": 6,
239
+ "I-class_type": 18,
240
+ "I-cost_relative": 21,
241
+ "I-depart_date.day_number": 62,
242
+ "I-depart_date.today_relative": 100,
243
+ "I-depart_time.end_time": 102,
244
+ "I-depart_time.period_of_day": 49,
245
+ "I-depart_time.start_time": 105,
246
+ "I-depart_time.time": 15,
247
+ "I-depart_time.time_relative": 107,
248
+ "I-economy": 33,
249
+ "I-fare_amount": 41,
250
+ "I-flight_mod": 81,
251
+ "I-flight_stop": 99,
252
+ "I-flight_time": 56,
253
+ "I-fromloc.airport_name": 37,
254
+ "I-fromloc.city_name": 10,
255
+ "I-fromloc.state_name": 101,
256
+ "I-meal_code": 89,
257
+ "I-meal_description": 72,
258
+ "I-restriction_code": 73,
259
+ "I-round_trip": 43,
260
+ "I-stoploc.city_name": 48,
261
+ "I-time": 91,
262
+ "I-today_relative": 84,
263
+ "I-toloc.airport_name": 24,
264
+ "I-toloc.city_name": 12,
265
+ "I-toloc.state_name": 95,
266
+ "I-transport_type": 26,
267
+ "O": 0
268
+ }
269
+ },
270
+ "_name_or_path": "OpenSLUv1",
271
+ "_num_labels": {
272
+ "intent": 17,
273
+ "slot": 111
274
+ },
275
+ "architectures": [
276
+ "PretrainedModelForSLUToSave"
277
+ ],
278
+ "is_decoder": true,
279
+ "model": {
280
+ "_model_target_": "model.OpenSLUModel",
281
+ "decoder": {
282
+ "_model_target_": "model.decoder.AGIFDecoder",
283
+ "intent_classifier": {
284
+ "_model_target_": "model.decoder.classifier.MLPClassifier",
285
+ "dropout_rate": 0.4,
286
+ "ignore_index": -100,
287
+ "input_dim": 384,
288
+ "intent_label_num": 17,
289
+ "loss_fn": {
290
+ "_model_target_": "torch.nn.BCEWithLogitsLoss"
291
+ },
292
+ "mlp": [
293
+ {
294
+ "_model_target_": "torch.nn.Linear",
295
+ "in_features": "{model.encoder.output_dim}",
296
+ "out_features": 256
297
+ },
298
+ {
299
+ "_model_target_": "torch.nn.LeakyReLU",
300
+ "negative_slope": 0.2
301
+ },
302
+ {
303
+ "_model_target_": "torch.nn.Linear",
304
+ "in_features": 256,
305
+ "out_features": "{base.intent_label_num}"
306
+ }
307
+ ],
308
+ "mode": "intent",
309
+ "multi_threshold": 0.5,
310
+ "return_sentence_level": true,
311
+ "use_intent": true,
312
+ "use_multi": "{base.multi_intent}",
313
+ "weight": 0.3
314
+ },
315
+ "interaction": {
316
+ "_model_target_": "model.decoder.interaction.AGIFInteraction",
317
+ "alpha": 0.2,
318
+ "dropout_rate": 0.4,
319
+ "hidden_dim": 128,
320
+ "input_dim": "{model.encoder.output_dim}",
321
+ "intent_embedding_dim": 128,
322
+ "num_heads": 4,
323
+ "num_layers": 2,
324
+ "output_dim": "{model.decoder.interaction.intent_embedding_dim}",
325
+ "row_normalized": true
326
+ },
327
+ "slot_classifier": {
328
+ "_model_target_": "model.decoder.classifier.AutoregressiveLSTMClassifier",
329
+ "bidirectional": false,
330
+ "dropout_rate": 0.4,
331
+ "embedding_dim": 128,
332
+ "force_ratio": 0.9,
333
+ "hidden_dim": "{model.decoder.interaction.intent_embedding_dim}",
334
+ "ignore_index": -100,
335
+ "input_dim": "{model.encoder.output_dim}",
336
+ "layer_num": 1,
337
+ "mode": "slot",
338
+ "multi_threshold": 0.5,
339
+ "return_sentence_level": false,
340
+ "slot_label_num": 111,
341
+ "use_multi": false,
342
+ "use_slot": true,
343
+ "weight": 0.7
344
+ }
345
+ },
346
+ "encoder": {
347
+ "_model_target_": "model.encoder.AutoEncoder",
348
+ "attention": {
349
+ "dropout_rate": 0.4,
350
+ "hidden_dim": 1024,
351
+ "output_dim": 128
352
+ },
353
+ "embedding": {
354
+ "dropout_rate": 0.4,
355
+ "embedding_dim": 128
356
+ },
357
+ "encoder_name": "self-attention-lstm",
358
+ "lstm": {
359
+ "bidirectional": true,
360
+ "dropout_rate": 0.4,
361
+ "layer_num": 1,
362
+ "output_dim": 256
363
+ },
364
+ "output_dim": 384,
365
+ "return_sentence_level_hidden": true,
366
+ "return_with_input": true,
367
+ "unflat_attention": {
368
+ "dropout_rate": 0.4
369
+ }
370
+ }
371
+ },
372
+ "return_dict": false,
373
+ "tokenizer_class": "OpenSLUv1",
374
+ "torch_dtype": "float32",
375
+ "transformers_version": "4.25.1",
376
+ "vocab_size": 790
377
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f6e60fa2a64b89d3e58ef15156a539842a6038b38a34c7157ec7113915ddc155
3
+ size 4030475
tokenizer.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5435e3d70b344234613dbde8264bbcd6c4a1c990b3848a0801907aa0d5426361
3
+ size 12218