joint-bert-slu-snips / config.json
LightChen2333's picture
Upload 3 files
60fab05
raw
history blame
5.25 kB
{
"_id2label": {
"intent": [
"PlayMusic",
"AddToPlaylist",
"RateBook",
"SearchScreeningEvent",
"BookRestaurant",
"GetWeather",
"SearchCreativeWork"
],
"slot": [
"O",
"B-artist",
"B-album",
"B-service",
"I-service",
"B-entity_name",
"I-entity_name",
"B-playlist",
"I-playlist",
"B-object_select",
"B-object_type",
"B-rating_value",
"B-best_rating",
"B-music_item",
"B-track",
"I-track",
"I-artist",
"B-playlist_owner",
"B-year",
"B-sort",
"B-movie_name",
"I-movie_name",
"B-party_size_number",
"B-state",
"B-city",
"B-timeRange",
"I-timeRange",
"B-object_part_of_series_type",
"I-object_type",
"B-movie_type",
"B-spatial_relation",
"I-spatial_relation",
"B-geographic_poi",
"I-geographic_poi",
"B-restaurant_type",
"I-city",
"B-party_size_description",
"I-party_size_description",
"B-object_location_type",
"I-object_location_type",
"B-object_name",
"I-object_name",
"I-movie_type",
"B-rating_unit",
"I-sort",
"B-location_name",
"I-location_name",
"B-current_location",
"I-current_location",
"I-playlist_owner",
"B-served_dish",
"B-country",
"B-condition_temperature",
"B-poi",
"I-poi",
"B-condition_description",
"B-genre",
"B-restaurant_name",
"I-restaurant_name",
"I-state",
"I-served_dish",
"B-cuisine",
"I-album",
"I-country",
"B-facility",
"I-facility",
"I-cuisine",
"I-music_item",
"I-object_select",
"I-restaurant_type",
"I-object_part_of_series_type",
"I-genre"
]
},
"_label2id": {
"intent": {
"AddToPlaylist": 1,
"BookRestaurant": 4,
"GetWeather": 5,
"PlayMusic": 0,
"RateBook": 2,
"SearchCreativeWork": 6,
"SearchScreeningEvent": 3
},
"slot": {
"B-album": 2,
"B-artist": 1,
"B-best_rating": 12,
"B-city": 24,
"B-condition_description": 55,
"B-condition_temperature": 52,
"B-country": 51,
"B-cuisine": 61,
"B-current_location": 47,
"B-entity_name": 5,
"B-facility": 64,
"B-genre": 56,
"B-geographic_poi": 32,
"B-location_name": 45,
"B-movie_name": 20,
"B-movie_type": 29,
"B-music_item": 13,
"B-object_location_type": 38,
"B-object_name": 40,
"B-object_part_of_series_type": 27,
"B-object_select": 9,
"B-object_type": 10,
"B-party_size_description": 36,
"B-party_size_number": 22,
"B-playlist": 7,
"B-playlist_owner": 17,
"B-poi": 53,
"B-rating_unit": 43,
"B-rating_value": 11,
"B-restaurant_name": 57,
"B-restaurant_type": 34,
"B-served_dish": 50,
"B-service": 3,
"B-sort": 19,
"B-spatial_relation": 30,
"B-state": 23,
"B-timeRange": 25,
"B-track": 14,
"B-year": 18,
"I-album": 62,
"I-artist": 16,
"I-city": 35,
"I-country": 63,
"I-cuisine": 66,
"I-current_location": 48,
"I-entity_name": 6,
"I-facility": 65,
"I-genre": 71,
"I-geographic_poi": 33,
"I-location_name": 46,
"I-movie_name": 21,
"I-movie_type": 42,
"I-music_item": 67,
"I-object_location_type": 39,
"I-object_name": 41,
"I-object_part_of_series_type": 70,
"I-object_select": 68,
"I-object_type": 28,
"I-party_size_description": 37,
"I-playlist": 8,
"I-playlist_owner": 49,
"I-poi": 54,
"I-restaurant_name": 58,
"I-restaurant_type": 69,
"I-served_dish": 60,
"I-service": 4,
"I-sort": 44,
"I-spatial_relation": 31,
"I-state": 59,
"I-timeRange": 26,
"I-track": 15,
"O": 0
}
},
"_name_or_path": "OpenSLUv1",
"_num_labels": {
"intent": 7,
"slot": 72
},
"architectures": [
"PretrainedModelForSLUToSave"
],
"is_decoder": true,
"model": {
"_model_target_": "model.open_slu_model.OpenSLUModel",
"decoder": {
"_model_target_": "model.decoder.base_decoder.BaseDecoder",
"intent_classifier": {
"_model_target_": "model.decoder.classifier.LinearClassifier",
"ignore_index": -100,
"input_dim": 768,
"intent_label_num": 7,
"mode": "intent",
"use_intent": true
},
"slot_classifier": {
"_model_target_": "model.decoder.classifier.LinearClassifier",
"ignore_index": -100,
"input_dim": 768,
"mode": "slot",
"slot_label_num": 72,
"use_slot": true
}
},
"encoder": {
"_model_target_": "model.encoder.AutoEncoder",
"encoder_name": "bert-base-uncased",
"output_dim": 768,
"return_sentence_level_hidden": true,
"return_with_input": true
},
"ignore_index": -100
},
"return_dict": false,
"tokenizer_class": "OpenSLUv1",
"torch_dtype": "float32",
"transformers_version": "4.25.1",
"vocab_size": 30522
}