
胶囊文本分类
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
胶囊文本分类是一种先进的机器学习技术,用于自动识别和归类不同类型的文本数据,通过模拟人类大脑处理信息的方式提高准确性和效率。
我们的论文关于胶囊网络的文本分类在EMNLP18上被接受了,并且我们提供了相应的实现代码。代码使用Python 2.7编写,需要TensorFlow 1.4.1的支持。
资料准备脚本reuters_process.py可以清除原始数据并生成Reuters-Multilabel和Reuters-Full两个数据集。要快速开始,请参考相关说明以获取Reuters-Multilabel数据集的指导;其他数据集的信息请参照相应指南。
utils.py文件包含了几个基础功能,比如_conv2d_wrapper、_separable_conv2d_wrapper 和 _get_variable_wrapper等。layers.py中实现了胶囊网络的主要组件,包括主胶囊层、卷积胶囊层、扁平化胶囊层和全连接(FC)胶囊层。network.py提供了两种不同类型的胶囊网络实现,并且包含了一个基础版本用于比较参考。
全部评论 (0)
还没有任何评论哟~


