目录
1 了解
2 文本分类
1 了解
1 依赖安装
!pip install transformers -i https://pypi.tuna.tsinghua.edu.cn/simple some-package
2 了解transformers 能做什么
from transformers.pipelines import SUPPORTED_TASKS
SUPPORTED_TASKS.items()
2 文本分类
我没外网所以需要自己下载模型。
2.1 下载模型
https://huggingface.co/docs/transformers/installation#offline-mode
下载之后一起放在一个文件(dianping)下:
from transformers import pipeline
pipe = pipeline("text-classification",model='./dianping')# ,from_pt=True
pipe("我并不是太喜欢杨柳依依的样子!")
[{'label': 'negative (stars 1, 2 and 3)', 'score': 0.8996551632881165}]
情感分析的二分类文本任务就结束了。自己安装环境试试吧!
【手把手带你实战HuggingFace Transformers-入门篇】基础组件之Datasets_哔哩哔哩_bilibili
GitHub - zyds/transformers-code: 手把手带你实战 Huggingface Transformers 课程视频同步更新在B站与YouTube