hitty28/branch-switch-augmented

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: wants_branch_switch, no_branch_switch

Usage

from transformers import pipeline

classifier = pipeline("text-classification", model="hitty28/branch-switch-augmented")
result = classifier("I want to switch to Mumbai branch")
print(result)

Training Data

Trained on custom dataset with statements about branch switching intentions.