LightChen2333 commited on
Commit
a030b65
1 Parent(s): ea78836

Upload 3 files

Browse files
Files changed (3) hide show
  1. config.json +216 -0
  2. pytorch_model.bin +3 -0
  3. tokenizer.pkl +3 -0
config.json ADDED
@@ -0,0 +1,216 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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.open_slu_model.OpenSLUModel",
183
+ "decoder": {
184
+ "_model_target_": "model.decoder.base_decoder.BaseDecoder",
185
+ "intent_classifier": {
186
+ "_model_target_": "model.decoder.classifier.LinearClassifier",
187
+ "ignore_index": -100,
188
+ "input_dim": 768,
189
+ "intent_label_num": 7,
190
+ "mode": "intent",
191
+ "use_intent": true
192
+ },
193
+ "slot_classifier": {
194
+ "_model_target_": "model.decoder.classifier.LinearClassifier",
195
+ "ignore_index": -100,
196
+ "input_dim": 768,
197
+ "mode": "slot",
198
+ "slot_label_num": 72,
199
+ "use_slot": true
200
+ }
201
+ },
202
+ "encoder": {
203
+ "_model_target_": "model.encoder.AutoEncoder",
204
+ "encoder_name": "roberta-base",
205
+ "output_dim": 768,
206
+ "return_sentence_level_hidden": true,
207
+ "return_with_input": true
208
+ },
209
+ "ignore_index": -100
210
+ },
211
+ "return_dict": false,
212
+ "tokenizer_class": "OpenSLUv1",
213
+ "torch_dtype": "float32",
214
+ "transformers_version": "4.25.1",
215
+ "vocab_size": 50265
216
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:484cdc70b1991324b204e4fa0ab9972b9baa49e35c9d438cf1dceb1b7f5539e6
3
+ size 498905129
tokenizer.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5b322f670b44bcbd2fc5993385820f03bd5fde4cfe4fb36dd4ba803629736362
3
+ size 1357087