metadata
license: apache-2.0
base_model: distilbert-base-uncased
tags:
- text-classification
- branch-switching
- intent-classification
datasets:
- custom
language:
- en
pipeline_tag: text-classification
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-v6")
result = classifier("I want to switch to Mumbai branch")
print(result)
Training Data
Trained on custom dataset with statements about branch switching intentions.