Modelpublic

hitty28/branch-switch-augmented

sourceHugging Faceapache-2.0updated 1y agoView on Hugging Face

0likes8downloads

Model Card

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: wantsbranchswitch, nobranchswitch

Usage

python

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.