LightChen2333 commited on
Commit
e4bf7a2
1 Parent(s): 39ab68d

Upload 3 files

Browse files
Files changed (3) hide show
  1. config.json +161 -285
  2. pytorch_model.bin +2 -2
  3. tokenizer.pkl +2 -2
config.json CHANGED
@@ -1,302 +1,178 @@
1
  {
2
  "_id2label": {
3
  "intent": [
4
- "atis_flight",
5
- "atis_airfare",
6
- "atis_airline",
7
- "atis_ground_service",
8
- "atis_quantity",
9
- "atis_city",
10
- "atis_flight#atis_airfare",
11
- "atis_abbreviation",
12
- "atis_aircraft",
13
- "atis_distance",
14
- "atis_ground_fare",
15
- "atis_capacity",
16
- "atis_flight_time",
17
- "atis_meal",
18
- "atis_aircraft#atis_flight#atis_flight_no",
19
- "atis_flight_no",
20
- "atis_restriction",
21
- "atis_airport",
22
- "atis_airline#atis_flight_no",
23
- "atis_cheapest",
24
- "atis_ground_service#atis_ground_fare"
25
  ],
26
  "slot": [
27
  "O",
28
- "B-fromloc.city_name",
29
- "B-toloc.city_name",
30
- "B-round_trip",
31
- "I-round_trip",
32
- "B-cost_relative",
33
- "B-fare_amount",
34
- "I-fare_amount",
35
- "B-arrive_date.month_name",
36
- "B-arrive_date.day_number",
37
- "I-fromloc.city_name",
38
- "B-stoploc.city_name",
39
- "B-arrive_time.time_relative",
40
- "B-arrive_time.time",
41
- "I-arrive_time.time",
42
- "B-toloc.state_code",
43
- "I-toloc.city_name",
44
- "I-stoploc.city_name",
45
- "B-meal_description",
46
- "B-depart_date.month_name",
47
- "B-depart_date.day_number",
48
- "B-airline_name",
49
- "I-airline_name",
50
- "B-depart_time.period_of_day",
51
- "B-depart_date.day_name",
52
- "B-toloc.state_name",
53
- "B-depart_time.time_relative",
54
- "B-depart_time.time",
55
- "B-toloc.airport_name",
56
- "I-toloc.airport_name",
57
- "B-depart_date.date_relative",
58
- "B-or",
59
- "B-airline_code",
60
- "B-class_type",
61
- "I-class_type",
62
- "I-cost_relative",
63
- "I-depart_time.time",
64
- "B-fromloc.airport_name",
65
- "I-fromloc.airport_name",
66
- "B-city_name",
67
- "B-flight_mod",
68
- "B-meal",
69
- "B-economy",
70
- "B-fare_basis_code",
71
- "I-depart_date.day_number",
72
- "B-depart_date.today_relative",
73
- "B-flight_stop",
74
- "B-airport_code",
75
- "B-fromloc.state_name",
76
- "I-fromloc.state_name",
77
- "I-city_name",
78
- "B-connect",
79
- "B-arrive_date.day_name",
80
- "B-fromloc.state_code",
81
- "B-arrive_date.today_relative",
82
- "B-depart_date.year",
83
- "B-depart_time.start_time",
84
- "I-depart_time.start_time",
85
- "B-depart_time.end_time",
86
- "I-depart_time.end_time",
87
- "B-arrive_time.start_time",
88
- "B-arrive_time.end_time",
89
- "I-arrive_time.end_time",
90
- "I-flight_mod",
91
- "B-flight_days",
92
- "B-mod",
93
- "B-flight_number",
94
- "I-toloc.state_name",
95
- "B-meal_code",
96
- "I-meal_code",
97
- "B-airport_name",
98
- "I-airport_name",
99
- "I-flight_stop",
100
- "B-transport_type",
101
- "I-transport_type",
102
- "B-state_code",
103
- "B-aircraft_code",
104
- "B-toloc.country_name",
105
- "I-arrive_date.day_number",
106
- "B-toloc.airport_code",
107
- "B-return_date.date_relative",
108
- "I-return_date.date_relative",
109
- "B-flight_time",
110
- "I-economy",
111
- "B-fromloc.airport_code",
112
- "B-arrive_time.period_of_day",
113
- "B-depart_time.period_mod",
114
- "I-flight_time",
115
- "B-return_date.day_name",
116
- "B-arrive_date.date_relative",
117
- "B-restriction_code",
118
- "I-restriction_code",
119
- "B-arrive_time.period_mod",
120
- "I-arrive_time.period_of_day",
121
- "B-period_of_day",
122
- "B-stoploc.state_code",
123
- "I-depart_date.today_relative",
124
- "I-fare_basis_code",
125
- "I-arrive_time.start_time",
126
- "B-time",
127
- "B-today_relative",
128
- "I-today_relative",
129
- "B-state_name",
130
- "B-days_code",
131
- "I-depart_time.period_of_day",
132
- "I-arrive_time.time_relative",
133
- "B-time_relative",
134
- "I-time",
135
- "B-return_date.month_name",
136
- "B-return_date.day_number",
137
- "I-depart_time.time_relative",
138
- "B-stoploc.airport_name",
139
- "B-day_name",
140
- "B-month_name",
141
- "B-day_number",
142
- "B-return_time.period_mod",
143
- "B-return_time.period_of_day",
144
- "B-return_date.today_relative",
145
- "I-return_date.today_relative",
146
- "I-meal_description"
147
  ]
148
  },
149
  "_label2id": {
150
  "intent": {
151
- "atis_abbreviation": 7,
152
- "atis_aircraft": 8,
153
- "atis_aircraft#atis_flight#atis_flight_no": 14,
154
- "atis_airfare": 1,
155
- "atis_airline": 2,
156
- "atis_airline#atis_flight_no": 18,
157
- "atis_airport": 17,
158
- "atis_capacity": 11,
159
- "atis_cheapest": 19,
160
- "atis_city": 5,
161
- "atis_distance": 9,
162
- "atis_flight": 0,
163
- "atis_flight#atis_airfare": 6,
164
- "atis_flight_no": 15,
165
- "atis_flight_time": 12,
166
- "atis_ground_fare": 10,
167
- "atis_ground_service": 3,
168
- "atis_ground_service#atis_ground_fare": 20,
169
- "atis_meal": 13,
170
- "atis_quantity": 4,
171
- "atis_restriction": 16
172
  },
173
  "slot": {
174
- "B-aircraft_code": 76,
175
- "B-airline_code": 32,
176
- "B-airline_name": 21,
177
- "B-airport_code": 47,
178
- "B-airport_name": 70,
179
- "B-arrive_date.date_relative": 89,
180
- "B-arrive_date.day_name": 52,
181
- "B-arrive_date.day_number": 9,
182
- "B-arrive_date.month_name": 8,
183
- "B-arrive_date.today_relative": 54,
184
- "B-arrive_time.end_time": 61,
185
- "B-arrive_time.period_mod": 92,
186
- "B-arrive_time.period_of_day": 85,
187
- "B-arrive_time.start_time": 60,
188
- "B-arrive_time.time": 13,
189
- "B-arrive_time.time_relative": 12,
190
- "B-city_name": 39,
191
- "B-class_type": 33,
192
- "B-connect": 51,
193
- "B-cost_relative": 5,
194
- "B-day_name": 112,
195
- "B-day_number": 114,
196
- "B-days_code": 103,
197
- "B-depart_date.date_relative": 30,
198
- "B-depart_date.day_name": 24,
199
- "B-depart_date.day_number": 20,
200
- "B-depart_date.month_name": 19,
201
- "B-depart_date.today_relative": 45,
202
- "B-depart_date.year": 55,
203
- "B-depart_time.end_time": 58,
204
- "B-depart_time.period_mod": 86,
205
- "B-depart_time.period_of_day": 23,
206
- "B-depart_time.start_time": 56,
207
- "B-depart_time.time": 27,
208
- "B-depart_time.time_relative": 26,
209
- "B-economy": 42,
210
- "B-fare_amount": 6,
211
- "B-fare_basis_code": 43,
212
- "B-flight_days": 64,
213
- "B-flight_mod": 40,
214
- "B-flight_number": 66,
215
- "B-flight_stop": 46,
216
- "B-flight_time": 82,
217
- "B-fromloc.airport_code": 84,
218
- "B-fromloc.airport_name": 37,
219
- "B-fromloc.city_name": 1,
220
- "B-fromloc.state_code": 53,
221
- "B-fromloc.state_name": 48,
222
- "B-meal": 41,
223
- "B-meal_code": 68,
224
- "B-meal_description": 18,
225
- "B-mod": 65,
226
- "B-month_name": 113,
227
- "B-or": 31,
228
- "B-period_of_day": 94,
229
- "B-restriction_code": 90,
230
- "B-return_date.date_relative": 80,
231
- "B-return_date.day_name": 88,
232
- "B-return_date.day_number": 109,
233
- "B-return_date.month_name": 108,
234
- "B-return_date.today_relative": 117,
235
- "B-return_time.period_mod": 115,
236
- "B-return_time.period_of_day": 116,
237
- "B-round_trip": 3,
238
- "B-state_code": 75,
239
- "B-state_name": 102,
240
- "B-stoploc.airport_name": 111,
241
- "B-stoploc.city_name": 11,
242
- "B-stoploc.state_code": 95,
243
- "B-time": 99,
244
- "B-time_relative": 106,
245
- "B-today_relative": 100,
246
- "B-toloc.airport_code": 79,
247
- "B-toloc.airport_name": 28,
248
- "B-toloc.city_name": 2,
249
- "B-toloc.country_name": 77,
250
- "B-toloc.state_code": 15,
251
- "B-toloc.state_name": 25,
252
- "B-transport_type": 73,
253
- "I-airline_name": 22,
254
- "I-airport_name": 71,
255
- "I-arrive_date.day_number": 78,
256
- "I-arrive_time.end_time": 62,
257
- "I-arrive_time.period_of_day": 93,
258
- "I-arrive_time.start_time": 98,
259
- "I-arrive_time.time": 14,
260
- "I-arrive_time.time_relative": 105,
261
- "I-city_name": 50,
262
- "I-class_type": 34,
263
- "I-cost_relative": 35,
264
- "I-depart_date.day_number": 44,
265
- "I-depart_date.today_relative": 96,
266
- "I-depart_time.end_time": 59,
267
- "I-depart_time.period_of_day": 104,
268
- "I-depart_time.start_time": 57,
269
- "I-depart_time.time": 36,
270
- "I-depart_time.time_relative": 110,
271
- "I-economy": 83,
272
- "I-fare_amount": 7,
273
- "I-fare_basis_code": 97,
274
- "I-flight_mod": 63,
275
- "I-flight_stop": 72,
276
- "I-flight_time": 87,
277
- "I-fromloc.airport_name": 38,
278
- "I-fromloc.city_name": 10,
279
- "I-fromloc.state_name": 49,
280
- "I-meal_code": 69,
281
- "I-meal_description": 119,
282
- "I-restriction_code": 91,
283
- "I-return_date.date_relative": 81,
284
- "I-return_date.today_relative": 118,
285
- "I-round_trip": 4,
286
- "I-stoploc.city_name": 17,
287
- "I-time": 107,
288
- "I-today_relative": 101,
289
- "I-toloc.airport_name": 29,
290
- "I-toloc.city_name": 16,
291
- "I-toloc.state_name": 67,
292
- "I-transport_type": 74,
293
  "O": 0
294
  }
295
  },
296
  "_name_or_path": "OpenSLUv1",
297
  "_num_labels": {
298
- "intent": 21,
299
- "slot": 120
300
  },
301
  "architectures": [
302
  "PretrainedModelForSLUToSave"
@@ -310,7 +186,7 @@
310
  "_model_target_": "model.decoder.classifier.LinearClassifier",
311
  "ignore_index": -100,
312
  "input_dim": 256,
313
- "intent_label_num": 21,
314
  "mode": "intent",
315
  "use_intent": true
316
  },
@@ -326,7 +202,7 @@
326
  "ignore_index": -100,
327
  "input_dim": 256,
328
  "mode": "slot",
329
- "slot_label_num": 120,
330
  "use_slot": true
331
  }
332
  },
@@ -353,5 +229,5 @@
353
  "tokenizer_class": "OpenSLUv1",
354
  "torch_dtype": "float32",
355
  "transformers_version": "4.25.1",
356
- "vocab_size": 900
357
  }
 
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"
 
186
  "_model_target_": "model.decoder.classifier.LinearClassifier",
187
  "ignore_index": -100,
188
  "input_dim": 256,
189
+ "intent_label_num": 7,
190
  "mode": "intent",
191
  "use_intent": true
192
  },
 
202
  "ignore_index": -100,
203
  "input_dim": 256,
204
  "mode": "slot",
205
+ "slot_label_num": 72,
206
  "use_slot": true
207
  }
208
  },
 
229
  "tokenizer_class": "OpenSLUv1",
230
  "torch_dtype": "float32",
231
  "transformers_version": "4.25.1",
232
+ "vocab_size": 11767
233
  }
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:de052b3699dac9f3160ff7e6407f0cdf27a7fa35cc466f4af8d91becdc6dd508
3
- size 4499652
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:39f8829ef5bc935550c23bdf18940e7a5efebf4156664f7a512cd225a6410880
3
+ size 15563716
tokenizer.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:91a3b819f2a4f8d82698577c606d5b7f832f808f48748d63c522e41a0802810c
3
- size 14048
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fee59a56fb6cbbe664361aaf51d6a4e518c78a6af7beeac249f21b01fba4ab2a
3
+ size 205516