File size: 731 Bytes
21f5187
c1d8b1e
21f5187
 
 
 
 
 
 
 
 
 
 
c1d8b1e
 
21f5187
c1d8b1e
21f5187
c1d8b1e
 
21f5187
 
 
c1d8b1e
21f5187
c1d8b1e
21f5187
 
c1d8b1e
21f5187
 
 
 
c1d8b1e
21f5187
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

---
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

```python
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.