Branch Switch Classifier
This model classifies whether a user statement indicates a desire to switch branches or not.
Model Details
- Base Model: DistilBERT
- Task: Binary Text Classification
- Labels: True, False
Usage
from transformers import pipeline
classifier = pipeline("text-classification", model="hitty28/branch-switch-v3")
result = classifier("I want to switch to Mumbai branch")
print(result)
Training Data
Trained on custom dataset with statements about branch switching intentions.
- Downloads last month
- 2
Model tree for hitty28/branch-switch-v3
Base model
distilbert/distilbert-base-uncased