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