您好，使用paddlehub2.0进行文本分类，如何修改模型可以支持一个文本对应多个分类环境是ai studio，运行项目PaddleHub2.0——使用动态图版预训练模型ERNIE实现文新闻本分类但是当前项目只能实现一篇文本一个分类，如何实现一个文本对应多个分类，需要修改什么地方，怎么去修改谢谢~
4uqofj5v1#
你好，可以在预测时获取前向传播的结果，这个时候输出的就是经过softmax的软标签，可以自己选择top k。但是有个点需要注意，如果想要模型预测得准，训练时的数据也需要有软标签的样本。PaddleHub/paddlehub/module/nlp_module.pyLine 565 in ab3a163| | probs=self(input_ids, segment_ids) |
1条答案
按热度按时间4uqofj5v1#
你好，可以在预测时获取前向传播的结果，这个时候输出的就是经过softmax的软标签，可以自己选择top k。但是有个点需要注意，如果想要模型预测得准，训练时的数据也需要有软标签的样本。
PaddleHub/paddlehub/module/nlp_module.py
Line 565 in ab3a163
| | probs=self(input_ids, segment_ids) |