| { | |
| "architecture": "V5 Flat 8-Class MentalBERT", | |
| "dataset": "mohamedasem318/mental-health-dataset-extended-v5", | |
| "backbone": "mental/mental-bert-base-uncased", | |
| "num_classes": 8, | |
| "classes": [ | |
| "Anxiety", | |
| "Bipolar", | |
| "Depression", | |
| "Directed Aggression", | |
| "Normal", | |
| "Personality Disorder", | |
| "Stress", | |
| "Suicidal" | |
| ], | |
| "id2label": { | |
| "0": "Anxiety", | |
| "1": "Bipolar", | |
| "2": "Depression", | |
| "3": "Directed Aggression", | |
| "4": "Normal", | |
| "5": "Personality Disorder", | |
| "6": "Stress", | |
| "7": "Suicidal" | |
| }, | |
| "label2id": { | |
| "Anxiety": 0, | |
| "Bipolar": 1, | |
| "Depression": 2, | |
| "Directed Aggression": 3, | |
| "Normal": 4, | |
| "Personality Disorder": 5, | |
| "Stress": 6, | |
| "Suicidal": 7 | |
| }, | |
| "max_len": 128, | |
| "batch_size": 32, | |
| "learning_rate": 2e-05, | |
| "epochs": 4, | |
| "label_smoothing": 0.05, | |
| "imbalance": "WeightedRandomSampler + class-weighted CE (capped at 3.0)", | |
| "best_val_f1": 0.8307269417033605, | |
| "test_metrics": { | |
| "accuracy": 0.8283594767540631, | |
| "f1_macro": 0.8350278478225386, | |
| "f1_weighted": 0.8280225059714043, | |
| "dep_to_sui": 733, | |
| "sui_to_dep": 516, | |
| "total_bleed": 1249, | |
| "roc_auc_macro": 0.9638191731454191 | |
| }, | |
| "baselines": { | |
| "mb_flat_v3_deployed": { | |
| "acc": 0.8198, | |
| "f1": 0.7646, | |
| "bleed": 1224 | |
| }, | |
| "mb_flat_v5_ph10": { | |
| "acc": 0.8241, | |
| "f1": 0.8308, | |
| "bleed": 1265 | |
| }, | |
| "v5_3stage_flat_lf": { | |
| "acc": 0.8404, | |
| "f1": 0.846, | |
| "bleed": 1179, | |
| "sui_to_dep": 274 | |
| } | |
| } | |
| } |