深入学习 Transformers进行nlp任务之情感分析

302 阅读1分钟

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)

image.png 以上代码将输出对给定文本进行情感分析的结果,显示了情感以及相应的置信度。

注意 目前pipeline只支持分析英文,分析中文需要导入中文的训练模型,以遥遥领先为例,如果不导入中文模型遥遥领先会被认为是消极的情感

image.png

导入国内的中文模型后打印的结果就是积极的情感

from transformers import pipeline
classifier = pipeline('sentiment-analysis', 'uer/roberta-base-finetuned-dianping-chinese')
result = classifier('遥遥领先')
result

image.png