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。
阅读量:21
阅读量:88
阅读量:64
阅读量:40
阅读量:60