mlx-examples/gcn

Graph Convolutional Network

An example of GCN implementation with MLX.

Install requirements

First, install the few dependencies with pip.

pip install -r requirements.txt

Run

To try the model, just run the main.py file. This will download the Cora dataset, run the training and testing.

python main.py