Upload 3 files

config.json

@@ -1,178 +1,276 @@
 {
   "_id2label": {
     "intent": [
-      "PlayMusic",
-      "RateBook",
-      "SearchCreativeWork",
-      "SearchScreeningEvent",
-      "AddToPlaylist",
-      "BookRestaurant",
-      "GetWeather"
+      "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-artist",
-      "I-artist",
-      "B-object_name",
-      "I-object_name",
-      "B-rating_value",
-      "B-rating_unit",
-      "B-sort",
-      "B-music_item",
-      "B-object_type",
-      "I-object_type",
-      "B-best_rating",
-      "B-location_name",
-      "I-location_name",
-      "B-playlist",
-      "I-playlist",
-      "B-movie_name",
-      "I-movie_name",
-      "B-party_size_description",
-      "I-party_size_description",
-      "B-city",
-      "B-restaurant_type",
-      "B-playlist_owner",
-      "B-party_size_number",
-      "I-city",
-      "B-condition_temperature",
-      "B-timeRange",
-      "I-timeRange",
-      "B-country",
-      "B-served_dish",
-      "I-served_dish",
-      "B-movie_type",
-      "B-spatial_relation",
-      "I-spatial_relation",
-      "B-restaurant_name",
-      "I-restaurant_name",
-      "B-entity_name",
-      "I-entity_name",
-      "B-service",
-      "B-cuisine",
-      "B-poi",
-      "I-poi",
-      "I-service",
-      "B-state",
-      "B-track",
-      "I-track",
-      "B-object_select",
-      "B-object_part_of_series_type",
-      "I-restaurant_type",
-      "I-playlist_owner",
-      "B-year",
-      "I-sort",
-      "B-geographic_poi",
-      "I-geographic_poi",
-      "B-genre",
-      "B-object_location_type",
-      "I-object_location_type",
-      "B-current_location",
-      "I-state",
-      "B-condition_description",
-      "I-current_location",
-      "I-country",
-      "I-genre",
-      "I-movie_type",
-      "B-facility",
-      "B-album",
-      "I-cuisine",
-      "I-album",
-      "I-facility",
-      "I-object_part_of_series_type",
-      "I-music_item",
-      "I-object_select"
+      "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": {
-      "AddToPlaylist": 4,
-      "BookRestaurant": 5,
-      "GetWeather": 6,
-      "PlayMusic": 0,
-      "RateBook": 1,
-      "SearchCreativeWork": 2,
-      "SearchScreeningEvent": 3
+      "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-album": 65,
-      "B-artist": 1,
-      "B-best_rating": 11,
-      "B-city": 20,
-      "B-condition_description": 59,
-      "B-condition_temperature": 25,
-      "B-country": 28,
-      "B-cuisine": 39,
-      "B-current_location": 57,
-      "B-entity_name": 36,
-      "B-facility": 64,
-      "B-genre": 54,
-      "B-geographic_poi": 52,
-      "B-location_name": 12,
-      "B-movie_name": 16,
-      "B-movie_type": 31,
-      "B-music_item": 8,
-      "B-object_location_type": 55,
-      "B-object_name": 3,
-      "B-object_part_of_series_type": 47,
-      "B-object_select": 46,
-      "B-object_type": 9,
-      "B-party_size_description": 18,
-      "B-party_size_number": 23,
-      "B-playlist": 14,
-      "B-playlist_owner": 22,
-      "B-poi": 40,
-      "B-rating_unit": 6,
-      "B-rating_value": 5,
-      "B-restaurant_name": 34,
-      "B-restaurant_type": 21,
-      "B-served_dish": 29,
-      "B-service": 38,
-      "B-sort": 7,
-      "B-spatial_relation": 32,
-      "B-state": 43,
-      "B-timeRange": 26,
-      "B-track": 44,
-      "B-year": 50,
-      "I-album": 67,
-      "I-artist": 2,
-      "I-city": 24,
-      "I-country": 61,
-      "I-cuisine": 66,
-      "I-current_location": 60,
-      "I-entity_name": 37,
-      "I-facility": 68,
-      "I-genre": 62,
-      "I-geographic_poi": 53,
-      "I-location_name": 13,
-      "I-movie_name": 17,
-      "I-movie_type": 63,
-      "I-music_item": 70,
-      "I-object_location_type": 56,
-      "I-object_name": 4,
-      "I-object_part_of_series_type": 69,
-      "I-object_select": 71,
-      "I-object_type": 10,
-      "I-party_size_description": 19,
-      "I-playlist": 15,
-      "I-playlist_owner": 49,
-      "I-poi": 41,
-      "I-restaurant_name": 35,
-      "I-restaurant_type": 48,
-      "I-served_dish": 30,
-      "I-service": 42,
-      "I-sort": 51,
-      "I-spatial_relation": 33,
-      "I-state": 58,
-      "I-timeRange": 27,
-      "I-track": 45,
+      "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": 7,
-    "slot": 72
+    "intent": 17,
+    "slot": 111
   },
   "architectures": [
     "PretrainedModelForSLUToSave"
@@ -188,7 +286,7 @@
         "dropout_rate": 0.4,
         "ignore_index": "{base.ignore_index}",
         "input_dim": 384,
-        "intent_label_num": 7,
+        "intent_label_num": 17,
         "loss_fn": {
           "_model_target_": "torch.nn.BCEWithLogitsLoss"
         },
@@ -252,7 +350,7 @@
         "mode": "slot",
         "multi_threshold": 0.5,
         "return_sentence_level": false,
-        "slot_label_num": 72,
+        "slot_label_num": 111,
         "use_multi": false,
         "use_slot": true
       }
@@ -284,5 +382,5 @@
   "tokenizer_class": "OpenSLUv1",
   "torch_dtype": "float32",
   "transformers_version": "4.25.1",
-  "vocab_size": 11758
+  "vocab_size": 790
 }

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:fab10e1a878df673d73d73377b7f5916ded21754f71edd66a6a97c1aa2f7b576
-size 20445104
+oid sha256:cf7b6cd788f0b8211327a1e690f8b16febbf5e479c09e445825f67297c94b0a6
+size 9196096

tokenizer.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cede86a9c60e470d6e281f1bea2aaadbf8876d573c186fae33da4e007e943b59
-size 205345
+oid sha256:5435e3d70b344234613dbde8264bbcd6c4a1c990b3848a0801907aa0d5426361
+size 12218