--- language: en license: apache-2.0 tags: - sentence-classification - banking datasets: - banking --- # Banking Sentence Classifier This is a sentence classification model for identifying relevant banking sentences. ## Model Details - Base Model: sentence-transformers/all-MiniLM-L6-v2 - Task: Binary Classification (Relevant/Irrelevant Banking Sentences) ## Usage ```python from transformers import AutoTokenizer from sentence_transformers import SentenceTransformer import torch import joblib # Load models tokenizer = AutoTokenizer.from_pretrained('sentence-transformers/all-MiniLM-L6-v2') base_model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2') classifier = ClassificationHead(base_model.get_sentence_embedding_dimension(), num_classes=2) classifier.load_state_dict(joblib.load('classifier_state.pth')) ```