CoolFace
Modelpublic

LightChen2333/stack-propagation-slu-atis

sourceHugging Facemitupdated 4y agoView on Hugging Face
0likes6downloads
config.json387 linesDownload Raw Back to root
1{2  "_id2label": {3    "intent": [4      "atis_flight",5      "atis_airfare",6      "atis_airline",7      "atis_ground_service",8      "atis_quantity",9      "atis_city",10      "atis_flight#atis_airfare",11      "atis_abbreviation",12      "atis_aircraft",13      "atis_distance",14      "atis_ground_fare",15      "atis_capacity",16      "atis_flight_time",17      "atis_meal",18      "atis_aircraft#atis_flight#atis_flight_no",19      "atis_flight_no",20      "atis_restriction",21      "atis_airport",22      "atis_airline#atis_flight_no",23      "atis_cheapest",24      "atis_ground_service#atis_ground_fare"25    ],26    "slot": [27      "O",28      "B-fromloc.city_name",29      "B-toloc.city_name",30      "B-round_trip",31      "I-round_trip",32      "B-cost_relative",33      "B-fare_amount",34      "I-fare_amount",35      "B-arrive_date.month_name",36      "B-arrive_date.day_number",37      "I-fromloc.city_name",38      "B-stoploc.city_name",39      "B-arrive_time.time_relative",40      "B-arrive_time.time",41      "I-arrive_time.time",42      "B-toloc.state_code",43      "I-toloc.city_name",44      "I-stoploc.city_name",45      "B-meal_description",46      "B-depart_date.month_name",47      "B-depart_date.day_number",48      "B-airline_name",49      "I-airline_name",50      "B-depart_time.period_of_day",51      "B-depart_date.day_name",52      "B-toloc.state_name",53      "B-depart_time.time_relative",54      "B-depart_time.time",55      "B-toloc.airport_name",56      "I-toloc.airport_name",57      "B-depart_date.date_relative",58      "B-or",59      "B-airline_code",60      "B-class_type",61      "I-class_type",62      "I-cost_relative",63      "I-depart_time.time",64      "B-fromloc.airport_name",65      "I-fromloc.airport_name",66      "B-city_name",67      "B-flight_mod",68      "B-meal",69      "B-economy",70      "B-fare_basis_code",71      "I-depart_date.day_number",72      "B-depart_date.today_relative",73      "B-flight_stop",74      "B-airport_code",75      "B-fromloc.state_name",76      "I-fromloc.state_name",77      "I-city_name",78      "B-connect",79      "B-arrive_date.day_name",80      "B-fromloc.state_code",81      "B-arrive_date.today_relative",82      "B-depart_date.year",83      "B-depart_time.start_time",84      "I-depart_time.start_time",85      "B-depart_time.end_time",86      "I-depart_time.end_time",87      "B-arrive_time.start_time",88      "B-arrive_time.end_time",89      "I-arrive_time.end_time",90      "I-flight_mod",91      "B-flight_days",92      "B-mod",93      "B-flight_number",94      "I-toloc.state_name",95      "B-meal_code",96      "I-meal_code",97      "B-airport_name",98      "I-airport_name",99      "I-flight_stop",100      "B-transport_type",101      "I-transport_type",102      "B-state_code",103      "B-aircraft_code",104      "B-toloc.country_name",105      "I-arrive_date.day_number",106      "B-toloc.airport_code",107      "B-return_date.date_relative",108      "I-return_date.date_relative",109      "B-flight_time",110      "I-economy",111      "B-fromloc.airport_code",112      "B-arrive_time.period_of_day",113      "B-depart_time.period_mod",114      "I-flight_time",115      "B-return_date.day_name",116      "B-arrive_date.date_relative",117      "B-restriction_code",118      "I-restriction_code",119      "B-arrive_time.period_mod",120      "I-arrive_time.period_of_day",121      "B-period_of_day",122      "B-stoploc.state_code",123      "I-depart_date.today_relative",124      "I-fare_basis_code",125      "I-arrive_time.start_time",126      "B-time",127      "B-today_relative",128      "I-today_relative",129      "B-state_name",130      "B-days_code",131      "I-depart_time.period_of_day",132      "I-arrive_time.time_relative",133      "B-time_relative",134      "I-time",135      "B-return_date.month_name",136      "B-return_date.day_number",137      "I-depart_time.time_relative",138      "B-stoploc.airport_name",139      "B-day_name",140      "B-month_name",141      "B-day_number",142      "B-return_time.period_mod",143      "B-return_time.period_of_day",144      "B-return_date.today_relative",145      "I-return_date.today_relative",146      "I-meal_description"147    ]148  },149  "_label2id": {150    "intent": {151      "atis_abbreviation": 7,152      "atis_aircraft": 8,153      "atis_aircraft#atis_flight#atis_flight_no": 14,154      "atis_airfare": 1,155      "atis_airline": 2,156      "atis_airline#atis_flight_no": 18,157      "atis_airport": 17,158      "atis_capacity": 11,159      "atis_cheapest": 19,160      "atis_city": 5,161      "atis_distance": 9,162      "atis_flight": 0,163      "atis_flight#atis_airfare": 6,164      "atis_flight_no": 15,165      "atis_flight_time": 12,166      "atis_ground_fare": 10,167      "atis_ground_service": 3,168      "atis_ground_service#atis_ground_fare": 20,169      "atis_meal": 13,170      "atis_quantity": 4,171      "atis_restriction": 16172    },173    "slot": {174      "B-aircraft_code": 76,175      "B-airline_code": 32,176      "B-airline_name": 21,177      "B-airport_code": 47,178      "B-airport_name": 70,179      "B-arrive_date.date_relative": 89,180      "B-arrive_date.day_name": 52,181      "B-arrive_date.day_number": 9,182      "B-arrive_date.month_name": 8,183      "B-arrive_date.today_relative": 54,184      "B-arrive_time.end_time": 61,185      "B-arrive_time.period_mod": 92,186      "B-arrive_time.period_of_day": 85,187      "B-arrive_time.start_time": 60,188      "B-arrive_time.time": 13,189      "B-arrive_time.time_relative": 12,190      "B-city_name": 39,191      "B-class_type": 33,192      "B-connect": 51,193      "B-cost_relative": 5,194      "B-day_name": 112,195      "B-day_number": 114,196      "B-days_code": 103,197      "B-depart_date.date_relative": 30,198      "B-depart_date.day_name": 24,199      "B-depart_date.day_number": 20,200      "B-depart_date.month_name": 19,201      "B-depart_date.today_relative": 45,202      "B-depart_date.year": 55,203      "B-depart_time.end_time": 58,204      "B-depart_time.period_mod": 86,205      "B-depart_time.period_of_day": 23,206      "B-depart_time.start_time": 56,207      "B-depart_time.time": 27,208      "B-depart_time.time_relative": 26,209      "B-economy": 42,210      "B-fare_amount": 6,211      "B-fare_basis_code": 43,212      "B-flight_days": 64,213      "B-flight_mod": 40,214      "B-flight_number": 66,215      "B-flight_stop": 46,216      "B-flight_time": 82,217      "B-fromloc.airport_code": 84,218      "B-fromloc.airport_name": 37,219      "B-fromloc.city_name": 1,220      "B-fromloc.state_code": 53,221      "B-fromloc.state_name": 48,222      "B-meal": 41,223      "B-meal_code": 68,224      "B-meal_description": 18,225      "B-mod": 65,226      "B-month_name": 113,227      "B-or": 31,228      "B-period_of_day": 94,229      "B-restriction_code": 90,230      "B-return_date.date_relative": 80,231      "B-return_date.day_name": 88,232      "B-return_date.day_number": 109,233      "B-return_date.month_name": 108,234      "B-return_date.today_relative": 117,235      "B-return_time.period_mod": 115,236      "B-return_time.period_of_day": 116,237      "B-round_trip": 3,238      "B-state_code": 75,239      "B-state_name": 102,240      "B-stoploc.airport_name": 111,241      "B-stoploc.city_name": 11,242      "B-stoploc.state_code": 95,243      "B-time": 99,244      "B-time_relative": 106,245      "B-today_relative": 100,246      "B-toloc.airport_code": 79,247      "B-toloc.airport_name": 28,248      "B-toloc.city_name": 2,249      "B-toloc.country_name": 77,250      "B-toloc.state_code": 15,251      "B-toloc.state_name": 25,252      "B-transport_type": 73,253      "I-airline_name": 22,254      "I-airport_name": 71,255      "I-arrive_date.day_number": 78,256      "I-arrive_time.end_time": 62,257      "I-arrive_time.period_of_day": 93,258      "I-arrive_time.start_time": 98,259      "I-arrive_time.time": 14,260      "I-arrive_time.time_relative": 105,261      "I-city_name": 50,262      "I-class_type": 34,263      "I-cost_relative": 35,264      "I-depart_date.day_number": 44,265      "I-depart_date.today_relative": 96,266      "I-depart_time.end_time": 59,267      "I-depart_time.period_of_day": 104,268      "I-depart_time.start_time": 57,269      "I-depart_time.time": 36,270      "I-depart_time.time_relative": 110,271      "I-economy": 83,272      "I-fare_amount": 7,273      "I-fare_basis_code": 97,274      "I-flight_mod": 63,275      "I-flight_stop": 72,276      "I-flight_time": 87,277      "I-fromloc.airport_name": 38,278      "I-fromloc.city_name": 10,279      "I-fromloc.state_name": 49,280      "I-meal_code": 69,281      "I-meal_description": 119,282      "I-restriction_code": 91,283      "I-return_date.date_relative": 81,284      "I-return_date.today_relative": 118,285      "I-round_trip": 4,286      "I-stoploc.city_name": 17,287      "I-time": 107,288      "I-today_relative": 101,289      "I-toloc.airport_name": 29,290      "I-toloc.city_name": 16,291      "I-toloc.state_name": 67,292      "I-transport_type": 74,293      "O": 0294    }295  },296  "_name_or_path": "OpenSLUv1",297  "_num_labels": {298    "intent": 21,299    "slot": 120300  },301  "architectures": [302    "PretrainedModelForSLUToSave"303  ],304  "is_decoder": true,305  "model": {306    "_model_target_": "model.OpenSLUModel",307    "decoder": {308      "_model_target_": "model.decoder.StackPropagationDecoder",309      "intent_classifier": {310        "_model_target_": "model.decoder.classifier.AutoregressiveLSTMClassifier",311        "bidirectional": false,312        "dropout_rate": 0.5,313        "embedding_dim": 8,314        "force_ratio": 0.9,315        "hidden_dim": 64,316        "ignore_index": -100,317        "input_dim": 384,318        "intent_label_num": 21,319        "layer_num": 1,320        "mode": "token-level-intent",321        "return_sentence_level": true,322        "use_intent": true,323        "use_multi": false324      },325      "interaction": {326        "_model_target_": "model.decoder.interaction.StackInteraction",327        "differentiable": false,328        "input_dim": 384,329        "intent_label_num": 21,330        "slot_label_num": 120331      },332      "slot_classifier": {333        "_model_target_": "model.decoder.classifier.AutoregressiveLSTMClassifier",334        "bidirectional": false,335        "dropout_rate": 0.55,336        "embedding_dim": 32,337        "force_ratio": 0.9,338        "hidden_dim": 64,339        "ignore_index": -100,340        "input_dim": 405,341        "layer_num": 1,342        "mode": "slot",343        "return_sentence_level": false,344        "slot_label_num": 120,345        "use_multi": false,346        "use_slot": true347      }348    },349    "encoder": {350      "_model_target_": "model.encoder.AutoEncoder",351      "attention": {352        "dropout_rate": 0.6,353        "hidden_dim": 1024,354        "output_dim": 128355      },356      "embedding": {357        "dropout_rate": 0.55,358        "embedding_dim": 256,359        "vocab_size": 900360      },361      "encoder_name": "self-attention-lstm",362      "lstm": {363        "bidirectional": true,364        "dropout_rate": 0.5,365        "layer_num": 1,366        "output_dim": 256367      },368      "output_dim": 384,369      "return_sentence_level_hidden": false,370      "return_with_input": true371    }372  },373  "return_dict": false,374  "tokenizer": {375    "_align_mode_": "fast",376    "_padding_side_": "right",377    "_to_lower_case_": true,378    "_tokenizer_name_": "word_tokenizer",379    "add_special_tokens": false,380    "max_length": 512381  },382  "tokenizer_class": "OpenSLUv1",383  "torch_dtype": "float32",384  "transformers_version": "4.25.1",385  "vocab_size": 900386}387