mlx-examples/gcn/README.md

18 lines
372 B
Markdown
Raw Normal View History

2023-12-12 00:48:07 +08:00
# Graph Convolutional Network
2023-12-12 06:10:46 +08:00
An example of [GCN](https://arxiv.org/abs/1609.02907) implementation with MLX.
2023-12-12 00:48:07 +08:00
### 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
```