| { | |
| "_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.BaseDecoder", | |
| "intent_classifier": { | |
| "_model_target_": "model.decoder.classifier.LinearClassifier", | |
| "ignore_index": -100, | |
| "input_dim": 256, | |
| "intent_label_num": 7, | |
| "mode": "intent", | |
| "use_intent": true | |
| }, | |
| "interaction": { | |
| "_model_target_": "model.decoder.interaction.BiModelInteraction", | |
| "dropout_rate": 0.5, | |
| "input_dim": 512, | |
| "intent_label_num": 7, | |
| "output_dim": 256, | |
| "slot_label_num": 72 | |
| }, | |
| "slot_classifier": { | |
| "_model_target_": "model.decoder.classifier.LinearClassifier", | |
| "ignore_index": -100, | |
| "input_dim": 256, | |
| "mode": "slot", | |
| "slot_label_num": 72, | |
| "use_slot": true | |
| } | |
| }, | |
| "encoder": { | |
| "_model_target_": "model.encoder.BiEncoder", | |
| "intent_encoder": { | |
| "_model_target_": "model.encoder.AutoEncoder", | |
| "embedding": { | |
| "dropout_rate": 0.5, | |
| "embedding_dim": 256, | |
| "vocab_size": 11767 | |
| }, | |
| "encoder_name": "lstm", | |
| "lstm": { | |
| "bidirectional": true, | |
| "dropout_rate": 0.5, | |
| "layer_num": 2, | |
| "output_dim": 256 | |
| }, | |
| "return_sentence_level_hidden": false, | |
| "return_with_input": true | |
| }, | |
| "output_dim": 512, | |
| "slot_encoder": { | |
| "_model_target_": "model.encoder.AutoEncoder", | |
| "embedding": { | |
| "dropout_rate": 0.5, | |
| "embedding_dim": 256, | |
| "vocab_size": 11767 | |
| }, | |
| "encoder_name": "lstm", | |
| "lstm": { | |
| "bidirectional": true, | |
| "dropout_rate": 0.5, | |
| "layer_num": 2, | |
| "output_dim": 256 | |
| }, | |
| "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 | |
| } | |