56 lines
1.2 KiB
JSON
56 lines
1.2 KiB
JSON
{
|
|
"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"
|
|
} |