zechengz / ctf-gcn

Graph convolutional network in CTF Python, NumPy and SciPy

Graph Convolutional Network in Cyclops Tensor Framework

Graph convolutional network in CTF Python, NumPy and SciPy

Building

• pip install -r requirements.txt to install the required python package first.
• To install Cyclops Tensor Framework (CTF) which is a parallel (distributed-memory) numerical library for multidimensional arrays (tensors) in C++ and Python, please refer to Cyclops Tensor Framework Building and Testing.

Usage

• Enter python train.py -h to see args parameters and default values.
• Example: python train.py --save_best=True --epochs=100 --package=scipy
• To run with CTF in parallel: mpirun -n 8 python train.py --package=ctf --epochs=100