上传者: 42165980
|
上传时间: 2021-11-10 12:55:15
|
文件大小: 270.57MB
|
文件类型: -
text_gcn
本文中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
在为其他数据集生成结果时,将上述3个命令行中的20ng更改为R8 , R52 , ohsumed和mr 。
输入数据示例
/data/20ng.txt表示文档名称,培训/测试组,文档标签。 每行都是一个文档。
/data/corpus/20ng.txt包含每个文档的原始文本,每行对应/data/20ng.txt的相应行
prepare_data.py是准备自己的数据的示例,请注意,文档或句子中的“ \ n”已删除。
归纳版
文本GCN的归纳版本是 ,其中培训过程中未包括测试文档。