TheBgGirl/stress-detection-distilbert

Binary text classifier for stress detection.

Labels

• 0: not_stressed
• 1: stressed

Evaluation

• accuracy: 0.9987
• f1: 0.9974
• macro_f1: 0.9983
• macro_precision: 0.9988
• macro_recall: 0.9977
• precision: 0.9991
• recall: 0.9957
• roc_auc: 1.0000

Usage

from transformers import pipeline

classifier = pipeline("text-classification", model="TheBgGirl/stress-detection-distilbert")
classifier("i feel overwhelmed and anxious all the time")