mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-09-03 15:04:35 +08:00
f45a1ab83c
* update a few examples to use compile * update mnist * add compile to vae and rename some stuff for simplicity * update reqs * use state in eval * GCN example with RNG + dropout * add a bit of prefetching
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