mlx-examples/gcn/README.md
2023-12-11 23:10:46 +01:00

372 B

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