An example of GCN implementation with MLX.
First, install the few dependencies with pip
.
pip install -r requirements.txt
To try the model, just run the main.py
file. This will download the Cora dataset, run the training and testing.
python main.py