add GCN implementation

This commit is contained in:
Tristan Bilot
2023-12-11 17:48:07 +01:00
parent ecd96acfe4
commit ed5a830626
5 changed files with 286 additions and 0 deletions

17
gcn/README.md Normal file
View File

@@ -0,0 +1,17 @@
# Graph Convolutional Network
An example of [GCN](https://arxiv.org/pdf/1609.02907.pdf%EF%BC%89) 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
```