💳 Consumer Complaint Classification — DistilBERT
This is the fine-tuned DistilBERT model for automated multi-class classification of consumer financial complaints across five resolution categories:
credit_card(Credit card or prepaid card)credit_reporting(Credit reporting, credit repair services, or other personal consumer reports)debt_collection(Debt collection)mortgages_and_loans(Mortgage, payday loan, title loan, or personal loan)retail_banking(Checking or savings account, money transfers, or virtual currency)
Performance
- Test Accuracy: 85.43%
- Test Macro F1: 0.8467
- Test Loss: 0.4215
- CPU Median Latency: ~28 ms
Repository
Complete code, Gradio web app, training reproduction, and evaluation notebooks are available at: GitHub: Solayman-Sameh/consumer-complaint-classification
- Downloads last month
- 20