🚀 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 |