项目文件夹

文件

T

文件历史

Hongzhi (Steve), Chen 704bcaf6dd examples (#5323 )

Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>

2023-02-19 08:35:15 +08:00

.gitignore

2019-11-03 17:29:42 -05:00

coarsening.py

examples (#5323 )

2023-02-19 08:35:15 +08:00

coordinate.py

2022-09-28 10:28:24 +08:00

grid_graph.py

2022-09-28 10:28:24 +08:00

mnist.py

examples (#5323 )

2023-02-19 08:35:15 +08:00

README.md

2019-11-03 17:29:42 -05:00

Geometric Deep Learning models

This example shows how to use geometric deep learning models defined in dgl.nn.pytorch.conv for graph classification.

Currently we support following models:

By transforming images to graphs, graph classifcation algorithms could be applied to image classification problems.

python mnist.py --model cheb --gpu 0
python mnist.py --model monet --gpu 0

We thank Xavier Bresson for providing code for graph coarsening algorithm and grid graph building in
CE7454_2019 Labs.