waimai / README.md
lightning-cheetah's picture
Create README.md
90d752e verified
metadata
license: apache-2.0
datasets:
  - XiangPan/waimai_10k
language:
  - zh
base_model:
  - huawei-noah/TinyBERT_4L_zh
pipeline_tag: text-classification

基于TinyBERT训练的一个模型,使用方法如下:

>>> from transformers import pipeline
>>> classifer = pipeline("text-classification", model="waimai")
>>> classifer("这家店明码标价、分量足!")
[{'label': '差评', 'score': 0.9906463623046875}]