Model Performance Metrics

DistilBERT Fine-Tuning Results Across 3 Epochs

Epoch Training Loss Validation Accuracy Accuracy F1 Precision Recall Precision Fun Precision Surprise Precision Neutral Precision Enthusiasm Precision Happiness Precision Hate Precision Sadness Precision Empty Precision Love Precision Relief Precision Anger
1 0.349200 0.367162 0.885096 0.888268 0.894924 0.885096 0.923581 0.7550882 0.963037 0.998424 0.729825 0.967809 0.678454 0.992661 0.973849 0.978587 0.919028
2 0.282900 0.354562 0.889610 0.891328 0.894542 0.889610 0.894949 0.771513 0.968844 0.974869 0.784091 0.971846 0.708262 0.954545 0.947528 0.963427 0.921986
3 0.213400 0.388078 0.889046 0.890193 0.891950 0.889046 0.887663 0.754302 0.975758 0.960825 0.808297 0.963768 0.735971 0.927242 0.946573 0.949949 0.939355
GitHub