mentBERT / model_performance_metrics.txt
reab5555's picture
Upload 7 files
d8a5741 verified
raw
history blame contribute delete
No virus
1.15 kB
Classification Report, AUC Score, F1 Score, and Losses
Train Losses: [0.20890531058768613, 0.15252333594386344]
Validation Losses: [0.1771729415499725, 0.17168622303959782]
F1 Score: [0.6797109149921196]
AUC Score: [0.7942126157213086]
Classification Report:
_Borderline:
Precision: 0.6682600382409177
Recall: 0.5923728813559322
F1-score: 0.628032345013477
_Anxiety:
Precision: 0.6620808254514188
Recall: 0.6497890295358649
F1-score: 0.65587734241908
_Depression:
Precision: 0.7261363636363637
Recall: 0.5424448217317487
F1-score: 0.6209912536443148
_Bipolar:
Precision: 0.8055555555555556
Recall: 0.5233798195242002
F1-score: 0.6345101939333666
_OCD:
Precision: 0.8200431034482759
Recall: 0.6532188841201717
F1-score: 0.7271858576206404
_ADHD:
Precision: 0.8740068104426788
Recall: 0.660377358490566
F1-score: 0.7523204689789936
_Schizophrenia:
Precision: 0.8017524644030668
Recall: 0.6472148541114059
F1-score: 0.716242661448141
_Asperger:
Precision: 0.7368421052631579
Recall: 0.6570397111913358
F1-score: 0.6946564885496183
_PTSD:
Precision: 0.8612244897959184
Recall: 0.581267217630854
F1-score: 0.6940789473684211