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 |
||
|---|---|---|
| .. | ||
| .gitignore | ||
| datasets.py | ||
| gcn.py | ||
| main.py | ||
| README.md | ||
| requirements.txt | ||
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