CoolFace
Modelpublic

Dikshith4500/SpendingCategoryBERT

sourceHugging Facemitupdated 6mo agoView on Hugging Face
0likes20downloads
Model Card

SpendingCategoryBERT

Fine-tuned BERT for classifying raw bank transaction strings into 8 spending categories. Built as part of Foresight — a proactive AI financial OS with 12 LangGraph agents and 6 custom MCP servers.

Categories

LabelCategoryExample
0groceryWHOLEFDS MKT #1523
1restaurantSTARBUCKS #12345
2transportUBER *TRIP
3shoppingAMZN MKTP US*2K4J9
4entertainmentNETFLIX.COM
5utilitiesATT*BILL PAYMENT
6healthcareCVS PHARMACY #1234
7generalVENMO PAYMENT

Performance

MetricScore
Accuracy100%
F1 (weighted)1.0
Precision1.0
Recall1.0

Usage

from transformers import pipeline classifier = pipeline("text-classification", model="Dikshith4500/SpendingCategoryBERT") classifier("WHOLEFDS MKT #1523")

Training

  • —Base model: bert-base-uncased
  • —Dataset: Dikshith4500/bank-transaction-categories-50k
  • —Examples: 6,973 labeled bank transaction strings
  • —Epochs: 3
  • —Learning rate: 2e-5
  • —Hardware: T4 GPU