Skip to content
This repository was archived by the owner on Feb 21, 2025. It is now read-only.

Files

Latest commit

0b9df9d · Sep 28, 2022

History

History
This branch is 1690 commits behind dmlc/dgl:master.

geometric

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:

Image Classification on MNIST

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

Usage

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

Acknowledgement

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