文件历史

176 次代码提交

作者 SHA1 备注 提交日期
xiang song(charlie.song) e17add5602 [NN] Add MXNet impl for TAGCN module. (#799)
* upd

* fig edgebatch edges

* add test

* trigger

* Update README.md for pytorch PinSage example.

Add noting that the PinSage model example under
example/pytorch/recommendation only work with Python 3.6+
as its dataset loader depends on stanfordnlp package
which work only with Python 3.6+.

* Provid a frame agnostic API to test nn modules on both CPU and CUDA side.

1. make dgl.nn.xxx frame agnostic
2. make test.backend include dgl.nn modules
3. modify test_edge_softmax of test/mxnet/test_nn.py and
    test/pytorch/test_nn.py work on both CPU and GPU

* Fix style

* Delete unused code

* Make agnostic test only related to tests/backend

1. clear all agnostic related code in dgl.nn
2. make test_graph_conv agnostic to cpu/gpu

* Fix code style

* fix

* doc

* Make all test code under tests.mxnet/pytorch.test_nn.py
work on both CPU and GPU.

* Fix syntex

* Remove rand

* Add TAGCN nn.module and example

* Now tagcn can run on CPU.

* Add unitest for TGConv

* Fix style

* For pubmed dataset, using --lr=0.005 can achieve better acc

* Fix style

* Fix some descriptions

* trigger

* Fix doc

* Add nn.TGConv and example

* Fix bug

* Update data in mxnet.tagcn test acc.

* Fix some comments and code

* delete useless code

* Fix namming

* Fix bug

* Fix bug

* Add test code for mxnet TAGCov

* Update some docs

* Fix some code

* Update docs dgl.nn.mxnet

* Update weight init

* Fix
2019-08-28 13:19:17 +08:00
Quan (Andy) Gan 14bffe9728 [NN] Renaming NearestNeighborGraph to KNNGraph (#802)
* initial commit

* second commit

* another commit

* change docstring

* migrating to dgl.nn

* fixes

* docs

* lint

* multiple fixes

* doc

* renaming nearest neighbor graph
2019-08-28 12:20:35 +08:00
Quan (Andy) Gan dc19cd5687 [Example] Dynamic Graph CNN on Point Cloud (#789)
* initial commit

* second commit

* another commit

* change docstring

* migrating to dgl.nn

* fixes

* docs

* lint

* multiple fixes

* doc
2019-08-28 09:21:57 +08:00
Mufei Li e590feeb62 [Model Zoo] GAT on Tox21 (#793)
* GAT

* Fix mistake

* Fix

* hotfix

* Fix

* Fix

* Fix

* Fix

* Fix

* Fix

* Fix

* Update

* Update

* Update

* Fix style

* Hotfix

* Hotfix

* Hotfix

* Fix

* Fix

* Update

* CI trial

* Update

* Update

* Update
2019-08-28 04:47:16 +08:00
VoVAllen 3192beb42d [Model zoo] JTNN model zoo (#790)
* jtnn model zoo

* poke ci

* fix line sep

* fix

* Fix import order

* fix render

* fix render

* revert

* fix

* Resolve conflict

* dix

* remove create_var

* refactor

* fix

* refactor

* readme

* format

* fix lint

* fix lint

* pylint

* lint

* fix lint

* fix lint

* add hint

* fix

* Remove vocab

* Add explanation for warning

* add directory

* Load model to cpu by default

* Update
2019-08-28 04:04:54 +08:00
Mufei Li 331337fe5f [Model Zoo] Clean up for Alchemy dataset (#800)
* Make dataset framework agnostic

* Update

* Update

* Fix

* Remove unused import
2019-08-28 00:08:59 +08:00
Zihao Ye 9314aabd1f [Refactor] Interface of nn modules (#798)
* refactor

* upd mpnn
2019-08-27 22:29:25 +08:00
Zihao Ye 650f6ee1e0 [NN] Add commonly used GNN models from examples to dgl.nn modules. (#748)
* gat

* upd

* upd sage

* upd

* upd

* upd

* upd

* upd

* add gmmconv

* upd ggnn

* upd

* upd

* upd

* upd

* add citation examples

* add README

* fix cheb

* improve doc

* formula

* upd

* trigger

* lint

* lint

* upd

* add test for transform

* add test

* check

* upd

* improve doc

* shape check

* upd

* densechebconv, currently not correct (?)

* fix cheb

* fix

* upd

* upd sgc-reddit

* upd

* trigger
2019-08-27 18:21:19 +08:00
Mufei Li 8079d98617 Misc update (#796) 2019-08-27 16:08:36 +08:00
lunar 3bc73931f8 [Model Zoo] Molecule Regression (#779)
* [Model] MPNN

* [Model] MPNN 🔨 reorganize the mpnn/sch/mgcn model & alchemy dataset

* [Model] MPNN alchemy dataloader refactoring

* [Model] Chem model zoo minor change

* [Model] Chem Model Zoo 🔥 remove old samples

* [Model Zoo] molecule regression minor change

* Fix dataset import

* Fix dataset import

* [Model Zoo] molecule regression test set

* [Model Zoo] molecule prediction MPNN model hyperparameter tuning

* [Model Zoo] molecule prediction ✏️ mpnn performance update
2019-08-26 17:42:33 +08:00
Mufei Li 73b2668fe2 [Model Zoo] DGMG for molecule generation (#783)
* DGMG for molecule generation

* Fix CI check

* Fix for CI

* Trial for CI due to shared memory

* Update

* Better interface for dataset configuration

* Update

* Handle corner cases

* Update README

* Fix

* Fix

* Fix

* Fix

* Fix

* Refactor

* Fix

* Fix

* Fix

* Fix

* Fix

* Fix

* Update

* Fix

* Fix

* Fix

* Fix

* Fix

* Finallly
2019-08-26 03:39:20 +08:00
xiang song(charlie.song) 11fb217a76 [NN] Add TAGCN nn.module and example (#788)
* upd

* fig edgebatch edges

* add test

* trigger

* Update README.md for pytorch PinSage example.

Add noting that the PinSage model example under
example/pytorch/recommendation only work with Python 3.6+
as its dataset loader depends on stanfordnlp package
which work only with Python 3.6+.

* Provid a frame agnostic API to test nn modules on both CPU and CUDA side.

1. make dgl.nn.xxx frame agnostic
2. make test.backend include dgl.nn modules
3. modify test_edge_softmax of test/mxnet/test_nn.py and
    test/pytorch/test_nn.py work on both CPU and GPU

* Fix style

* Delete unused code

* Make agnostic test only related to tests/backend

1. clear all agnostic related code in dgl.nn
2. make test_graph_conv agnostic to cpu/gpu

* Fix code style

* fix

* doc

* Make all test code under tests.mxnet/pytorch.test_nn.py
work on both CPU and GPU.

* Fix syntex

* Remove rand

* Add TAGCN nn.module and example

* Now tagcn can run on CPU.

* Add unitest for TGConv

* Fix style

* For pubmed dataset, using --lr=0.005 can achieve better acc

* Fix style

* Fix some descriptions

* trigger

* Fix doc
2019-08-25 22:17:46 +08:00
Minjie Wang 708765f0a1 [NN] RGCN modules (#744)
* rgcn module

* support id input

* WIP: model codes

* use faster index select

* dropout

* self loop

* WIP: link prediction

* fix lint

* WIP: docs

* docstring

* docstring

* merge two child classes

* mxnet rgcn module

* fix lint

* fix lint

* fix rename bug

* add uniform edge sampler

* fix fn name

* docstring

* fix mxnet rgcn module

* fix mx rgcn

* enable test on cuda
2019-08-23 16:38:48 -04:00
VoVAllen 0f12763725 [Model] Early stop GAT (#750)
* Add early stop

* add mxnet version

* Poke ci
2019-08-21 14:58:56 +08:00
Sahand 77c58289a7 [Bugfix][Model] Fixing RGCN evaluation bug (#778) 2019-08-20 15:54:03 -04:00
Mufei Li 35bed2a9d0 Rename early stop (#773) 2019-08-18 13:46:44 +08:00
xiang song(charlie.song) 165d453806 [Doc] Update README.md for pytorch PinSage example. (#770)
Add noting that the PinSage model example under
example/pytorch/recommendation only work with Python 3.6+
as its dataset loader depends on stanfordnlp package
which work only with Python 3.6+.
2019-08-16 02:23:05 -03:00
Mufei Li cd9fb7baa0 [Model Zoo] Refactor GCN on Tox21 (#766)
* [Model zoo] Model zoo (#765)

* tox21

* fix ci

* fix ci

* fix urls to url

* add doc

* remove binary

* model zoo

* test

* markdown

* fix typo

* fix typo

* fix typo

* raise error

* fix lint

* remove unnecessary

* fix doc

* fix

* fix

* fix

* fix

* fix

* fix

* Update

* CI

* Fix

* Fix

* Fix

* Fix

* Fix

* CI
2019-08-16 02:49:25 +08:00
Xiagkun Hu b2f7f0ee7c [Model] Recurrent Relational Network (RRN) on sudoku (#733)
* rrn model and sudoku

* add README

* refine the code, add doc strings

* add sudoku solver
2019-08-10 02:00:11 +08:00
lunar e1f08644fb [Model] Schnet & MGCN (#726)
* [Model] SchNet

* [Model] SchNet model

* [Model] Schnet Model fix device-related bug

* [Model] SchNet fix bugs

* [Model] SchNet fix some bugs

* [Model] Schnet 🎨 code indent format

* [Model] SchNet ✏️ fix some typos
2019-08-07 14:08:31 +08:00
HQ bf3994eea6 [Model][Hotfix] DiffPool formatting fix (#696)
* formatting

* formatting
2019-07-02 15:19:57 +08:00
HQ 684a61ad09 [Model] DiffPool with both DGL and tensorized operatons (#665)
* diffpool original file added

* make diffpool fuse up and running

* minor tweak on tu dataset statistics method

* fix tu

* break

* delete break

* pre_org

* diffpool fuse reorg

* fix random shuffling

* fix bn

* add dgl layers

* early stopping

* add readme

* fix

* add diffpool preprocess script

* tweak tu dataset

* tweak

* tweak

* tweak

* tweak

* tweak

* preprocess dataset

* fix early stopping

* fix

* fix

* fix

* tweak

* readme

* code review

* code review

* dataset code review

* update README

* code review

* tu doc
2019-07-01 20:00:15 +08:00
Quan (Andy) Gan 714f2e1ced fix node id dtype (#674) 2019-06-20 10:53:26 +08:00
Lingfan Yu 82bb9a9b6a [README] Update speed numbers in README (#661)
* update speed numbers in readme

* numbers in examples/pytorch/readme
2019-06-15 14:27:58 -07:00
Tianyi f3d3fdf8cd add reddit example. (#658) 2019-06-14 22:53:59 +08:00
Da Zheng 94ecb8eb96 [BUGFIX] copy graph index to shared memory. (#634)
* copy graph index to shared memory.

* fix.

* fix.

* fix.

* use a diff name for in-csr and out-csr.

* fix lint.

* remove print.

* add test.

* add comments.
2019-06-12 01:32:25 -07:00
Da Zheng 411bcd7e6c [BUGFIX] extand DGLGraph graph_data. (#632)
* enable tutorial test in CI.

* extand DGLGraph graph_data.

* update doc.

* Revert "enable tutorial test in CI."

This reverts commit cd774067180922bb6ae979bde4aecbffc61c8147.

* accept DGLGraph in graph store.
2019-06-09 17:22:04 -07:00
Da Zheng fc7775a20f Define node/edge Ids in NodeFlow more clearly (#628)
* add tests.

* distinguish layer-local nid and nodeflow nid.

* use numpy assert_array_equal and assert_allclose

* fix map_from_parent_nid

* fix test

* fix test.

* renmae remap.

* update doc.

* update doc.

* update doc.

* fix test.

* fix test.
2019-06-09 12:10:07 -07:00
Minjie Wang dec8b49b5d [Examples] Run all the examples (#623)
* all pytorch examples

* scan through mxnet examples

* change reddit data

* tweak numerical range for unittest

* fix ci

* fix ci

* fix

* add seed to workaround
2019-06-09 00:26:13 -04:00
Lingfan Yu 74e13eea61 [Model] Update GAT model code (#622)
* fix gat code to use latest edge softmax module

* avoid transpose

* update README

* use edge_softmax op

* mxnet edge softmax op

* mxnet gat

* update README

* fix unittest

* fix ci

* fix mxnet nn test; relax criteria for prod reducer
2019-06-08 21:41:38 -04:00
Minjie Wang 01a4cc5b7b [Graph] Add API to convert graph to simple graph (#587)
* to simple

* WIP: multigraph flag

* graph index refactor; pass basic testing

* graph index refactor; pass basic testing

* fix bug in to_simple; pass torch test

* fix mx utest

* fix example

* fix lint

* fix ci

* poke ci

* poke ci

* WIP

* poke ci

* poke ci

* poke ci

* change ci workspace

* poke ci

* poke ci

* poke ci

* poke ci

* delete ci

* use enum for multigraph flag
2019-06-01 23:32:46 -04:00
Chao Ma e0ce190dfe [Doc] Update README.MD of demo for distributed sampler (#579)
* Update README.md for the demo of distributed sampler

* Update README.MD of demo for distributed sampler
2019-05-30 01:24:13 -04:00
Da Zheng e0004f7199 [Doc] sampler readme update. (#581) 2019-05-29 23:47:57 -04:00
Chao Ma 08d4900fd1 Update README.md for the demo of distributed sampler (#577) 2019-05-28 17:24:33 +08:00
Chao Ma 9aa5ffcab1 [DEMO] Update demo of distributed sampler (#564)
* update

* update

* update demo
2019-05-26 12:10:53 +08:00
Da Zheng 8f378d90ad [BUGFIX] fix bugs for running GCN on giant graphs. (#561)
* load mxnet csr.

* enable load large csr.

* fix

* fix.

* fix int overflow.

* fix test.
2019-05-25 11:09:33 -07:00
Chao Ma 86d60a1ff6 [DEMO] Add Pytorch demo for distributed sampler (#562)
* update

* update

* update

* add sender

* update

* remove duplicate cpde
2019-05-24 11:53:28 +08:00
Da Zheng 924efc6520 [Perf] Improve performance of graph store. (#554)
* fix.

* use inplace.

* move to shared memory graph store.

* fix.

* add more unit tests.

* fix.

* fix test.

* fix test.

* disable test.

* fix.
2019-05-23 12:23:31 -07:00
Chao Ma 49c4a9e4cc [DEMO] Remove duplicate code for sampling (#557)
* update

* update

* re-use single-machine code

* update

* use relative path

* update

* update

* update

* add __init__.py

* add __init__.py

* import sys, os

* fix typo

* update
2019-05-23 16:57:35 +08:00
Chao Ma 28379f927f [DEMO] Reproduce numbers of distributed training in AMLC giant graph paper (#556)
* update

* update

* update

* update num_hops

* fix bug

* update

* report numbers of distributed training in AMLC giant graph paper
2019-05-23 14:23:14 +08:00
Chao Ma f99725adbc all demo use python-3 (#555) 2019-05-23 10:38:05 +08:00
Da Zheng b2b8be25bb [API] update graph store API. (#549)
* add init_ndata and init_edata in DGLGraph.

* adjust SharedMemoryGraph API.

* print warning.

* fix comment.

* update example

* fix.

* fix examples.

* add unit tests.

* add comments.
2019-05-21 03:27:05 -07:00
Da Zheng cdfca992e1 [BUGFix] Improve multi-processing training (#526)
* fix.

* add comment.

* remove.

* temp fix.

* initialize for shared memory.

* fix graphsage.

* fix gcn.

* add more unit tests.

* add more tests.

* avoid creating shared-memory exclusively.

* redefine remote initializer.

* improve initializer.

* fix unit test.

* fix lint.

* fix lint.

* initialize data in the graph store server properly.

* fix test.

* fix test.

* fix test.

* small fix.

* add comments.

* cleanup server.

* test graph store with a random port.

* print.

* print to stderr.

* test1

* test2

* remove comment.

* adjust the initializer signature.
2019-05-20 14:23:19 -07:00
Quan (Andy) Gan d8c69d53e2 Fixing typo in JTNN after interface change (#536) 2019-05-16 09:43:03 +08:00
Da Zheng 3a1392e641 [Model] add multiprocessing training with sampling. (#484)
* reorganize sampling code.

* add multi-process training.

* speed up gcn_cv

* fix graphsage_cv.

* add new API in graph store.

* update barrier impl.

* support both local and distributed training.

* fix multiprocess train.

* fix.

* fix barrier.

* add script for loading data.

* multiprocessing sampling.

* accel training.

* replace pull with spmv for speedup.

* nodeflow copy from parent with context.

* enable GPU.

* fix a bug in graph store.

* enable multi-GPU training.

* fix lint.

* add comments.

* rename to run_store_server.py

* fix gcn_cv.

* fix a minor bug in sampler.

* handle error better in graph store.

* improve graphsage_cv for distributed mode.

* update README.

* fix.

* update.
2019-05-06 23:51:56 -07:00
Ziyue Huang e89519150f disable pytorch reddit (#512) 2019-04-28 11:24:14 +08:00
Chao Ma 2ff8ecfbac update test_batch_size (#510) 2019-04-25 13:28:36 +08:00
Ziyue Huang 3b96299a45 sampling (#505) 2019-04-23 12:14:38 +08:00
Zhengwei ad9da36add [Model] Add DGI Model (#501) 2019-04-22 11:37:06 +08:00
Chao Ma fe7d5e9b8b [Sampler] Change Distributed Sampler API (#499)
* Change Distributed Sampler API

* fix lint

* fix lint

* update demo

* update

* update

* update

* update demo

* update demo
2019-04-22 09:41:47 +08:00