Update project and configurations
This commit is contained in:
104
intelligent_cabin/models/local_bert_multi_intent/config.json
Normal file
104
intelligent_cabin/models/local_bert_multi_intent/config.json
Normal file
@@ -0,0 +1,104 @@
|
||||
{
|
||||
"architectures": [
|
||||
"BertForSequenceClassification"
|
||||
],
|
||||
"attention_probs_dropout_prob": 0.1,
|
||||
"classifier_dropout": null,
|
||||
"directionality": "bidi",
|
||||
"dtype": "float32",
|
||||
"gradient_checkpointing": false,
|
||||
"hidden_act": "gelu",
|
||||
"hidden_dropout_prob": 0.1,
|
||||
"hidden_size": 768,
|
||||
"id2label": {
|
||||
"0": "cabin_ac_off",
|
||||
"1": "cabin_ac_on",
|
||||
"2": "cabin_defog_front_on",
|
||||
"3": "cabin_defog_rear_on",
|
||||
"4": "cabin_fan_down",
|
||||
"5": "cabin_fan_up",
|
||||
"6": "cabin_lights_off",
|
||||
"7": "cabin_lights_on",
|
||||
"8": "cabin_lock_doors",
|
||||
"9": "cabin_mirror_fold",
|
||||
"10": "cabin_mirror_unfold",
|
||||
"11": "cabin_nav_cancel",
|
||||
"12": "cabin_nav_to",
|
||||
"13": "cabin_next_track",
|
||||
"14": "cabin_pause_music",
|
||||
"15": "cabin_play_music",
|
||||
"16": "cabin_previous_track",
|
||||
"17": "cabin_seat_heat_off",
|
||||
"18": "cabin_seat_heat_on",
|
||||
"19": "cabin_set_ac",
|
||||
"20": "cabin_sunroof_close",
|
||||
"21": "cabin_sunroof_open",
|
||||
"22": "cabin_unlock_doors",
|
||||
"23": "cabin_volume_down",
|
||||
"24": "cabin_volume_mute",
|
||||
"25": "cabin_volume_up",
|
||||
"26": "cabin_window_close",
|
||||
"27": "cabin_window_open",
|
||||
"28": "cabin_wiper_off",
|
||||
"29": "cabin_wiper_on",
|
||||
"30": "cs_cancel_order",
|
||||
"31": "cs_query_logistics",
|
||||
"32": "cs_query_order",
|
||||
"33": "cs_transfer_human"
|
||||
},
|
||||
"initializer_range": 0.02,
|
||||
"intermediate_size": 3072,
|
||||
"label2id": {
|
||||
"cabin_ac_off": 0,
|
||||
"cabin_ac_on": 1,
|
||||
"cabin_defog_front_on": 2,
|
||||
"cabin_defog_rear_on": 3,
|
||||
"cabin_fan_down": 4,
|
||||
"cabin_fan_up": 5,
|
||||
"cabin_lights_off": 6,
|
||||
"cabin_lights_on": 7,
|
||||
"cabin_lock_doors": 8,
|
||||
"cabin_mirror_fold": 9,
|
||||
"cabin_mirror_unfold": 10,
|
||||
"cabin_nav_cancel": 11,
|
||||
"cabin_nav_to": 12,
|
||||
"cabin_next_track": 13,
|
||||
"cabin_pause_music": 14,
|
||||
"cabin_play_music": 15,
|
||||
"cabin_previous_track": 16,
|
||||
"cabin_seat_heat_off": 17,
|
||||
"cabin_seat_heat_on": 18,
|
||||
"cabin_set_ac": 19,
|
||||
"cabin_sunroof_close": 20,
|
||||
"cabin_sunroof_open": 21,
|
||||
"cabin_unlock_doors": 22,
|
||||
"cabin_volume_down": 23,
|
||||
"cabin_volume_mute": 24,
|
||||
"cabin_volume_up": 25,
|
||||
"cabin_window_close": 26,
|
||||
"cabin_window_open": 27,
|
||||
"cabin_wiper_off": 28,
|
||||
"cabin_wiper_on": 29,
|
||||
"cs_cancel_order": 30,
|
||||
"cs_query_logistics": 31,
|
||||
"cs_query_order": 32,
|
||||
"cs_transfer_human": 33
|
||||
},
|
||||
"layer_norm_eps": 1e-12,
|
||||
"max_position_embeddings": 512,
|
||||
"model_type": "bert",
|
||||
"num_attention_heads": 12,
|
||||
"num_hidden_layers": 12,
|
||||
"pad_token_id": 0,
|
||||
"pooler_fc_size": 768,
|
||||
"pooler_num_attention_heads": 12,
|
||||
"pooler_num_fc_layers": 3,
|
||||
"pooler_size_per_head": 128,
|
||||
"pooler_type": "first_token_transform",
|
||||
"position_embedding_type": "absolute",
|
||||
"problem_type": "multi_label_classification",
|
||||
"transformers_version": "4.57.3",
|
||||
"type_vocab_size": 2,
|
||||
"use_cache": true,
|
||||
"vocab_size": 21128
|
||||
}
|
||||
@@ -0,0 +1,36 @@
|
||||
{
|
||||
"LABEL_0": "cabin_ac_off",
|
||||
"LABEL_1": "cabin_ac_on",
|
||||
"LABEL_2": "cabin_defog_front_on",
|
||||
"LABEL_3": "cabin_defog_rear_on",
|
||||
"LABEL_4": "cabin_fan_down",
|
||||
"LABEL_5": "cabin_fan_up",
|
||||
"LABEL_6": "cabin_lights_off",
|
||||
"LABEL_7": "cabin_lights_on",
|
||||
"LABEL_8": "cabin_lock_doors",
|
||||
"LABEL_9": "cabin_mirror_fold",
|
||||
"LABEL_10": "cabin_mirror_unfold",
|
||||
"LABEL_11": "cabin_nav_cancel",
|
||||
"LABEL_12": "cabin_nav_to",
|
||||
"LABEL_13": "cabin_next_track",
|
||||
"LABEL_14": "cabin_pause_music",
|
||||
"LABEL_15": "cabin_play_music",
|
||||
"LABEL_16": "cabin_previous_track",
|
||||
"LABEL_17": "cabin_seat_heat_off",
|
||||
"LABEL_18": "cabin_seat_heat_on",
|
||||
"LABEL_19": "cabin_set_ac",
|
||||
"LABEL_20": "cabin_sunroof_close",
|
||||
"LABEL_21": "cabin_sunroof_open",
|
||||
"LABEL_22": "cabin_unlock_doors",
|
||||
"LABEL_23": "cabin_volume_down",
|
||||
"LABEL_24": "cabin_volume_mute",
|
||||
"LABEL_25": "cabin_volume_up",
|
||||
"LABEL_26": "cabin_window_close",
|
||||
"LABEL_27": "cabin_window_open",
|
||||
"LABEL_28": "cabin_wiper_off",
|
||||
"LABEL_29": "cabin_wiper_on",
|
||||
"LABEL_30": "cs_cancel_order",
|
||||
"LABEL_31": "cs_query_logistics",
|
||||
"LABEL_32": "cs_query_order",
|
||||
"LABEL_33": "cs_transfer_human"
|
||||
}
|
||||
@@ -0,0 +1,37 @@
|
||||
{
|
||||
"cls_token": {
|
||||
"content": "[CLS]",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false
|
||||
},
|
||||
"mask_token": {
|
||||
"content": "[MASK]",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false
|
||||
},
|
||||
"pad_token": {
|
||||
"content": "[PAD]",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false
|
||||
},
|
||||
"sep_token": {
|
||||
"content": "[SEP]",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false
|
||||
},
|
||||
"unk_token": {
|
||||
"content": "[UNK]",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false
|
||||
}
|
||||
}
|
||||
21292
intelligent_cabin/models/local_bert_multi_intent/tokenizer.json
Normal file
21292
intelligent_cabin/models/local_bert_multi_intent/tokenizer.json
Normal file
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,56 @@
|
||||
{
|
||||
"added_tokens_decoder": {
|
||||
"0": {
|
||||
"content": "[PAD]",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
},
|
||||
"100": {
|
||||
"content": "[UNK]",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
},
|
||||
"101": {
|
||||
"content": "[CLS]",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
},
|
||||
"102": {
|
||||
"content": "[SEP]",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
},
|
||||
"103": {
|
||||
"content": "[MASK]",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
}
|
||||
},
|
||||
"clean_up_tokenization_spaces": false,
|
||||
"cls_token": "[CLS]",
|
||||
"do_lower_case": true,
|
||||
"extra_special_tokens": {},
|
||||
"mask_token": "[MASK]",
|
||||
"model_max_length": 1000000000000000019884624838656,
|
||||
"pad_token": "[PAD]",
|
||||
"sep_token": "[SEP]",
|
||||
"strip_accents": null,
|
||||
"tokenize_chinese_chars": true,
|
||||
"tokenizer_class": "BertTokenizer",
|
||||
"unk_token": "[UNK]"
|
||||
}
|
||||
@@ -0,0 +1,56 @@
|
||||
{
|
||||
"task_type": "multi_label_intent_detection",
|
||||
"base_model": "hfl/chinese-macbert-base",
|
||||
"epochs": 12,
|
||||
"batch_size": 8,
|
||||
"learning_rate": 2e-05,
|
||||
"threshold": 0.5,
|
||||
"top_k": 4,
|
||||
"train_size": 862,
|
||||
"dev_size": 236,
|
||||
"label_count": 34,
|
||||
"labels": [
|
||||
"cabin_ac_off",
|
||||
"cabin_ac_on",
|
||||
"cabin_defog_front_on",
|
||||
"cabin_defog_rear_on",
|
||||
"cabin_fan_down",
|
||||
"cabin_fan_up",
|
||||
"cabin_lights_off",
|
||||
"cabin_lights_on",
|
||||
"cabin_lock_doors",
|
||||
"cabin_mirror_fold",
|
||||
"cabin_mirror_unfold",
|
||||
"cabin_nav_cancel",
|
||||
"cabin_nav_to",
|
||||
"cabin_next_track",
|
||||
"cabin_pause_music",
|
||||
"cabin_play_music",
|
||||
"cabin_previous_track",
|
||||
"cabin_seat_heat_off",
|
||||
"cabin_seat_heat_on",
|
||||
"cabin_set_ac",
|
||||
"cabin_sunroof_close",
|
||||
"cabin_sunroof_open",
|
||||
"cabin_unlock_doors",
|
||||
"cabin_volume_down",
|
||||
"cabin_volume_mute",
|
||||
"cabin_volume_up",
|
||||
"cabin_window_close",
|
||||
"cabin_window_open",
|
||||
"cabin_wiper_off",
|
||||
"cabin_wiper_on",
|
||||
"cs_cancel_order",
|
||||
"cs_query_logistics",
|
||||
"cs_query_order",
|
||||
"cs_transfer_human"
|
||||
],
|
||||
"best_dev_metrics": {
|
||||
"micro_precision": 1.0,
|
||||
"micro_recall": 0.9929,
|
||||
"micro_f1": 0.9964,
|
||||
"exact_match": 0.9915,
|
||||
"recall_at_k": 1.0
|
||||
},
|
||||
"device": "mps"
|
||||
}
|
||||
21128
intelligent_cabin/models/local_bert_multi_intent/vocab.txt
Normal file
21128
intelligent_cabin/models/local_bert_multi_intent/vocab.txt
Normal file
File diff suppressed because it is too large
Load Diff
Reference in New Issue
Block a user