| { | |
| "_id2label": { | |
| "intent": [ | |
| "atis_abbreviation", | |
| "atis_airport", | |
| "atis_city", | |
| "atis_capacity", | |
| "atis_aircraft", | |
| "atis_ground_service", | |
| "atis_meal", | |
| "atis_quantity", | |
| "atis_flight_no", | |
| "atis_flight_time", | |
| "atis_ground_fare", | |
| "atis_cheapest", | |
| "atis_flight", | |
| "atis_distance", | |
| "atis_restriction", | |
| "atis_airfare", | |
| "atis_airline" | |
| ], | |
| "slot": [ | |
| "O", | |
| "B-airline_code", | |
| "B-airline_name", | |
| "I-airline_name", | |
| "B-aircraft_code", | |
| "B-city_name", | |
| "I-city_name", | |
| "B-meal", | |
| "B-flight_number", | |
| "B-fromloc.city_name", | |
| "I-fromloc.city_name", | |
| "B-toloc.city_name", | |
| "I-toloc.city_name", | |
| "B-depart_time.time_relative", | |
| "B-depart_time.time", | |
| "I-depart_time.time", | |
| "B-depart_time.period_of_day", | |
| "B-class_type", | |
| "I-class_type", | |
| "B-flight_time", | |
| "B-cost_relative", | |
| "I-cost_relative", | |
| "B-depart_date.day_name", | |
| "B-toloc.airport_name", | |
| "I-toloc.airport_name", | |
| "B-transport_type", | |
| "I-transport_type", | |
| "B-arrive_time.time_relative", | |
| "B-arrive_time.time", | |
| "I-arrive_time.time", | |
| "B-arrive_date.day_name", | |
| "B-or", | |
| "B-economy", | |
| "I-economy", | |
| "B-depart_date.month_name", | |
| "B-depart_date.day_number", | |
| "B-fromloc.airport_name", | |
| "I-fromloc.airport_name", | |
| "B-airport_name", | |
| "I-airport_name", | |
| "B-fare_amount", | |
| "I-fare_amount", | |
| "B-round_trip", | |
| "I-round_trip", | |
| "B-fare_basis_code", | |
| "B-toloc.state_code", | |
| "B-restriction_code", | |
| "B-stoploc.city_name", | |
| "I-stoploc.city_name", | |
| "I-depart_time.period_of_day", | |
| "B-fromloc.state_name", | |
| "B-toloc.state_name", | |
| "B-flight_mod", | |
| "B-toloc.airport_code", | |
| "B-mod", | |
| "B-depart_date.year", | |
| "I-flight_time", | |
| "B-airport_code", | |
| "B-fromloc.airport_code", | |
| "B-arrive_date.month_name", | |
| "B-arrive_date.day_number", | |
| "B-depart_date.date_relative", | |
| "I-depart_date.day_number", | |
| "B-state_code", | |
| "B-flight_stop", | |
| "B-depart_time.start_time", | |
| "B-depart_time.end_time", | |
| "B-depart_date.today_relative", | |
| "B-day_name", | |
| "B-month_name", | |
| "B-day_number", | |
| "B-meal_description", | |
| "I-meal_description", | |
| "I-restriction_code", | |
| "B-fromloc.state_code", | |
| "I-arrive_date.day_number", | |
| "B-arrive_time.start_time", | |
| "I-arrive_time.start_time", | |
| "B-arrive_time.end_time", | |
| "I-arrive_time.end_time", | |
| "B-meal_code", | |
| "I-flight_mod", | |
| "B-days_code", | |
| "B-today_relative", | |
| "I-today_relative", | |
| "B-time", | |
| "B-flight_days", | |
| "B-connect", | |
| "B-period_of_day", | |
| "I-meal_code", | |
| "B-time_relative", | |
| "I-time", | |
| "B-arrive_time.period_of_day", | |
| "B-return_date.month_name", | |
| "B-return_date.day_number", | |
| "I-toloc.state_name", | |
| "B-state_name", | |
| "B-depart_time.period_mod", | |
| "I-arrive_time.period_of_day", | |
| "I-flight_stop", | |
| "I-depart_date.today_relative", | |
| "I-fromloc.state_name", | |
| "I-depart_time.end_time", | |
| "B-arrive_date.date_relative", | |
| "B-return_date.date_relative", | |
| "I-depart_time.start_time", | |
| "B-stoploc.state_code", | |
| "I-depart_time.time_relative", | |
| "B-stoploc.airport_name", | |
| "B-arrive_time.period_mod", | |
| "B-toloc.country_name" | |
| ] | |
| }, | |
| "_label2id": { | |
| "intent": { | |
| "atis_abbreviation": 0, | |
| "atis_aircraft": 4, | |
| "atis_airfare": 15, | |
| "atis_airline": 16, | |
| "atis_airport": 1, | |
| "atis_capacity": 3, | |
| "atis_cheapest": 11, | |
| "atis_city": 2, | |
| "atis_distance": 13, | |
| "atis_flight": 12, | |
| "atis_flight_no": 8, | |
| "atis_flight_time": 9, | |
| "atis_ground_fare": 10, | |
| "atis_ground_service": 5, | |
| "atis_meal": 6, | |
| "atis_quantity": 7, | |
| "atis_restriction": 14 | |
| }, | |
| "slot": { | |
| "B-aircraft_code": 4, | |
| "B-airline_code": 1, | |
| "B-airline_name": 2, | |
| "B-airport_code": 57, | |
| "B-airport_name": 38, | |
| "B-arrive_date.date_relative": 103, | |
| "B-arrive_date.day_name": 30, | |
| "B-arrive_date.day_number": 60, | |
| "B-arrive_date.month_name": 59, | |
| "B-arrive_time.end_time": 78, | |
| "B-arrive_time.period_mod": 109, | |
| "B-arrive_time.period_of_day": 92, | |
| "B-arrive_time.start_time": 76, | |
| "B-arrive_time.time": 28, | |
| "B-arrive_time.time_relative": 27, | |
| "B-city_name": 5, | |
| "B-class_type": 17, | |
| "B-connect": 87, | |
| "B-cost_relative": 20, | |
| "B-day_name": 68, | |
| "B-day_number": 70, | |
| "B-days_code": 82, | |
| "B-depart_date.date_relative": 61, | |
| "B-depart_date.day_name": 22, | |
| "B-depart_date.day_number": 35, | |
| "B-depart_date.month_name": 34, | |
| "B-depart_date.today_relative": 67, | |
| "B-depart_date.year": 55, | |
| "B-depart_time.end_time": 66, | |
| "B-depart_time.period_mod": 97, | |
| "B-depart_time.period_of_day": 16, | |
| "B-depart_time.start_time": 65, | |
| "B-depart_time.time": 14, | |
| "B-depart_time.time_relative": 13, | |
| "B-economy": 32, | |
| "B-fare_amount": 40, | |
| "B-fare_basis_code": 44, | |
| "B-flight_days": 86, | |
| "B-flight_mod": 52, | |
| "B-flight_number": 8, | |
| "B-flight_stop": 64, | |
| "B-flight_time": 19, | |
| "B-fromloc.airport_code": 58, | |
| "B-fromloc.airport_name": 36, | |
| "B-fromloc.city_name": 9, | |
| "B-fromloc.state_code": 74, | |
| "B-fromloc.state_name": 50, | |
| "B-meal": 7, | |
| "B-meal_code": 80, | |
| "B-meal_description": 71, | |
| "B-mod": 54, | |
| "B-month_name": 69, | |
| "B-or": 31, | |
| "B-period_of_day": 88, | |
| "B-restriction_code": 46, | |
| "B-return_date.date_relative": 104, | |
| "B-return_date.day_number": 94, | |
| "B-return_date.month_name": 93, | |
| "B-round_trip": 42, | |
| "B-state_code": 63, | |
| "B-state_name": 96, | |
| "B-stoploc.airport_name": 108, | |
| "B-stoploc.city_name": 47, | |
| "B-stoploc.state_code": 106, | |
| "B-time": 85, | |
| "B-time_relative": 90, | |
| "B-today_relative": 83, | |
| "B-toloc.airport_code": 53, | |
| "B-toloc.airport_name": 23, | |
| "B-toloc.city_name": 11, | |
| "B-toloc.country_name": 110, | |
| "B-toloc.state_code": 45, | |
| "B-toloc.state_name": 51, | |
| "B-transport_type": 25, | |
| "I-airline_name": 3, | |
| "I-airport_name": 39, | |
| "I-arrive_date.day_number": 75, | |
| "I-arrive_time.end_time": 79, | |
| "I-arrive_time.period_of_day": 98, | |
| "I-arrive_time.start_time": 77, | |
| "I-arrive_time.time": 29, | |
| "I-city_name": 6, | |
| "I-class_type": 18, | |
| "I-cost_relative": 21, | |
| "I-depart_date.day_number": 62, | |
| "I-depart_date.today_relative": 100, | |
| "I-depart_time.end_time": 102, | |
| "I-depart_time.period_of_day": 49, | |
| "I-depart_time.start_time": 105, | |
| "I-depart_time.time": 15, | |
| "I-depart_time.time_relative": 107, | |
| "I-economy": 33, | |
| "I-fare_amount": 41, | |
| "I-flight_mod": 81, | |
| "I-flight_stop": 99, | |
| "I-flight_time": 56, | |
| "I-fromloc.airport_name": 37, | |
| "I-fromloc.city_name": 10, | |
| "I-fromloc.state_name": 101, | |
| "I-meal_code": 89, | |
| "I-meal_description": 72, | |
| "I-restriction_code": 73, | |
| "I-round_trip": 43, | |
| "I-stoploc.city_name": 48, | |
| "I-time": 91, | |
| "I-today_relative": 84, | |
| "I-toloc.airport_name": 24, | |
| "I-toloc.city_name": 12, | |
| "I-toloc.state_name": 95, | |
| "I-transport_type": 26, | |
| "O": 0 | |
| } | |
| }, | |
| "_name_or_path": "OpenSLUv1", | |
| "_num_labels": { | |
| "intent": 17, | |
| "slot": 111 | |
| }, | |
| "architectures": [ | |
| "PretrainedModelForSLUToSave" | |
| ], | |
| "is_decoder": true, | |
| "model": { | |
| "_model_target_": "model.OpenSLUModel", | |
| "decoder": { | |
| "_model_target_": "model.decoder.GLGINDecoder", | |
| "dropout_rate": 0.4, | |
| "intent_classifier": { | |
| "_model_target_": "model.decoder.classifier.MLPClassifier", | |
| "dropout_rate": 0.4, | |
| "ignore_index": -100, | |
| "input_dim": 384, | |
| "intent_label_num": 17, | |
| "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": 17 | |
| } | |
| ], | |
| "mode": "token-level-intent", | |
| "multi_threshold": 0.5, | |
| "return_sentence_level": true, | |
| "use_intent": true, | |
| "use_multi": true | |
| }, | |
| "interaction": { | |
| "_model_target_": "model.decoder.interaction.GLGINInteraction", | |
| "alpha": 0.2, | |
| "dropout_rate": 0.4, | |
| "hidden_dim": 256, | |
| "input_dim": 384, | |
| "intent_embedding_dim": 64, | |
| "intent_label_num": 17, | |
| "num_heads": 8, | |
| "num_layers": 2, | |
| "output_dim": 64, | |
| "row_normalized": true, | |
| "slot_graph_window": 1, | |
| "slot_label_num": 111 | |
| }, | |
| "slot_classifier": { | |
| "_model_target_": "model.decoder.classifier.MLPClassifier", | |
| "dropout_rate": 0.4, | |
| "ignore_index": -100, | |
| "input_dim": 384, | |
| "mlp": [ | |
| { | |
| "_model_target_": "torch.nn.Linear", | |
| "in_features": 64, | |
| "out_features": 64 | |
| }, | |
| { | |
| "_model_target_": "torch.nn.LeakyReLU", | |
| "negative_slope": 0.2 | |
| }, | |
| { | |
| "_model_target_": "torch.nn.Linear", | |
| "in_features": 64, | |
| "out_features": 111 | |
| } | |
| ], | |
| "mode": "slot", | |
| "multi_threshold": 0.5, | |
| "return_sentence_level": false, | |
| "slot_label_num": 111, | |
| "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": 128, | |
| "vocab_size": 790 | |
| }, | |
| "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": 790 | |
| } | |