Downloads · 30 days
89
0% of all-time downloads
distilroberta-finetuned-banking77
mrm8488/distilroberta-finetuned-banking77
distilroberta-finetuned-banking77 is a text classification model from mrm8488. Use it when you need a label for a piece of text. It is set up for transformers.
All-time downloads
122K
Public
Parameters
82.2M
657 MB on disk
Likes
9
Public
Repo makeup
Click a slice to open those files.
.bin329 MB · 50%
Parameter types
How the weights are stored.
F3282.2M · 100%
Try a prompt
Datasets
Model card
From the Hugging Face model README
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 <span style="color: #e25555;">♥</span> in Spain
- Task
- Text Classification
- Library
- transformers
- Type
- roberta
- Languages
- en
- Created
- Mar 2, 2022
- Updated
- Apr 4, 2023