cartesinus's picture
Upload folder using huggingface_hub (#1)
54d42ed
{
"model_name": "distilbert-base-multilingual-cased",
"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": "distillb-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
}
}