Loading…
Loading…
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"))
distilbert/distilbert-base-uncased