mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-06-26 10:41:18 +08:00
18 lines
385 B
Markdown
18 lines
385 B
Markdown
|
# 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
|
||
|
|
```
|