首页 PaddleNLP 帖子详情
如何用ERNIE作特殊领域的文本分类?
收藏
快速回复
PaddleNLP 问答NLP 1430 3
如何用ERNIE作特殊领域的文本分类?
收藏
快速回复
PaddleNLP 问答NLP 1430 3

我想用ERNIE作文本分类,是一种特殊的业务领域的分类方式,是否可以?有没有例子可以借鉴?

在Studio里面看到的项目,都是用ERNIE作情感分类的,我理解的是,ERNIE肯定是针对情感分类作了预训练,

而我要作的是特殊领域的分类,好像不是那么简单吧?需要自己从头训练一个模型吗?而且似乎ERNIE的模型训练,

必须在配置GPU的环境下运行,普通的开发电脑似乎扛不住!!!

0
收藏
回复
全部评论(3)
时间顺序
学习委员
#2 回复于2020-03

可以直接用的,ERNIE是基于大规模预料训练的语义模型,与领域无关,并没有针对情感分类做特别的处理。

只要参照情感分类的例子,整理好数据,修改一下类别数,就可以跑任务看看结果。

0
回复
CChan
#3 回复于2020-03

收集相关数据做好标注,然后用预训练模型微调。

训练可以在aistudio上用免费的gpu,推断用cpu也可以吧

0
回复
AIStudio810258
#4 回复于2020-03

长知识了

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户