| { | |
| "model_name": "mental/mental-bert-base-uncased", | |
| "max_len": 128, | |
| "n_classes": 8, | |
| "classes": [ | |
| "Anxiety", | |
| "Bipolar", | |
| "Depression", | |
| "Directed Aggression", | |
| "Normal", | |
| "Personality Disorder", | |
| "Stress", | |
| "Suicidal" | |
| ], | |
| "n_sources": 6, | |
| "source_map": { | |
| "cssrs": 0, | |
| "huggingface": 1, | |
| "kaggle": 2, | |
| "kaggle_bpd": 3, | |
| "olid": 4, | |
| "swmh": 5 | |
| }, | |
| "source_reliability": { | |
| "cssrs": 1.0, | |
| "olid": 1.0, | |
| "kaggle_bpd": 0.95, | |
| "huggingface": 0.7, | |
| "kaggle": 0.7, | |
| "swmh": 0.5 | |
| }, | |
| "lambda_src": 0.3, | |
| "best_epoch": 3, | |
| "val_f1_macro": 0.8395463482978247, | |
| "test_accuracy": 0.8322668327764879, | |
| "test_f1_macro": 0.8380939220912206, | |
| "test_f1_weighted": 0.8321028523440407, | |
| "dep_to_sui": 715, | |
| "sui_to_dep": 544, | |
| "total_bleed": 1259, | |
| "dataset_version": "v5", | |
| "architecture": "source_aware_multitask_flat", | |
| "hub_repo": "itsLu/mentalbert-v5-source-aware" | |
| } |