text_gcn

text_gcn

我们论文中 Text GCN 的实现: 梁耀,毛承胜,罗元. "图形卷积网络用于文本分类。" 第 33 届人工智能大会 (AAAI-19) 论文集, 7370-7377

要求

Python 2.7 或 3.6 Tensorflow >= 1.4.0

重现结果

1. 运行 Python remove_words.py 20ng
2. 运行 Python build_graph.py 20ng
3. 运行 Python train.py 20ng
4. 为生成其他数据集的结果,将上述 3 行命令中的 20ng 更改为 R8,R52,ohsumed 和 mr。

示例输入数据

1. /data/20ng.txt 指出文档名称、训练/测试分割、文档标签。每一行表示一个文档。

2. /data/corpus/20ng.txt 包含每个文档的原始文本,每一行对应 /data/20ng.txt 中的相应行。