Thanks to visit codestin.com
Credit goes to github.com

Skip to content

xuehui1991/text_gcn

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

52 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

text_gcn

The implementation of Text GCN in our paper:

Liang Yao, Chengsheng Mao, Yuan Luo. "Graph Convolutional Networks for Text Classification." In 33rd AAAI Conference on Artificial Intelligence (AAAI-19)

Require

Python 2.7 or 3.6

Tensorflow >= 1.4.0

Reproducing Results

  1. Run python remove_words.py 20ng

  2. Run python build_graph.py 20ng

  3. Run python train.py 20ng

  4. Change 20ng in above 3 command lines to R8, R52, ohsumed and mr when producing results for other datasets.

Example input data

  1. /data/20ng.txt indicates document names, training/test split, document labels. Each line is for a document.

  2. /data/corpus/20ng.txt contains raw text of each document, each line is for the corresponding line in /data/20ng.txt

  3. prepare_data.py is an example for preparing your own data, note that '\n' is removed in your documents or sentences.

About

Graph Convolutional Networks for Text Classification. AAAI 2019

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Yacc 99.7%
  • Python 0.3%