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.

Downloads last month
20
Safetensors
Model size
67M params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for hitty28/branch-switch-noise-roberta-base

Finetuned
(11359)
this model