Downloads · 30 days
17
20% of all-time downloads
kubler-ross
itopcu/kubler-ross
kubler-ross is a machine learning model from itopcu. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
This model is a fine-tuned version of dbmdz/bert-base-turkish-128k-cased on the Kübler-Ross grief stages dataset.
All-time downloads
86
Public
Parameters
184M
2.2 GB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors737 MB · 99%
At a glance
- Model type
- bert
- Access
- Public
- Created
- Sep 27, 2024
- Updated
- Sep 27, 2024
- SHA
- c426af9c
Model card
From the Hugging Face model README
Kubler-Ross Grief Stage Classification
This model is a fine-tuned version of dbmdz/bert-base-turkish-128k-cased on the Kübler-Ross grief stages dataset.
It classifies text into one of five stages of grief: denial, anger, bargaining, depression, acceptance.
Results on the Test Set:
| Metric | Value |
|---|---|
| Accuracy | 0.5200 |
| Precision Macro | 0.2089 |
| Recall Macro | 0.2618 |
| F1 Macro | 0.2315 |
| Precision Micro | 0.5200 |
| Recall Micro | 0.5200 |
| F1 Micro | 0.5200 |
| MCC | 0.2140 |
Training Procedure
- Learning Rate: 1e-5
- Batch Size: 16
- Epochs: 10
This model was trained on a balanced dataset of texts categorized into the five stages of grief.
- Type
- bert
- Created
- Sep 27, 2024
- Updated
- Sep 27, 2024