transformers库入门方法

2025-07-26 20:05:04

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

transformers库入门方法

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

transformers库入门方法

3、下面介绍一个使用例子首先导入这个包,在python中执行:from transformers import *

4、这里以一个情感分类为例,直接使用pipeline,选择一个模型nlp_sentence_classif = pipeline("sentiment-analysis")首次使用会从网络下载模型,需要等待一段时间,这里日志还能看到这个模型的基本情况。

transformers库入门方法

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

transformers库入门方法

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

transformers库入门方法
声明:本网站引用、摘录或转载内容仅供网站访问者交流或参考,不代表本站立场,如存在版权或非法内容,请联系站长删除,联系邮箱:site.kefu@qq.com。
猜你喜欢