mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-06-24 09:21:18 +08:00
|
|
||
|---|---|---|
| .. | ||
| .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