Downloads · 30 days
13
2% of all-time downloads
Severity_Predictor
AliArshad/Severity_Predictor
Severity_Predictor is a text classification model from AliArshad. Use it when you need a label for a piece of text. It is set up for transformers.
All-time downloads
583
Public
Parameters
124M
498 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors498 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- gpt2
- Access
- Public
- Created
- Dec 21, 2023
- Updated
- Dec 21, 2023
- SHA
- a7991375
Try a prompt
Model card
From the Hugging Face model README
Model Training Summary
Data Splitting
- Split Ratio: 80:20 (Training:Test)
Label Mapping
label_mapping = {'non-severe': 0, 'severe': 1}
Training Loss Over Iterations
| Step | Training Loss |
|---|---|
| 500 | 0.518800 |
| 1000 | 0.321000 |
| 1500 | 0.210000 |
| 2000 | 0.120900 |
Evaluation Metrics
- Accuracy: 0.8452921586436573
- Precision: 0.8686567164179104
- Recall: 0.909375
- F1-score: 0.8885496183206106
- MCC (Matthews Correlation Coefficient): 0.637922907223529
Confusion Matrix
[[ 755 308]
[ 203 2037]]
- Task
- Text Classification
- Library
- transformers
- Type
- gpt2
- Languages
- en
- Created
- Dec 21, 2023
- Updated
- Dec 21, 2023