| 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')) | |
| ``` | |