mlx-examples/gcn
Awni Hannun 27c0a8c002
Add llms subdir + update README (#145)
* add llms subdir + update README

* nits

* use same pre-commit as mlx

* update readmes a bit

* format
2023-12-20 10:22:25 -08:00
..
.gitignore fix comments before merge 2023-12-11 23:10:46 +01:00
datasets.py Add llms subdir + update README (#145) 2023-12-20 10:22:25 -08:00
gcn.py add GCN implementation 2023-12-11 17:48:07 +01:00
main.py Add llms subdir + update README (#145) 2023-12-20 10:22:25 -08:00
README.md fix comments before merge 2023-12-11 23:10:46 +01:00
requirements.txt fix comments before merge 2023-12-11 23:10:46 +01:00

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