branch-switch-v6 / README.md
hitty28's picture
Add model card
21f5187 verified
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.