iworldtong / text_gcn.pytorch Public

Notifications You must be signed in to change notification settings
Fork 67
Star 262

PyTorch implementation of "Graph Convolutional Networks for Text Classification. Yao et al. AAAI2019."

262 stars 67 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 10 Commits
data		data
models		models
preprocess		preprocess
src		src
utils		utils
.DS_Store		.DS_Store
README.md		README.md
config.py		config.py
train.py		train.py
tsne.py		tsne.py

Repository files navigation

text_gcn.pytorch

This project implements "Graph Convolutional Networks for Text Classification. Yao et al. AAAI2019." in PyTorch.

This implementation highly based on official code yao8839836/text_gcn.

Require

Python 3.6
PyTorch 1.0

Running training and evaluation

cd ./preprocess
Run python remove_words.py <dataset>
Run python build_graph.py <dataset>
cd ..
Run python train.py <dataset>
Replace <dataset> with 20ng, R8, R52, ohsumed or mr

Visualization

R8

R8_gcn_test

About

PyTorch implementation of "Graph Convolutional Networks for Text Classification. Yao et al. AAAI2019."

Report repository

Releases

No releases published

Packages

No packages published

Languages

Python 100.0%