如何用ERNIE作特殊领域的文本分类?
收藏
我想用ERNIE作文本分类,是一种特殊的业务领域的分类方式,是否可以?有没有例子可以借鉴?
在Studio里面看到的项目,都是用ERNIE作情感分类的,我理解的是,ERNIE肯定是针对情感分类作了预训练,
而我要作的是特殊领域的分类,好像不是那么简单吧?需要自己从头训练一个模型吗?而且似乎ERNIE的模型训练,
必须在配置GPU的环境下运行,普通的开发电脑似乎扛不住!!!
0
收藏
请登录后评论
可以直接用的,ERNIE是基于大规模预料训练的语义模型,与领域无关,并没有针对情感分类做特别的处理。
只要参照情感分类的例子,整理好数据,修改一下类别数,就可以跑任务看看结果。
收集相关数据做好标注,然后用预训练模型微调。
训练可以在aistudio上用免费的gpu,推断用cpu也可以吧
长知识了