egynuha-classifier / training_config.json
SafwanLjd's picture
Upload folder using huggingface_hub
d030f62 verified
{
"model_path": "faisalq/EgyBERT",
"id2label": {
"0": "محايد",
"1": "اعتراض/رفض",
"2": "السب او التنمر",
"3": "صور نمطية ضارة",
"4": "التأثيم والإتهام",
"5": "شتائم جنسية",
"6": "التحرش الجنسي اللفظي",
"7": "العنف الجنسي",
"8": "التحريض/استعداء السلطات",
"9": "التهديد"
},
"label2id": {
"محايد": 0,
"اعتراض/رفض": 1,
"السب او التنمر": 2,
"صور نمطية ضارة": 3,
"التأثيم والإتهام": 4,
"شتائم جنسية": 5,
"التحرش الجنسي اللفظي": 6,
"العنف الجنسي": 7,
"التحريض/استعداء السلطات": 8,
"التهديد": 9
},
"num_labels": 10,
"max_length": 64,
"training": {
"learning_rate": 2e-05,
"batch_size": 16,
"gradient_accumulation": 2,
"epochs": 15,
"warmup_ratio": 0.1,
"lr_scheduler": "cosine"
},
"loss": {
"type": "class_balanced_focal",
"beta": 0.999,
"gamma": 0.5
},
"sampling": {
"method": "sqrt_oversampling",
"min_samples_target": 1200,
"max_oversample_ratio": 20
},
"ensemble": {
"seeds": [
42,
123,
456
],
"weights": [
0.33705821050623624,
0.3220079415508732,
0.3409338479428906
],
"best_model_seed": 456
},
"calibration": {
"temperature": 1.6999999999999997,
"per_class_thresholds": [
1.1999999999999997,
0.49999999999999994,
0.7,
0.95,
1.2499999999999998,
1.0999999999999999,
1.3999999999999997,
0.3,
0.8999999999999999,
1.3999999999999997
]
},
"results": {
"accuracy": 0.817786693413574,
"f1_macro": 0.6878295290398673,
"f1_weighted": 0.8161890790250527,
"individual_model_f1s": [
0.6852009859861107,
0.677420596947488,
0.6871622318853237
]
}
}