| { |
| "_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 |
| } |
|
|
