🚀 BERT - Banking77模型
BERT - Banking77模型是使用AutoTrain訓練的多分類文本分類模型,可有效處理銀行業相關文本的分類問題,在BANKING77數據集上有較高的準確率和F1值。
🚀 快速開始
你可以使用以下方式快速使用該模型:
使用cURL調用模型
$ 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
使用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?')
✨ 主要特性
- 問題類型:多分類文本分類。
- 模型ID:940131041。
- 二氧化碳排放量(克):0.03330651014155927。
📚 詳細文檔
驗證指標
屬性 |
詳情 |
損失值 |
0.3505457043647766 |
準確率 |
0.9263261296660118 |
宏F1值 |
0.9268371013605569 |
微F1值 |
0.9263261296660118 |
加權F1值 |
0.9259954221865809 |
宏精確率 |
0.9305746406646502 |
微精確率 |
0.9263261296660118 |
加權精確率 |
0.929031563971418 |
宏召回率 |
0.9263724620088746 |
微召回率 |
0.9263261296660118 |
加權召回率 |
0.9263261296660118 |
模型評估結果
任務 |
數據集 |
指標 |
值 |
文本分類 |
BANKING77 |
準確率 |
92.64 |
文本分類 |
BANKING77 |
宏F1值 |
92.64 |
文本分類 |
BANKING77 |
加權F1值 |
92.6 |
文本分類(測試集) |
banking77 |
準確率 |
0.9275974025974026 |
文本分類(測試集) |
banking77 |
宏精確率 |
0.9305185253845069 |
文本分類(測試集) |
banking77 |
微精確率 |
0.9275974025974026 |
文本分類(測試集) |
banking77 |
加權精確率 |
0.9305185253845071 |
文本分類(測試集) |
banking77 |
宏召回率 |
0.9275974025974028 |
文本分類(測試集) |
banking77 |
微召回率 |
0.9275974025974026 |
文本分類(測試集) |
banking77 |
加權召回率 |
0.9275974025974026 |
文本分類(測試集) |
banking77 |
宏F1值 |
0.927623314966026 |
文本分類(測試集) |
banking77 |
微F1值 |
0.9275974025974026 |
文本分類(測試集) |
banking77 |
加權F1值 |
0.927623314966026 |
文本分類(測試集) |
banking77 |
損失值 |
0.3199225962162018 |