rohin30n/finance-classifier-bert

Finance Classifier Model

This directory contains the fine-tuned mBERT model for binary financial conversation classification.

Model Files

The model directory should contain:

• config.json - Model configuration
• tokenizer_config.json - Tokenizer configuration
• special_tokens_map.json - Special tokens mapping
• pytorch_model.bin - Trained model weights (generated by training)

Training

To generate the trained model, run:

cd nlp/
python train_classifier.py

This will:

1. Load training data from ../classifier_training.json
2. Fine-tune bert-base-multilingual-cased on financial vs non-financial classification
3. Save the trained model to this directory

Model Details

• Base Model: bert-base-multilingual-cased
• Task: Binary Classification (financial: 1, non-financial: 0)
• Input: Text sentences
• Languages: Multilingual support
• Training File: classifier_training.json

Usage

from nlp.classifier import FinanceClassifier

clf = FinanceClassifier()
result = clf.predict("Loan lena chahiye")
print(result)  # {'prediction': 'financial', 'confidence': 0.95}