bi-model-slu-atis / config.json
LightChen2333's picture
Upload 3 files
7c45c74
{
"_id2label": {
"intent": [
"atis_flight",
"atis_airfare",
"atis_airline",
"atis_ground_service",
"atis_quantity",
"atis_city",
"atis_flight#atis_airfare",
"atis_abbreviation",
"atis_aircraft",
"atis_distance",
"atis_ground_fare",
"atis_capacity",
"atis_flight_time",
"atis_meal",
"atis_aircraft#atis_flight#atis_flight_no",
"atis_flight_no",
"atis_restriction",
"atis_airport",
"atis_airline#atis_flight_no",
"atis_cheapest",
"atis_ground_service#atis_ground_fare"
],
"slot": [
"O",
"B-fromloc.city_name",
"B-toloc.city_name",
"B-round_trip",
"I-round_trip",
"B-cost_relative",
"B-fare_amount",
"I-fare_amount",
"B-arrive_date.month_name",
"B-arrive_date.day_number",
"I-fromloc.city_name",
"B-stoploc.city_name",
"B-arrive_time.time_relative",
"B-arrive_time.time",
"I-arrive_time.time",
"B-toloc.state_code",
"I-toloc.city_name",
"I-stoploc.city_name",
"B-meal_description",
"B-depart_date.month_name",
"B-depart_date.day_number",
"B-airline_name",
"I-airline_name",
"B-depart_time.period_of_day",
"B-depart_date.day_name",
"B-toloc.state_name",
"B-depart_time.time_relative",
"B-depart_time.time",
"B-toloc.airport_name",
"I-toloc.airport_name",
"B-depart_date.date_relative",
"B-or",
"B-airline_code",
"B-class_type",
"I-class_type",
"I-cost_relative",
"I-depart_time.time",
"B-fromloc.airport_name",
"I-fromloc.airport_name",
"B-city_name",
"B-flight_mod",
"B-meal",
"B-economy",
"B-fare_basis_code",
"I-depart_date.day_number",
"B-depart_date.today_relative",
"B-flight_stop",
"B-airport_code",
"B-fromloc.state_name",
"I-fromloc.state_name",
"I-city_name",
"B-connect",
"B-arrive_date.day_name",
"B-fromloc.state_code",
"B-arrive_date.today_relative",
"B-depart_date.year",
"B-depart_time.start_time",
"I-depart_time.start_time",
"B-depart_time.end_time",
"I-depart_time.end_time",
"B-arrive_time.start_time",
"B-arrive_time.end_time",
"I-arrive_time.end_time",
"I-flight_mod",
"B-flight_days",
"B-mod",
"B-flight_number",
"I-toloc.state_name",
"B-meal_code",
"I-meal_code",
"B-airport_name",
"I-airport_name",
"I-flight_stop",
"B-transport_type",
"I-transport_type",
"B-state_code",
"B-aircraft_code",
"B-toloc.country_name",
"I-arrive_date.day_number",
"B-toloc.airport_code",
"B-return_date.date_relative",
"I-return_date.date_relative",
"B-flight_time",
"I-economy",
"B-fromloc.airport_code",
"B-arrive_time.period_of_day",
"B-depart_time.period_mod",
"I-flight_time",
"B-return_date.day_name",
"B-arrive_date.date_relative",
"B-restriction_code",
"I-restriction_code",
"B-arrive_time.period_mod",
"I-arrive_time.period_of_day",
"B-period_of_day",
"B-stoploc.state_code",
"I-depart_date.today_relative",
"I-fare_basis_code",
"I-arrive_time.start_time",
"B-time",
"B-today_relative",
"I-today_relative",
"B-state_name",
"B-days_code",
"I-depart_time.period_of_day",
"I-arrive_time.time_relative",
"B-time_relative",
"I-time",
"B-return_date.month_name",
"B-return_date.day_number",
"I-depart_time.time_relative",
"B-stoploc.airport_name",
"B-day_name",
"B-month_name",
"B-day_number",
"B-return_time.period_mod",
"B-return_time.period_of_day",
"B-return_date.today_relative",
"I-return_date.today_relative",
"I-meal_description"
]
},
"_label2id": {
"intent": {
"atis_abbreviation": 7,
"atis_aircraft": 8,
"atis_aircraft#atis_flight#atis_flight_no": 14,
"atis_airfare": 1,
"atis_airline": 2,
"atis_airline#atis_flight_no": 18,
"atis_airport": 17,
"atis_capacity": 11,
"atis_cheapest": 19,
"atis_city": 5,
"atis_distance": 9,
"atis_flight": 0,
"atis_flight#atis_airfare": 6,
"atis_flight_no": 15,
"atis_flight_time": 12,
"atis_ground_fare": 10,
"atis_ground_service": 3,
"atis_ground_service#atis_ground_fare": 20,
"atis_meal": 13,
"atis_quantity": 4,
"atis_restriction": 16
},
"slot": {
"B-aircraft_code": 76,
"B-airline_code": 32,
"B-airline_name": 21,
"B-airport_code": 47,
"B-airport_name": 70,
"B-arrive_date.date_relative": 89,
"B-arrive_date.day_name": 52,
"B-arrive_date.day_number": 9,
"B-arrive_date.month_name": 8,
"B-arrive_date.today_relative": 54,
"B-arrive_time.end_time": 61,
"B-arrive_time.period_mod": 92,
"B-arrive_time.period_of_day": 85,
"B-arrive_time.start_time": 60,
"B-arrive_time.time": 13,
"B-arrive_time.time_relative": 12,
"B-city_name": 39,
"B-class_type": 33,
"B-connect": 51,
"B-cost_relative": 5,
"B-day_name": 112,
"B-day_number": 114,
"B-days_code": 103,
"B-depart_date.date_relative": 30,
"B-depart_date.day_name": 24,
"B-depart_date.day_number": 20,
"B-depart_date.month_name": 19,
"B-depart_date.today_relative": 45,
"B-depart_date.year": 55,
"B-depart_time.end_time": 58,
"B-depart_time.period_mod": 86,
"B-depart_time.period_of_day": 23,
"B-depart_time.start_time": 56,
"B-depart_time.time": 27,
"B-depart_time.time_relative": 26,
"B-economy": 42,
"B-fare_amount": 6,
"B-fare_basis_code": 43,
"B-flight_days": 64,
"B-flight_mod": 40,
"B-flight_number": 66,
"B-flight_stop": 46,
"B-flight_time": 82,
"B-fromloc.airport_code": 84,
"B-fromloc.airport_name": 37,
"B-fromloc.city_name": 1,
"B-fromloc.state_code": 53,
"B-fromloc.state_name": 48,
"B-meal": 41,
"B-meal_code": 68,
"B-meal_description": 18,
"B-mod": 65,
"B-month_name": 113,
"B-or": 31,
"B-period_of_day": 94,
"B-restriction_code": 90,
"B-return_date.date_relative": 80,
"B-return_date.day_name": 88,
"B-return_date.day_number": 109,
"B-return_date.month_name": 108,
"B-return_date.today_relative": 117,
"B-return_time.period_mod": 115,
"B-return_time.period_of_day": 116,
"B-round_trip": 3,
"B-state_code": 75,
"B-state_name": 102,
"B-stoploc.airport_name": 111,
"B-stoploc.city_name": 11,
"B-stoploc.state_code": 95,
"B-time": 99,
"B-time_relative": 106,
"B-today_relative": 100,
"B-toloc.airport_code": 79,
"B-toloc.airport_name": 28,
"B-toloc.city_name": 2,
"B-toloc.country_name": 77,
"B-toloc.state_code": 15,
"B-toloc.state_name": 25,
"B-transport_type": 73,
"I-airline_name": 22,
"I-airport_name": 71,
"I-arrive_date.day_number": 78,
"I-arrive_time.end_time": 62,
"I-arrive_time.period_of_day": 93,
"I-arrive_time.start_time": 98,
"I-arrive_time.time": 14,
"I-arrive_time.time_relative": 105,
"I-city_name": 50,
"I-class_type": 34,
"I-cost_relative": 35,
"I-depart_date.day_number": 44,
"I-depart_date.today_relative": 96,
"I-depart_time.end_time": 59,
"I-depart_time.period_of_day": 104,
"I-depart_time.start_time": 57,
"I-depart_time.time": 36,
"I-depart_time.time_relative": 110,
"I-economy": 83,
"I-fare_amount": 7,
"I-fare_basis_code": 97,
"I-flight_mod": 63,
"I-flight_stop": 72,
"I-flight_time": 87,
"I-fromloc.airport_name": 38,
"I-fromloc.city_name": 10,
"I-fromloc.state_name": 49,
"I-meal_code": 69,
"I-meal_description": 119,
"I-restriction_code": 91,
"I-return_date.date_relative": 81,
"I-return_date.today_relative": 118,
"I-round_trip": 4,
"I-stoploc.city_name": 17,
"I-time": 107,
"I-today_relative": 101,
"I-toloc.airport_name": 29,
"I-toloc.city_name": 16,
"I-toloc.state_name": 67,
"I-transport_type": 74,
"O": 0
}
},
"_name_or_path": "OpenSLUv1",
"_num_labels": {
"intent": 21,
"slot": 120
},
"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": 21,
"mode": "intent",
"use_intent": true
},
"interaction": {
"_model_target_": "model.decoder.interaction.BiModelInteraction",
"dropout_rate": 0.4,
"input_dim": 512,
"intent_label_num": 21,
"output_dim": 256,
"slot_label_num": 120
},
"slot_classifier": {
"_model_target_": "model.decoder.classifier.LinearClassifier",
"ignore_index": -100,
"input_dim": 256,
"mode": "slot",
"slot_label_num": 120,
"use_slot": true
}
},
"encoder": {
"_model_target_": "model.encoder.BiEncoder",
"intent_encoder": {
"_model_target_": "model.encoder.AutoEncoder",
"embedding": {
"dropout_rate": 0.4,
"embedding_dim": 256,
"vocab_size": 900
},
"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.4,
"embedding_dim": 256,
"vocab_size": 900
},
"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": 900
}