text_gcn

收录理由

如果你正在做文本分类，又想拿图神经网络跟传统的词袋模型做对比，这个仓库就是AAAI 2019那篇Text GCN论文的官方实现。它不再把每篇文档看成独立的词袋，而是把文档和单词都当成节点，构建一张异构图，边同时编码词与文档的共现关系以及词与词的共现关系，然后在这张图上跑图卷积。仓库里把清洗语料、建图、训练这三步都写成了脚本，论文里用的20ng、R8、R52、ohsumed、mr这几个数据集也都直接接好了，跑出来的结果可以跟论文里的数字核对。另外它还提了一个归纳式版本的链接，适合测试文档不能出现在训练图里的场景。代码基于TensorFlow 1.x，现在用的话得自己改一改才能跑起来。