arabic-eou-marbert / training_info.json
Amr-h's picture
Upload Arabic EOU detection model (MARBERT fine-tuned)
265095b verified
{
"base_model": "UBC-NLP/MARBERT",
"task": "end-of-utterance-detection",
"language": "arabic",
"dialect": "saudi",
"num_labels": 2,
"label_mapping": {
"0": "INCOMPLETE",
"1": "COMPLETE"
},
"training_config": {
"epochs": 3,
"learning_rate": 2e-05,
"batch_size": 32,
"max_length": 64,
"warmup_ratio": 0.1
},
"performance": {
"test_accuracy": 0.99,
"ood_accuracy": 0.918,
"ood_weighted_f1": 0.9176,
"complete_f1": 0.9109,
"incomplete_f1": 0.9244
}
}