
Text_GCN:应用于文本分类的图卷积网络。AAAI 2019
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本文提出了一种名为Text-GCN的模型,在文本分类任务中应用图卷积网络以提高性能,该论文发表于AAAI 2019会议。
本段落介绍了Text GCN的实现:梁耀、毛成胜、罗源在第33届AAAI人工智能会议(AAAI-19)上发表的文章《图卷积网络用于文本分类》中的方法,需要使用Python 2.7或3.6版本以及Tensorflow >= 1.4.0来重现结果。具体步骤如下:
运行 `python remove_words.py 20ng`
运行 `python build_graph.py 20ng`
运行 `python train.py 20ng`
对于其他数据集,只需将上述命令中的“20ng”替换为R8、R52、ohsumed和mr。
输入文件示例:/data/20ng.txt 包含文档名称、培训测试分组以及文档标签。每行对应一个文档。
/data/corpus/20ng.txt包含每个文档的原始文本,与前一文件中的相应行一一对应。
prepare_data.py提供了准备自己数据集的一个例子,请注意,“\n”已经被从文档或句子中移除。
归纳版本:Text GCN的归纳版在训练过程中不包括测试文档。
全部评论 (0)
还没有任何评论哟~


