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