transformer pipeline文本分类
时间: 2023-10-18 20:06:27 浏览: 244
Transformer:自然语言处理的变革者
好的,针对您的问题,我可以告诉您,transformer pipeline是一个基于Hugging Face的transformer模型的Python库,它提供了一系列预训练的模型和工具,可以用于自然语言处理任务,包括文本分类。其中,文本分类是指将文本分为不同的类别或标签。使用transformer pipeline进行文本分类非常简单,只需要几行代码即可完成。
下面是一个使用transformer pipeline进行文本分类的示例代码:
```python
from transformers import pipeline
# 加载文本分类模型
classifier = pipeline('text-classification', model='distilbert-base-uncased-finetuned-sst-2-english')
# 进行文本分类
result = classifier('This movie is terrible!')
print(result)
```
输出结果如下:
```
[{'label': 'NEGATIVE', 'score': 0.9996823077201843}]
```
可以看到,该模型将输入的文本分类为“NEGATIVE”,即负面评价。
阅读全文