{ "_id2label": { "intent": [ "PlayMusic", "RateBook", "SearchCreativeWork", "SearchScreeningEvent", "AddToPlaylist", "BookRestaurant", "GetWeather" ], "slot": [ "O", "B-artist", "I-artist", "B-object_name", "I-object_name", "B-rating_value", "B-rating_unit", "B-sort", "B-music_item", "B-object_type", "I-object_type", "B-best_rating", "B-location_name", "I-location_name", "B-playlist", "I-playlist", "B-movie_name", "I-movie_name", "B-party_size_description", "I-party_size_description", "B-city", "B-restaurant_type", "B-playlist_owner", "B-party_size_number", "I-city", "B-condition_temperature", "B-timeRange", "I-timeRange", "B-country", "B-served_dish", "I-served_dish", "B-movie_type", "B-spatial_relation", "I-spatial_relation", "B-restaurant_name", "I-restaurant_name", "B-entity_name", "I-entity_name", "B-service", "B-cuisine", "B-poi", "I-poi", "I-service", "B-state", "B-track", "I-track", "B-object_select", "B-object_part_of_series_type", "I-restaurant_type", "I-playlist_owner", "B-year", "I-sort", "B-geographic_poi", "I-geographic_poi", "B-genre", "B-object_location_type", "I-object_location_type", "B-current_location", "I-state", "B-condition_description", "I-current_location", "I-country", "I-genre", "I-movie_type", "B-facility", "B-album", "I-cuisine", "I-album", "I-facility", "I-object_part_of_series_type", "I-music_item", "I-object_select" ] }, "_label2id": { "intent": { "AddToPlaylist": 4, "BookRestaurant": 5, "GetWeather": 6, "PlayMusic": 0, "RateBook": 1, "SearchCreativeWork": 2, "SearchScreeningEvent": 3 }, "slot": { "B-album": 65, "B-artist": 1, "B-best_rating": 11, "B-city": 20, "B-condition_description": 59, "B-condition_temperature": 25, "B-country": 28, "B-cuisine": 39, "B-current_location": 57, "B-entity_name": 36, "B-facility": 64, "B-genre": 54, "B-geographic_poi": 52, "B-location_name": 12, "B-movie_name": 16, "B-movie_type": 31, "B-music_item": 8, "B-object_location_type": 55, "B-object_name": 3, "B-object_part_of_series_type": 47, "B-object_select": 46, "B-object_type": 9, "B-party_size_description": 18, "B-party_size_number": 23, "B-playlist": 14, "B-playlist_owner": 22, "B-poi": 40, "B-rating_unit": 6, "B-rating_value": 5, "B-restaurant_name": 34, "B-restaurant_type": 21, "B-served_dish": 29, "B-service": 38, "B-sort": 7, "B-spatial_relation": 32, "B-state": 43, "B-timeRange": 26, "B-track": 44, "B-year": 50, "I-album": 67, "I-artist": 2, "I-city": 24, "I-country": 61, "I-cuisine": 66, "I-current_location": 60, "I-entity_name": 37, "I-facility": 68, "I-genre": 62, "I-geographic_poi": 53, "I-location_name": 13, "I-movie_name": 17, "I-movie_type": 63, "I-music_item": 70, "I-object_location_type": 56, "I-object_name": 4, "I-object_part_of_series_type": 69, "I-object_select": 71, "I-object_type": 10, "I-party_size_description": 19, "I-playlist": 15, "I-playlist_owner": 49, "I-poi": 41, "I-restaurant_name": 35, "I-restaurant_type": 48, "I-served_dish": 30, "I-service": 42, "I-sort": 51, "I-spatial_relation": 33, "I-state": 58, "I-timeRange": 27, "I-track": 45, "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.AGIFDecoder", "intent_classifier": { "_model_target_": "model.decoder.classifier.MLPClassifier", "dropout_rate": 0.4, "ignore_index": -100, "input_dim": 384, "intent_label_num": 7, "loss_fn": { "_model_target_": "torch.nn.BCEWithLogitsLoss" }, "mlp": [ { "_model_target_": "torch.nn.Linear", "in_features": 384, "out_features": 256 }, { "_model_target_": "torch.nn.LeakyReLU", "negative_slope": 0.2 }, { "_model_target_": "torch.nn.Linear", "in_features": 256, "out_features": 7 } ], "mode": "intent", "multi_threshold": 0.5, "return_sentence_level": true, "use_intent": true, "use_multi": true, "weight": 0.3 }, "interaction": { "_model_target_": "model.decoder.interaction.AGIFInteraction", "alpha": 0.2, "dropout_rate": 0.4, "hidden_dim": 128, "input_dim": 384, "intent_embedding_dim": 128, "intent_label_num": 7, "num_heads": 4, "num_layers": 2, "output_dim": 128, "row_normalized": true, "slot_label_num": 72 }, "slot_classifier": { "_model_target_": "model.decoder.classifier.AutoregressiveLSTMClassifier", "bidirectional": false, "dropout_rate": 0.4, "embedding_dim": 128, "force_ratio": 0.9, "hidden_dim": 128, "ignore_index": -100, "input_dim": 384, "layer_num": 1, "mode": "slot", "multi_threshold": 0.5, "return_sentence_level": false, "slot_label_num": 72, "use_multi": false, "use_slot": true, "weight": 0.7 } }, "encoder": { "_model_target_": "model.encoder.AutoEncoder", "attention": { "dropout_rate": 0.4, "hidden_dim": 1024, "output_dim": 128 }, "embedding": { "dropout_rate": 0.4, "embedding_dim": 128, "vocab_size": 11758 }, "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": true, "return_with_input": true, "unflat_attention": { "dropout_rate": 0.4 } } }, "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": 11758 }