Suchun-sv / Policy-GNN

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About


Languages

Language:Python 100.0%