label_mapping = {'non-severe': 0, 'severe': 1}


Step	Training Loss
500	0.518800
1000	0.321000
1500	0.210000
2000	0.120900
Accuracy: 0.8452921586436573
------------------------------
Precision: 0.8686567164179104
------------------------------
Recall: 0.909375
------------------------------
F1-score: 0.8885496183206106
------------------------------
MCC: 0.637922907223529
------------------------------
Confusion Matrix:
[[ 755  308]
 [ 203 2037]]
------------------------------