dmlc--dgl
69eae6de49
* Use ==/!= to compare constant literals (str, bytes, int, float, tuple) Avoid Syntax Warnings on Python >= 3.8 $ `python3` ``` >>> "" == "" True >>> "" is "" <stdin>:1: SyntaxWarning: "is" with a literal. Did you mean "=="? True ``` * Use ==/!= to compare constant literals (str, bytes, int, float, tuple)
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.