| { | |
| "model_name": "cartesinus/xlm_r-joint_nlu-custom_ds", | |
| "num_intents": 10, | |
| "num_slots": 7, | |
| "intent_id2label": { | |
| "0": "AddEventOnDateWithName", | |
| "1": "AddEventWithName", | |
| "2": "CheckCalendarEventName", | |
| "3": "CheckCalendarOnDate", | |
| "4": "NotifyOnEventInLocation", | |
| "5": "NotifyOnEventStart", | |
| "6": "NotifyWhenEventNameStart", | |
| "7": "NotNotifyOnEventInLocation", | |
| "8": "NotNotifyOnEventStart", | |
| "9": "OpenCalendar" | |
| }, | |
| "intent_label2id": { | |
| "AddEventOnDateWithName": "0", | |
| "AddEventWithName": "1", | |
| "CheckCalendarEventName": "2", | |
| "CheckCalendarOnDate": "3", | |
| "NotifyOnEventInLocation": "4", | |
| "NotifyOnEventStart": "5", | |
| "NotifyWhenEventNameStart": "6", | |
| "NotNotifyOnEventInLocation": "7", | |
| "NotNotifyOnEventStart": "8", | |
| "OpenCalendar": "9" | |
| }, | |
| "slot_id2label": { | |
| "0": "o", | |
| "1": "b-event_name", | |
| "2": "i-event_name", | |
| "3": "b-date", | |
| "4": "i-date", | |
| "5": "b-location", | |
| "6": "i-location" | |
| }, | |
| "slot_label2id": { | |
| "o": "0", | |
| "b-event_name": "1", | |
| "i-event_name": "2", | |
| "b-date": "3", | |
| "i-date": "4", | |
| "b-location": "5", | |
| "i-location": "6" | |
| }, | |
| "trainer_config": { | |
| "repository_id": "xlm_r-joint_nlu-custom_ds", | |
| "evaluation_strategy": "epoch", | |
| "learning_rate": 2e-05, | |
| "per_device_train_batch_size": 16, | |
| "per_device_eval_batch_size": 16, | |
| "num_train_epochs": 20, | |
| "weight_decay": 0.01, | |
| "push_to_hub": true | |
| }, | |
| "data_collator_config": { | |
| "padding": true, | |
| "max_length": 512 | |
| } | |
| } | |