LightChen2333
commited on
Commit
•
03b8cc0
1
Parent(s):
185fc6a
Upload 3 files
Browse files- config.json +250 -0
- pytorch_model.bin +3 -0
- tokenizer.pkl +3 -0
config.json
ADDED
@@ -0,0 +1,250 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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.BaseDecoder",
|
185 |
+
"intent_classifier": {
|
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 |
+
},
|
193 |
+
"interaction": {
|
194 |
+
"_model_target_": "model.decoder.interaction.BiModelInteraction",
|
195 |
+
"dropout_rate": 0.5,
|
196 |
+
"input_dim": 512,
|
197 |
+
"output_dim": 256
|
198 |
+
},
|
199 |
+
"slot_classifier": {
|
200 |
+
"_model_target_": "model.decoder.classifier.LinearClassifier",
|
201 |
+
"ignore_index": -100,
|
202 |
+
"input_dim": 256,
|
203 |
+
"mode": "slot",
|
204 |
+
"slot_label_num": 72,
|
205 |
+
"use_slot": true
|
206 |
+
}
|
207 |
+
},
|
208 |
+
"encoder": {
|
209 |
+
"_model_target_": "model.encoder.BiEncoder",
|
210 |
+
"intent_encoder": {
|
211 |
+
"_model_target_": "model.encoder.AutoEncoder",
|
212 |
+
"embedding": {
|
213 |
+
"dropout_rate": 0.5,
|
214 |
+
"embedding_dim": 256
|
215 |
+
},
|
216 |
+
"encoder_name": "lstm",
|
217 |
+
"lstm": {
|
218 |
+
"bidirectional": true,
|
219 |
+
"dropout_rate": 0.5,
|
220 |
+
"layer_num": 2,
|
221 |
+
"output_dim": 256
|
222 |
+
},
|
223 |
+
"return_sentence_level_hidden": false,
|
224 |
+
"return_with_input": true
|
225 |
+
},
|
226 |
+
"output_dim": 512,
|
227 |
+
"slot_encoder": {
|
228 |
+
"_model_target_": "model.encoder.AutoEncoder",
|
229 |
+
"embedding": {
|
230 |
+
"dropout_rate": 0.5,
|
231 |
+
"embedding_dim": 256
|
232 |
+
},
|
233 |
+
"encoder_name": "lstm",
|
234 |
+
"lstm": {
|
235 |
+
"bidirectional": true,
|
236 |
+
"dropout_rate": 0.5,
|
237 |
+
"layer_num": 2,
|
238 |
+
"output_dim": 256
|
239 |
+
},
|
240 |
+
"return_sentence_level_hidden": false,
|
241 |
+
"return_with_input": true
|
242 |
+
}
|
243 |
+
}
|
244 |
+
},
|
245 |
+
"return_dict": false,
|
246 |
+
"tokenizer_class": "OpenSLUv1",
|
247 |
+
"torch_dtype": "float32",
|
248 |
+
"transformers_version": "4.25.1",
|
249 |
+
"vocab_size": 11767
|
250 |
+
}
|
pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:37f1b05c81be66eec377c68b537cf1d408b0b1c13cb85ec608a49fcf96cb670d
|
3 |
+
size 37870151
|
tokenizer.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:fee59a56fb6cbbe664361aaf51d6a4e518c78a6af7beeac249f21b01fba4ab2a
|
3 |
+
size 205516
|