GautamR's picture
Update README.md
0b06148
|
raw
history blame
1.15 kB
metadata
license: mit
language:
  - en
metrics:
  - accuracy
library_name: transformers
pipeline_tag: text-classification
tags:
  - agriculture

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

  1. agricultural_scheme: The farmer query is about schemes in Odisha

  2. agriculture: General agri queries

  3. pest: The farmer query is about pests

  4. 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