File size: 720 Bytes
7fe9b22 319fd25 7fe9b22 ae13479 |
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 |
---
language: en
tags:
- banking
- intent
- multiclass
datasets:
- banking77
widget:
- text: "How long until my transfer goes through?"
---
# distilroberta-base fine-tuned on banking77 dataset for intent classification
Test set accuray: 0.896
## How to use
```py
from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
ckpt = 'mrm8488/distilroberta-finetuned-banking77'
tokenizer = AutoTokenizer.from_pretrained(ckpt)
model = AutoModelForSequenceClassification.from_pretrained(ckpt)
classifier = pipeline('text-classification', tokenizer=tokenizer, model=model)
classifier('What is the base of the exchange rates?')
# Output: [{'label': 'exchange_rate', 'score': 0.8509947657585144}]
``` |