mteb/banking77
Viewer โข Updated โข 13.1k โข 26.8k โข 22
How to use MithunKaveriappa/banking77-distilbert with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="MithunKaveriappa/banking77-distilbert") # pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("MithunKaveriappa/banking77-distilbert")
model = AutoModelForSequenceClassification.from_pretrained("MithunKaveriappa/banking77-distilbert", device_map="auto")DistilBERT fine-tuned to classify customer messages into 77 banking intents
(e.g. card_arrival, top_up_failed, pending_transfer).
| Model | Accuracy | Macro-F1 |
|---|---|---|
| TF-IDF + Logistic Regression | 0.8898 | 0.8899 |
| BGE-small embeddings + Logistic Regression | 0.9311 | 0.9310 |
| DistilBERT (this model) | 0.9217 | 0.9217 |
distilbert-base-uncasedmteb/banking77, 90/10 train/validation split (stratified), test set held outfrom transformers import pipeline
clf = pipeline("text-classification", model="MithunKaveriappa/banking77-distilbert", top_k=3)
print(clf("My card has not arrived yet"))
Base model
distilbert/distilbert-base-uncased