mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-09-01 12:49:50 +08:00
add GCN implementation
This commit is contained in:
17
gcn/README.md
Normal file
17
gcn/README.md
Normal 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
|
||||
```
|
Reference in New Issue
Block a user