1.安装 Transformers 库
首先,我们需要安装 Transformers 库,该库是一个用于自然语言处理(NLP)任务的强大工具。
!pip install transformers
2.使用 Transformers 中的 Pipeline 加载器模块
Transformers 提供了一个功能强大的模块,即 Pipeline 加载器,用于加载各种 NLP 任务的预训练模型。
from transformers import pipeline
# 定义一个情感分析分类器
classifier = pipeline('sentiment-analysis')
3.情感分析示例
让我们通过一个简单的情感分析任务来演示如何使用 Transformers 中的 Pipeline 加载器。
# 对文本进行情感分析
result = classifier('I love you')
# 打印结果
print(result)
以上代码将输出对给定文本进行情感分析的结果,显示了情感以及相应的置信度。
注意 目前pipeline只支持分析英文,分析中文需要导入中文的训练模型,以遥遥领先为例,如果不导入中文模型遥遥领先会被认为是消极的情感
导入国内的中文模型后打印的结果就是积极的情感
from transformers import pipeline
classifier = pipeline('sentiment-analysis', 'uer/roberta-base-finetuned-dianping-chinese')
result = classifier('遥遥领先')
result