mlx-examples/gcn/README.md
2023-12-11 23:10:46 +01:00

18 lines
372 B
Markdown

# Graph Convolutional Network
An example of [GCN](https://arxiv.org/abs/1609.02907) 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
```