1. 安装PaddleNLP
安装相关过程和问题可以参考PaddleNLP的 安装文档。
>>> pip install --upgrade paddlenlp -i https://pypi.org/simple
2. 一键加载预训练模型
情感分析本质是一个文本分类任务。PaddleNLP内置了ERNIE、BERT、RoBERTa、Electra等丰富的预训练模型,并且内置了各种预训练模型对于不同下游任务的Fine-tune网络。用户可以使用PaddleNLP提供的模型,完成问答、序列分类、token分类等任务。查阅 预训练模型 了解更多。这里以ERNIE模型为例,介绍如何将预训练模型Fine-tune完成文本分类任务。
加载预训练模型ERNIE
>>> MODEL_NAME = "ernie-3.0-medium-zh" >>> ernie_model = paddlenlp.transformers.ErnieModel.from_pretrained(MODEL_NAME)
加载预训练模型ERNIE用于文本分类任务的Fine-tune网络,只需指定想要使用的模型名称和文本分类的类别数即可完成网络定义。
>>> model = paddlenlp.transformers.ErnieForSeque