lightning-cheetah/waimai

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

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