Skip to content

Latest commit

 

History

History
15 lines (13 loc) · 321 Bytes

README.md

File metadata and controls

15 lines (13 loc) · 321 Bytes

Policy-GNN

This is the implementation of Policy-GNN.

Installation requirements

python3.5.2
numpy==1.16.4
torch-cluster==1.4.5
torch-scatter==1.4.0
torch-sparse==0.4.3
torch-geometric==1.3.2
torch==1.4.0

Example

python train_cora.py
python train_citeseer.py