setsumbt-dst-tm123 / config.json
CarelvNiekerk's picture
Upload config
57605c4
raw
history blame contribute delete
No virus
20 kB
{
"_name_or_path": "setsumbt-dst-tm123",
"adapters": {
"adapters": {},
"config_map": {},
"fusion_config_map": {},
"fusions": {}
},
"architectures": [
"RobertaSetSUMBT"
],
"attention_probs_dropout_prob": 0.1,
"bos_token_id": 0,
"candidate_embedding_model_name": "roberta-base",
"classifier_dropout": null,
"dev_batch_size": 8,
"distance_measure": "cosine",
"domain_ids": {
"car_repairs": [
1,
2,
3,
4,
5,
6,
7,
8
],
"coffee": [
9,
10,
11,
12,
13,
14
],
"flights": [
15,
16,
17,
18,
19,
20,
21,
22,
23,
24,
25,
26,
27,
28,
29,
30,
31,
32,
33,
34,
35,
36,
37,
38,
39
],
"hotel": [
40,
41,
42,
43,
44,
45,
46,
47,
48,
49,
50,
51,
52,
53,
54,
55,
56,
57,
58,
59,
60
],
"movies": [
61,
62,
63,
64,
65,
66,
67,
68,
69,
70,
71,
72,
73,
74,
75,
76,
77,
78,
79,
80,
81,
82,
83,
84,
85,
86,
87,
88,
89,
90,
91,
92,
93,
94,
95,
96
],
"music": [
97,
98,
99,
100,
101,
102,
103,
104,
105,
106,
107
],
"restaurant": [
108,
109,
110,
111,
112,
113,
114,
115,
116,
117,
118,
119,
120,
121,
122,
123,
124
],
"sports": [
128,
129,
130,
131,
132,
133,
134,
135,
136,
137,
138,
139,
140,
141,
142,
125,
126,
127
],
"takeout": [
143,
144,
145,
146,
147,
148,
149,
150,
151,
152,
153,
154,
155,
156,
157,
158,
159,
160,
161
],
"uber_lyft": [
162,
163,
164,
165,
166,
167,
168,
169
]
},
"dropout_rate": 0.3,
"eos_token_id": 2,
"freeze_encoder": false,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.1,
"hidden_size": 768,
"informable_slot_ids": {
"car_repairs-date_appt": 1,
"car_repairs-location_store": 2,
"car_repairs-name_customer": 3,
"car_repairs-name_store": 4,
"car_repairs-name_vehicle": 5,
"car_repairs-reason_appt": 6,
"car_repairs-time_appt": 7,
"car_repairs-year_vehicle": 8,
"coffee-location_store": 9,
"coffee-name_drink": 10,
"coffee-num_drink": 11,
"coffee-preference": 12,
"coffee-size_drink": 13,
"coffee-type_milk": 14,
"flights-airline": 15,
"flights-date": 16,
"flights-date_depart_intermediate": 17,
"flights-date_depart_origin": 18,
"flights-date_return": 19,
"flights-destination1": 20,
"flights-destination2": 21,
"flights-fare": 22,
"flights-flight_number": 23,
"flights-from": 24,
"flights-from_time": 25,
"flights-luggage": 26,
"flights-num_pax": 27,
"flights-origin": 28,
"flights-other_description": 29,
"flights-price_range": 30,
"flights-seat_location": 31,
"flights-seating_class": 32,
"flights-stops": 33,
"flights-stops_location": 34,
"flights-time_of_day": 35,
"flights-to": 36,
"flights-to_time": 37,
"flights-total_fare": 38,
"flights-type": 39,
"hotel-amenity": 40,
"hotel-check-in_date": 41,
"hotel-check-out_date": 42,
"hotel-customer_rating": 43,
"hotel-customer_review": 44,
"hotel-date_range": 45,
"hotel-location": 46,
"hotel-location_hotel": 47,
"hotel-name_hotel": 48,
"hotel-num_beds": 49,
"hotel-num_guests": 50,
"hotel-num_rooms": 51,
"hotel-other_detail": 52,
"hotel-other_request": 53,
"hotel-price_per_night": 54,
"hotel-price_range": 55,
"hotel-star_rating": 56,
"hotel-sub_location_hotel": 57,
"hotel-total_fare": 58,
"hotel-type_bed": 59,
"hotel-type_room": 60,
"movies-audience_rating": 61,
"movies-character": 62,
"movies-critic_rating": 63,
"movies-date_release": 64,
"movies-date_showing": 65,
"movies-duration_movie": 68,
"movies-genre": 69,
"movies-location": 70,
"movies-location_theater": 71,
"movies-movie_rating": 72,
"movies-name_character": 73,
"movies-name_genre": 74,
"movies-name_movie": 75,
"movies-name_person": 76,
"movies-name_theater": 77,
"movies-num_tickets": 78,
"movies-other_description": 79,
"movies-price_streaming": 80,
"movies-price_ticket": 81,
"movies-price_total": 82,
"movies-rating_movie": 83,
"movies-real_person": 84,
"movies-release_date": 85,
"movies-review_audience": 86,
"movies-review_critic": 87,
"movies-runtime": 88,
"movies-seating": 89,
"movies-streaming_service": 90,
"movies-time_duration": 91,
"movies-time_end": 92,
"movies-time_preference": 93,
"movies-time_showing": 94,
"movies-time_start": 95,
"movies-type_screening": 96,
"music-describes_album": 97,
"music-describes_artist": 98,
"music-describes_genre": 99,
"music-describes_track": 100,
"music-describes_type_music": 101,
"music-name_album": 102,
"music-name_artist": 103,
"music-name_genre": 104,
"music-name_track": 105,
"music-technical_difficulty": 106,
"music-type_music": 107,
"restaurant-business_hours": 108,
"restaurant-date_reservation": 109,
"restaurant-location": 110,
"restaurant-location_restaurant": 111,
"restaurant-menu_item": 112,
"restaurant-name_reservation": 113,
"restaurant-name_restaurant": 114,
"restaurant-num_guests": 115,
"restaurant-other_description": 116,
"restaurant-phone": 117,
"restaurant-price_range": 118,
"restaurant-rating": 119,
"restaurant-sub-location": 120,
"restaurant-time_reservation": 121,
"restaurant-type_food": 122,
"restaurant-type_meal": 123,
"restaurant-type_seating": 124,
"sports-date_match": 125,
"sports-day_match": 126,
"sports-name_non_player": 127,
"sports-name_player": 128,
"sports-name_team": 129,
"sports-other_description_match": 130,
"sports-other_description_person": 131,
"sports-other_description_team": 132,
"sports-place_team": 133,
"sports-position_player": 134,
"sports-record_games_ahead": 135,
"sports-record_games_back": 136,
"sports-record_player": 137,
"sports-record_team": 138,
"sports-result_match": 139,
"sports-score_match": 140,
"sports-time_match": 141,
"sports-venue": 142,
"takeout-location_restaurant": 143,
"takeout-location_store": 144,
"takeout-name_item": 145,
"takeout-name_pizza": 146,
"takeout-name_restaurant": 147,
"takeout-name_store": 148,
"takeout-num_people": 149,
"takeout-other_description_item": 150,
"takeout-preference": 151,
"takeout-price_range": 152,
"takeout-rating_restaurant": 153,
"takeout-size_pizza": 154,
"takeout-time_pickup": 155,
"takeout-total_price": 156,
"takeout-type_crust": 157,
"takeout-type_food": 158,
"takeout-type_meal": 159,
"takeout-type_retrieval": 160,
"takeout-type_topping": 161,
"uber_lyft-duration_estimate": 162,
"uber_lyft-location_from": 163,
"uber_lyft-location_to": 164,
"uber_lyft-num_people": 165,
"uber_lyft-price_estimate": 166,
"uber_lyft-time_dropoff": 167,
"uber_lyft-time_pickup": 168,
"uber_lyft-type_ride": 169
},
"initializer_range": 0.02,
"intermediate_size": 3072,
"label_smoothing": 0.05,
"layer_norm_eps": 1e-05,
"loss_function": "labelsmoothing",
"max_candidate_len": 12,
"max_dialogue_len": 12,
"max_position_embeddings": 514,
"max_slot_len": 12,
"max_turn_len": 64,
"model_type": "roberta",
"nbt_hidden_size": 300,
"nbt_layers": 1,
"nbt_type": "gru",
"num_attention_heads": 12,
"num_hidden_layers": 12,
"num_slots": 170,
"num_values": {
"car_repairs-date_appt": 399,
"car_repairs-location_store": 366,
"car_repairs-name_customer": 1960,
"car_repairs-name_store": 242,
"car_repairs-name_vehicle": 883,
"car_repairs-reason_appt": 2766,
"car_repairs-time_appt": 2306,
"car_repairs-year_vehicle": 96,
"coffee-location_store": 2869,
"coffee-name_drink": 1200,
"coffee-num_drink": 103,
"coffee-preference": 1149,
"coffee-size_drink": 163,
"coffee-type_milk": 261,
"flights-airline": 272,
"flights-date": 331,
"flights-date_depart_intermediate": 36,
"flights-date_depart_origin": 799,
"flights-date_return": 944,
"flights-destination1": 472,
"flights-destination2": 17,
"flights-fare": 940,
"flights-flight_number": 34,
"flights-from": 168,
"flights-from_time": 290,
"flights-luggage": 18,
"flights-num_pax": 75,
"flights-origin": 171,
"flights-other_description": 1348,
"flights-price_range": 831,
"flights-seat_location": 42,
"flights-seating_class": 145,
"flights-stops": 280,
"flights-stops_location": 144,
"flights-time_of_day": 290,
"flights-to": 228,
"flights-to_time": 290,
"flights-total_fare": 876,
"flights-type": 43,
"hotel-amenity": 1386,
"hotel-check-in_date": 365,
"hotel-check-out_date": 271,
"hotel-customer_rating": 513,
"hotel-customer_review": 38,
"hotel-date_range": 1774,
"hotel-location": 233,
"hotel-location_hotel": 349,
"hotel-name_hotel": 1954,
"hotel-num_beds": 38,
"hotel-num_guests": 201,
"hotel-num_rooms": 50,
"hotel-other_detail": 916,
"hotel-other_request": 1146,
"hotel-price_per_night": 856,
"hotel-price_range": 1027,
"hotel-star_rating": 479,
"hotel-sub_location_hotel": 773,
"hotel-total_fare": 29,
"hotel-type_bed": 153,
"hotel-type_room": 311,
"movies-audience_rating": 222,
"movies-character": 324,
"movies-critic_rating": 626,
"movies-date_release": 201,
"movies-date_showing": 1930,
"movies-description_other": 1,
"movies-description_plot": 1,
"movies-duration_movie": 12,
"movies-genre": 291,
"movies-location": 5102,
"movies-location_theater": 2564,
"movies-movie_rating": 41,
"movies-name_character": 404,
"movies-name_genre": 1424,
"movies-name_movie": 11020,
"movies-name_person": 1210,
"movies-name_theater": 8629,
"movies-num_tickets": 1944,
"movies-other_description": 966,
"movies-price_streaming": 52,
"movies-price_ticket": 1253,
"movies-price_total": 953,
"movies-rating_movie": 260,
"movies-real_person": 711,
"movies-release_date": 153,
"movies-review_audience": 656,
"movies-review_critic": 2266,
"movies-runtime": 290,
"movies-seating": 1119,
"movies-streaming_service": 75,
"movies-time_duration": 12,
"movies-time_end": 290,
"movies-time_preference": 290,
"movies-time_showing": 290,
"movies-time_start": 290,
"movies-type_screening": 478,
"music-describes_album": 102,
"music-describes_artist": 307,
"music-describes_genre": 16,
"music-describes_track": 1400,
"music-describes_type_music": 190,
"music-name_album": 430,
"music-name_artist": 2065,
"music-name_genre": 134,
"music-name_track": 3105,
"music-technical_difficulty": 201,
"music-type_music": 213,
"restaurant-business_hours": 938,
"restaurant-date_reservation": 30,
"restaurant-location": 1463,
"restaurant-location_restaurant": 1310,
"restaurant-menu_item": 514,
"restaurant-name_reservation": 286,
"restaurant-name_restaurant": 6071,
"restaurant-num_guests": 263,
"restaurant-other_description": 1940,
"restaurant-phone": 109,
"restaurant-price_range": 861,
"restaurant-rating": 710,
"restaurant-sub-location": 411,
"restaurant-time_reservation": 290,
"restaurant-type_food": 918,
"restaurant-type_meal": 57,
"restaurant-type_seating": 675,
"sports-date_match": 334,
"sports-day_match": 98,
"sports-name_non_player": 223,
"sports-name_player": 878,
"sports-name_team": 1206,
"sports-other_description_match": 19,
"sports-other_description_person": 41,
"sports-other_description_team": 412,
"sports-place_team": 254,
"sports-position_player": 151,
"sports-record_games_ahead": 14,
"sports-record_games_back": 168,
"sports-record_player": 65,
"sports-record_team": 1112,
"sports-result_match": 206,
"sports-score_match": 905,
"sports-time_match": 290,
"sports-venue": 183,
"takeout-location_restaurant": 9,
"takeout-location_store": 605,
"takeout-name_item": 1828,
"takeout-name_pizza": 1320,
"takeout-name_restaurant": 43,
"takeout-name_store": 808,
"takeout-num_people": 12,
"takeout-other_description_item": 679,
"takeout-preference": 597,
"takeout-price_range": 7,
"takeout-rating_restaurant": 5,
"takeout-size_pizza": 188,
"takeout-time_pickup": 290,
"takeout-total_price": 7,
"takeout-type_crust": 345,
"takeout-type_food": 120,
"takeout-type_meal": 8,
"takeout-type_retrieval": 23,
"takeout-type_topping": 689,
"uber_lyft-duration_estimate": 12,
"uber_lyft-location_from": 2839,
"uber_lyft-location_to": 2937,
"uber_lyft-num_people": 12,
"uber_lyft-price_estimate": 1835,
"uber_lyft-time_dropoff": 2306,
"uber_lyft-time_pickup": 2306,
"uber_lyft-type_ride": 549
},
"pad_token_id": 1,
"position_embedding_type": "absolute",
"predict_actions": false,
"requestable_slot_ids": {
"coffee-name_drink": 10,
"movies-num_tickets": 78,
"movies-rating_movie": 83,
"takeout-size_pizza": 154
},
"rnn_zero_init": false,
"seed": 0,
"set_pooling": "cnn",
"set_similarity": true,
"slot_attention_heads": 12,
"slot_ids": {
"car_repairs-date_appt": 1,
"car_repairs-location_store": 2,
"car_repairs-name_customer": 3,
"car_repairs-name_store": 4,
"car_repairs-name_vehicle": 5,
"car_repairs-reason_appt": 6,
"car_repairs-time_appt": 7,
"car_repairs-year_vehicle": 8,
"coffee-location_store": 9,
"coffee-name_drink": 10,
"coffee-num_drink": 11,
"coffee-preference": 12,
"coffee-size_drink": 13,
"coffee-type_milk": 14,
"flights-airline": 15,
"flights-date": 16,
"flights-date_depart_intermediate": 17,
"flights-date_depart_origin": 18,
"flights-date_return": 19,
"flights-destination1": 20,
"flights-destination2": 21,
"flights-fare": 22,
"flights-flight_number": 23,
"flights-from": 24,
"flights-from_time": 25,
"flights-luggage": 26,
"flights-num_pax": 27,
"flights-origin": 28,
"flights-other_description": 29,
"flights-price_range": 30,
"flights-seat_location": 31,
"flights-seating_class": 32,
"flights-stops": 33,
"flights-stops_location": 34,
"flights-time_of_day": 35,
"flights-to": 36,
"flights-to_time": 37,
"flights-total_fare": 38,
"flights-type": 39,
"hotel-amenity": 40,
"hotel-check-in_date": 41,
"hotel-check-out_date": 42,
"hotel-customer_rating": 43,
"hotel-customer_review": 44,
"hotel-date_range": 45,
"hotel-location": 46,
"hotel-location_hotel": 47,
"hotel-name_hotel": 48,
"hotel-num_beds": 49,
"hotel-num_guests": 50,
"hotel-num_rooms": 51,
"hotel-other_detail": 52,
"hotel-other_request": 53,
"hotel-price_per_night": 54,
"hotel-price_range": 55,
"hotel-star_rating": 56,
"hotel-sub_location_hotel": 57,
"hotel-total_fare": 58,
"hotel-type_bed": 59,
"hotel-type_room": 60,
"movies-audience_rating": 61,
"movies-character": 62,
"movies-critic_rating": 63,
"movies-date_release": 64,
"movies-date_showing": 65,
"movies-description_other": 66,
"movies-description_plot": 67,
"movies-duration_movie": 68,
"movies-genre": 69,
"movies-location": 70,
"movies-location_theater": 71,
"movies-movie_rating": 72,
"movies-name_character": 73,
"movies-name_genre": 74,
"movies-name_movie": 75,
"movies-name_person": 76,
"movies-name_theater": 77,
"movies-num_tickets": 78,
"movies-other_description": 79,
"movies-price_streaming": 80,
"movies-price_ticket": 81,
"movies-price_total": 82,
"movies-rating_movie": 83,
"movies-real_person": 84,
"movies-release_date": 85,
"movies-review_audience": 86,
"movies-review_critic": 87,
"movies-runtime": 88,
"movies-seating": 89,
"movies-streaming_service": 90,
"movies-time_duration": 91,
"movies-time_end": 92,
"movies-time_preference": 93,
"movies-time_showing": 94,
"movies-time_start": 95,
"movies-type_screening": 96,
"music-describes_album": 97,
"music-describes_artist": 98,
"music-describes_genre": 99,
"music-describes_track": 100,
"music-describes_type_music": 101,
"music-name_album": 102,
"music-name_artist": 103,
"music-name_genre": 104,
"music-name_track": 105,
"music-technical_difficulty": 106,
"music-type_music": 107,
"restaurant-business_hours": 108,
"restaurant-date_reservation": 109,
"restaurant-location": 110,
"restaurant-location_restaurant": 111,
"restaurant-menu_item": 112,
"restaurant-name_reservation": 113,
"restaurant-name_restaurant": 114,
"restaurant-num_guests": 115,
"restaurant-other_description": 116,
"restaurant-phone": 117,
"restaurant-price_range": 118,
"restaurant-rating": 119,
"restaurant-sub-location": 120,
"restaurant-time_reservation": 121,
"restaurant-type_food": 122,
"restaurant-type_meal": 123,
"restaurant-type_seating": 124,
"sports-date_match": 125,
"sports-day_match": 126,
"sports-name_non_player": 127,
"sports-name_player": 128,
"sports-name_team": 129,
"sports-other_description_match": 130,
"sports-other_description_person": 131,
"sports-other_description_team": 132,
"sports-place_team": 133,
"sports-position_player": 134,
"sports-record_games_ahead": 135,
"sports-record_games_back": 136,
"sports-record_player": 137,
"sports-record_team": 138,
"sports-result_match": 139,
"sports-score_match": 140,
"sports-time_match": 141,
"sports-venue": 142,
"takeout-location_restaurant": 143,
"takeout-location_store": 144,
"takeout-name_item": 145,
"takeout-name_pizza": 146,
"takeout-name_restaurant": 147,
"takeout-name_store": 148,
"takeout-num_people": 149,
"takeout-other_description_item": 150,
"takeout-preference": 151,
"takeout-price_range": 152,
"takeout-rating_restaurant": 153,
"takeout-size_pizza": 154,
"takeout-time_pickup": 155,
"takeout-total_price": 156,
"takeout-type_crust": 157,
"takeout-type_food": 158,
"takeout-type_meal": 159,
"takeout-type_retrieval": 160,
"takeout-type_topping": 161,
"uber_lyft-duration_estimate": 162,
"uber_lyft-location_from": 163,
"uber_lyft-location_to": 164,
"uber_lyft-num_people": 165,
"uber_lyft-price_estimate": 166,
"uber_lyft-time_dropoff": 167,
"uber_lyft-time_pickup": 168,
"uber_lyft-type_ride": 169
},
"test_batch_size": 8,
"tokenizer_name": "roberta-base",
"torch_dtype": "float32",
"train_batch_size": 3,
"transformers_version": "4.24.0",
"type_vocab_size": 1,
"use_cache": true,
"use_descriptions": true,
"vocab_size": 50265
}