项目文件夹

文件

T

Zihao Ye 9a0511c8e9 [NN] nn modules & examples update (#890 )

* upd

* damn it

* fuck

* fuck pylint

* fudge

* remove some comments about MXNet

* upd

* upd

* damn it

* damn it

* fuck

* fuck

* upd

* upd

* pylint bastard

* upd

* upd

* upd

* upd

* upd

* upd

* upd

* upd

* upd

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

814 B

原始文件 永久链接 Blame 文件历史

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:

ChebNet
MoNet

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.

814 B 原始文件 永久链接 Blame 文件历史

Geometric Deep Learning models

Image Classification on MNIST

Usage

Acknowledgement

814 B

原始文件永久链接 Blame 文件历史