distilroberta-base fine-tuned on banking77 dataset for intent classification
Test set accuray: 0.896
How to use
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}]
Created by Manuel Romero/@mrm8488 | LinkedIn Made with ♥ in Spain
- Downloads last month
- 34