基于强化学习和胶囊网络的文本分类模型及方法

摘要

本发明涉及自然语言处理、文本分类的技术领域，更具体地，涉及基于强化学习和胶囊网络的文本分类模型及方法。本发明以强化学习Actor‑Critic，胶囊网络CapsNet为基本的框架，胶囊网络提取文本信息的特征，强化学习判别胶囊层之间的联系。本发明的创新内容在于引入了强化学习去学习胶囊网络层之间的路由关系，引入胶囊网络去解决文本分类模型中的多标签分类的任务。利用胶囊网络在多标签分类任务上的优势，应用于文本多标签分类的任务上，从而达到更好的效果；利用强化学习遇错调整的机制，学习到更好的路由之间的联系方法。