{ "_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.AGIFDecoder", "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": "{model.encoder.output_dim}", "out_features": 256 }, { "_model_target_": "torch.nn.LeakyReLU", "negative_slope": 0.2 }, { "_model_target_": "torch.nn.Linear", "in_features": 256, "out_features": "{base.intent_label_num}" } ], "mode": "intent", "multi_threshold": 0.5, "return_sentence_level": true, "use_intent": true, "use_multi": "{base.multi_intent}", "weight": 0.3 }, "interaction": { "_model_target_": "model.decoder.interaction.AGIFInteraction", "alpha": 0.2, "dropout_rate": 0.4, "hidden_dim": 128, "input_dim": "{model.encoder.output_dim}", "intent_embedding_dim": 128, "num_heads": 4, "num_layers": 2, "output_dim": "{model.decoder.interaction.intent_embedding_dim}", "row_normalized": true }, "slot_classifier": { "_model_target_": "model.decoder.classifier.AutoregressiveLSTMClassifier", "bidirectional": false, "dropout_rate": 0.4, "embedding_dim": 128, "force_ratio": 0.9, "hidden_dim": "{model.decoder.interaction.intent_embedding_dim}", "ignore_index": -100, "input_dim": "{model.encoder.output_dim}", "layer_num": 1, "mode": "slot", "multi_threshold": 0.5, "return_sentence_level": false, "slot_label_num": 111, "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 }, "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_class": "OpenSLUv1", "torch_dtype": "float32", "transformers_version": "4.25.1", "vocab_size": 790 }