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]]
