Agri-flow Classification Model
This model classifies grievances into five distinct buckets:
- Label 0: agricultural_scheme
- Label 1: agriculture
- Label 2: pest
- Label 3: seed
Description of the Buckets
-
agricultural_scheme: The farmer query is about schemes in Odisha
-
agriculture: General agri queries
-
pest: The farmer query is about pests
-
seed: The farmer query is about seed varieties
Training Metrics
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 |