transformers库入门方法
1、安装依赖包,这里以pytorch版本为例,首先安装pytorch
执行:pip install torch

2、安装transformers包
执行:pip install transformers
到此就安装完成了

3、下面介绍一个使用例子
首先导入这个包,在python中执行:
from transformers import *
4、这里以一个情感分类为例,直接使用pipeline,选择一个模型
nlp_sentence_classif = pipeline("sentiment-analysis")
首次使用会从网络下载模型,需要等待一段时间,这里日志还能看到这个模型的基本情况。

5、加载完成就可以调用这个模型了
执行:nlp_sentence_classif("This apple is not delicious.")
就能看到这句话的分类,以及对应的打分情况了。

6、一个分类模型就这样完成了,例子中的模型对英文分类效果比较好,对中文的话就得要使用中文的模型,或者进行后续的训练了。

声明:本网站引用、摘录或转载内容仅供网站访问者交流或参考,不代表本站立场,如存在版权或非法内容,请联系站长删除,联系邮箱:site.kefu@qq.com。
阅读量:99
阅读量:72
阅读量:26
阅读量:143
阅读量:147