Downloads · 30 days
10
0% of all-time downloads
GautamR/model_grievance_class
model_grievance_class is a text classification model from GautamR. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as apache-2.0.
This model classifies grievances into five distinct buckets:
Downloads · 30 days
10
0% of all-time downloads
All-time downloads
56.4K
Public
Parameters
24.6M
2.6 GB on disk
Likes
0
Public
Click a slice to open those files.
.bin98.7 MB · 50%
From the Hugging Face model README
This model classifies grievances into five distinct buckets:
Account number is not Correct: This pertains to feedback or grievances related to the bank account number not being displayed correctly on the portal for farmers.
Installment not received: This pertains to feedback where the farmer has not received the installment or money for a given month.
Others: This covers a broad range of other grievances including, but not limited to:
Convo_starter: Any phrase that could be a conversation starter like 'Hi','Hello','Namasthe'
Convo_ender: Any phrase that could signal the end of a conversation like 'Thank you','Goodbye'
The code for training the model can be found here
The following training metrics were observed over 15 epochs:
| Epoch | Loss | Accuracy | F1 Score |
|---|---|---|---|
| 1 | 1.0574 | 0.4242 | 0.4202 |
| 2 | 0.7780 | 0.7500 | 0.6952 |
| 3 | 0.5212 | 0.8939 | 0.8825 |
| 4 | 0.3214 | 0.9773 | 0.9767 |
| 5 | 0.1612 | 1.0000 | 1.0000 |
| 6 | 0.0883 | 1.0000 | 1.0000 |
| 7 | 0.0592 | 1.0000 | 1.0000 |
| 8 | 0.0418 | 1.0000 | 1.0000 |
| 9 | 0.0321 | 1.0000 | 1.0000 |