|
{ |
|
"_id2label": { |
|
"intent": [ |
|
"PlayMusic", |
|
"AddToPlaylist", |
|
"RateBook", |
|
"SearchScreeningEvent", |
|
"BookRestaurant", |
|
"GetWeather", |
|
"SearchCreativeWork" |
|
], |
|
"slot": [ |
|
"O", |
|
"B-artist", |
|
"B-album", |
|
"B-service", |
|
"I-service", |
|
"B-entity_name", |
|
"I-entity_name", |
|
"B-playlist", |
|
"I-playlist", |
|
"B-object_select", |
|
"B-object_type", |
|
"B-rating_value", |
|
"B-best_rating", |
|
"B-music_item", |
|
"B-track", |
|
"I-track", |
|
"I-artist", |
|
"B-playlist_owner", |
|
"B-year", |
|
"B-sort", |
|
"B-movie_name", |
|
"I-movie_name", |
|
"B-party_size_number", |
|
"B-state", |
|
"B-city", |
|
"B-timeRange", |
|
"I-timeRange", |
|
"B-object_part_of_series_type", |
|
"I-object_type", |
|
"B-movie_type", |
|
"B-spatial_relation", |
|
"I-spatial_relation", |
|
"B-geographic_poi", |
|
"I-geographic_poi", |
|
"B-restaurant_type", |
|
"I-city", |
|
"B-party_size_description", |
|
"I-party_size_description", |
|
"B-object_location_type", |
|
"I-object_location_type", |
|
"B-object_name", |
|
"I-object_name", |
|
"I-movie_type", |
|
"B-rating_unit", |
|
"I-sort", |
|
"B-location_name", |
|
"I-location_name", |
|
"B-current_location", |
|
"I-current_location", |
|
"I-playlist_owner", |
|
"B-served_dish", |
|
"B-country", |
|
"B-condition_temperature", |
|
"B-poi", |
|
"I-poi", |
|
"B-condition_description", |
|
"B-genre", |
|
"B-restaurant_name", |
|
"I-restaurant_name", |
|
"I-state", |
|
"I-served_dish", |
|
"B-cuisine", |
|
"I-album", |
|
"I-country", |
|
"B-facility", |
|
"I-facility", |
|
"I-cuisine", |
|
"I-music_item", |
|
"I-object_select", |
|
"I-restaurant_type", |
|
"I-object_part_of_series_type", |
|
"I-genre" |
|
] |
|
}, |
|
"_label2id": { |
|
"intent": { |
|
"AddToPlaylist": 1, |
|
"BookRestaurant": 4, |
|
"GetWeather": 5, |
|
"PlayMusic": 0, |
|
"RateBook": 2, |
|
"SearchCreativeWork": 6, |
|
"SearchScreeningEvent": 3 |
|
}, |
|
"slot": { |
|
"B-album": 2, |
|
"B-artist": 1, |
|
"B-best_rating": 12, |
|
"B-city": 24, |
|
"B-condition_description": 55, |
|
"B-condition_temperature": 52, |
|
"B-country": 51, |
|
"B-cuisine": 61, |
|
"B-current_location": 47, |
|
"B-entity_name": 5, |
|
"B-facility": 64, |
|
"B-genre": 56, |
|
"B-geographic_poi": 32, |
|
"B-location_name": 45, |
|
"B-movie_name": 20, |
|
"B-movie_type": 29, |
|
"B-music_item": 13, |
|
"B-object_location_type": 38, |
|
"B-object_name": 40, |
|
"B-object_part_of_series_type": 27, |
|
"B-object_select": 9, |
|
"B-object_type": 10, |
|
"B-party_size_description": 36, |
|
"B-party_size_number": 22, |
|
"B-playlist": 7, |
|
"B-playlist_owner": 17, |
|
"B-poi": 53, |
|
"B-rating_unit": 43, |
|
"B-rating_value": 11, |
|
"B-restaurant_name": 57, |
|
"B-restaurant_type": 34, |
|
"B-served_dish": 50, |
|
"B-service": 3, |
|
"B-sort": 19, |
|
"B-spatial_relation": 30, |
|
"B-state": 23, |
|
"B-timeRange": 25, |
|
"B-track": 14, |
|
"B-year": 18, |
|
"I-album": 62, |
|
"I-artist": 16, |
|
"I-city": 35, |
|
"I-country": 63, |
|
"I-cuisine": 66, |
|
"I-current_location": 48, |
|
"I-entity_name": 6, |
|
"I-facility": 65, |
|
"I-genre": 71, |
|
"I-geographic_poi": 33, |
|
"I-location_name": 46, |
|
"I-movie_name": 21, |
|
"I-movie_type": 42, |
|
"I-music_item": 67, |
|
"I-object_location_type": 39, |
|
"I-object_name": 41, |
|
"I-object_part_of_series_type": 70, |
|
"I-object_select": 68, |
|
"I-object_type": 28, |
|
"I-party_size_description": 37, |
|
"I-playlist": 8, |
|
"I-playlist_owner": 49, |
|
"I-poi": 54, |
|
"I-restaurant_name": 58, |
|
"I-restaurant_type": 69, |
|
"I-served_dish": 60, |
|
"I-service": 4, |
|
"I-sort": 44, |
|
"I-spatial_relation": 31, |
|
"I-state": 59, |
|
"I-timeRange": 26, |
|
"I-track": 15, |
|
"O": 0 |
|
} |
|
}, |
|
"_name_or_path": "OpenSLUv1", |
|
"_num_labels": { |
|
"intent": 7, |
|
"slot": 72 |
|
}, |
|
"architectures": [ |
|
"PretrainedModelForSLUToSave" |
|
], |
|
"is_decoder": true, |
|
"model": { |
|
"_model_target_": "model.OpenSLUModel", |
|
"decoder": { |
|
"_model_target_": "model.decoder.StackPropagationDecoder", |
|
"intent_classifier": { |
|
"_model_target_": "model.decoder.classifier.AutoregressiveLSTMClassifier", |
|
"bidirectional": false, |
|
"dropout_rate": 0.4, |
|
"embedding_dim": 8, |
|
"force_ratio": 0.9, |
|
"hidden_dim": 64, |
|
"ignore_index": -100, |
|
"input_dim": 384, |
|
"intent_label_num": 7, |
|
"layer_num": 1, |
|
"mode": "token-level-intent", |
|
"return_sentence_level": true, |
|
"use_intent": true, |
|
"use_multi": false |
|
}, |
|
"interaction": { |
|
"_model_target_": "model.decoder.interaction.StackInteraction", |
|
"differentiable": false, |
|
"input_dim": 384, |
|
"intent_label_num": 7, |
|
"slot_label_num": 72 |
|
}, |
|
"slot_classifier": { |
|
"_model_target_": "model.decoder.classifier.AutoregressiveLSTMClassifier", |
|
"bidirectional": false, |
|
"dropout_rate": 0.4, |
|
"embedding_dim": 32, |
|
"force_ratio": 0.9, |
|
"hidden_dim": 64, |
|
"ignore_index": -100, |
|
"input_dim": 391, |
|
"layer_num": 1, |
|
"mode": "slot", |
|
"return_sentence_level": false, |
|
"slot_label_num": 72, |
|
"use_multi": false, |
|
"use_slot": true |
|
} |
|
}, |
|
"encoder": { |
|
"_model_target_": "model.encoder.AutoEncoder", |
|
"attention": { |
|
"dropout_rate": 0.4, |
|
"hidden_dim": 1024, |
|
"output_dim": 128 |
|
}, |
|
"embedding": { |
|
"dropout_rate": 0.4, |
|
"embedding_dim": 256, |
|
"vocab_size": 11767 |
|
}, |
|
"encoder_name": "self-attention-lstm", |
|
"lstm": { |
|
"bidirectional": true, |
|
"dropout_rate": 0.4, |
|
"layer_num": 1, |
|
"output_dim": 256 |
|
}, |
|
"output_dim": 384, |
|
"return_sentence_level_hidden": false, |
|
"return_with_input": true |
|
} |
|
}, |
|
"return_dict": false, |
|
"tokenizer": { |
|
"_align_mode_": "fast", |
|
"_padding_side_": "right", |
|
"_tokenizer_name_": "word_tokenizer", |
|
"add_special_tokens": false, |
|
"max_length": 512 |
|
}, |
|
"tokenizer_class": "OpenSLUv1", |
|
"torch_dtype": "float32", |
|
"transformers_version": "4.25.1", |
|
"vocab_size": 11767 |
|
} |
|
|