Text Classification
Transformers
PyTorch
English
bert
Trained with AutoTrain
Eval Results (legacy)
text-embeddings-inference
Instructions to use philschmid/BERT-Banking77 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use philschmid/BERT-Banking77 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="philschmid/BERT-Banking77")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("philschmid/BERT-Banking77") model = AutoModelForSequenceClassification.from_pretrained("philschmid/BERT-Banking77") - Notebooks
- Google Colab
- Kaggle
| tags: autotrain | |
| language: en | |
| widget: | |
| - text: I am still waiting on my card? | |
| datasets: | |
| - banking77 | |
| model-index: | |
| - name: BERT-Banking77 | |
| results: | |
| - task: | |
| name: Text Classification | |
| type: text-classification | |
| dataset: | |
| name: BANKING77 | |
| type: banking77 | |
| metrics: | |
| - name: Accuracy | |
| type: accuracy | |
| value: 92.64 | |
| - name: Macro F1 | |
| type: macro-f1 | |
| value: 92.64 | |
| - name: Weighted F1 | |
| type: weighted-f1 | |
| value: 92.6 | |
| - task: | |
| type: text-classification | |
| name: Text Classification | |
| dataset: | |
| name: banking77 | |
| type: banking77 | |
| config: default | |
| split: test | |
| metrics: | |
| - name: Accuracy | |
| type: accuracy | |
| value: 0.9275974025974026 | |
| verified: true | |
| - name: Precision Macro | |
| type: precision | |
| value: 0.9305185253845069 | |
| verified: true | |
| - name: Precision Micro | |
| type: precision | |
| value: 0.9275974025974026 | |
| verified: true | |
| - name: Precision Weighted | |
| type: precision | |
| value: 0.9305185253845071 | |
| verified: true | |
| - name: Recall Macro | |
| type: recall | |
| value: 0.9275974025974028 | |
| verified: true | |
| - name: Recall Micro | |
| type: recall | |
| value: 0.9275974025974026 | |
| verified: true | |
| - name: Recall Weighted | |
| type: recall | |
| value: 0.9275974025974026 | |
| verified: true | |
| - name: F1 Macro | |
| type: f1 | |
| value: 0.927623314966026 | |
| verified: true | |
| - name: F1 Micro | |
| type: f1 | |
| value: 0.9275974025974026 | |
| verified: true | |
| - name: F1 Weighted | |
| type: f1 | |
| value: 0.927623314966026 | |
| verified: true | |
| - name: loss | |
| type: loss | |
| value: 0.3199225962162018 | |
| verified: true | |
| co2_eq_emissions: 0.03330651014155927 | |
| # `BERT-Banking77` Model Trained Using AutoTrain | |
| - Problem type: Multi-class Classification | |
| - Model ID: 940131041 | |
| - CO2 Emissions (in grams): 0.03330651014155927 | |
| ## Validation Metrics | |
| - Loss: 0.3505457043647766 | |
| - Accuracy: 0.9263261296660118 | |
| - Macro F1: 0.9268371013605569 | |
| - Micro F1: 0.9263261296660118 | |
| - Weighted F1: 0.9259954221865809 | |
| - Macro Precision: 0.9305746406646502 | |
| - Micro Precision: 0.9263261296660118 | |
| - Weighted Precision: 0.929031563971418 | |
| - Macro Recall: 0.9263724620088746 | |
| - Micro Recall: 0.9263261296660118 | |
| - Weighted Recall: 0.9263261296660118 | |
| ## Usage | |
| You can use cURL to access this model: | |
| ``` | |
| $ curl -X POST -H "Authorization: Bearer YOUR_API_KEY" -H "Content-Type: application/json" -d '{"inputs": "I love AutoTrain"}' https://api-inference.huggingface.co/models/philschmid/autotrain-does-it-work-940131041 | |
| ``` | |
| Or Python API: | |
| ``` | |
| from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline | |
| model_id = 'philschmid/BERT-Banking77' | |
| tokenizer = AutoTokenizer.from_pretrained(model_id) | |
| model = AutoModelForSequenceClassification.from_pretrained(model_id) | |
| classifier = pipeline('text-classification', tokenizer=tokenizer, model=model) | |
| classifier('What is the base of the exchange rates?') | |
| ``` |