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
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
bf3994eea6
[Model][Hotfix] DiffPool formatting fix ( #696 )
...
* formatting
* formatting
2019-07-02 15:19:57 +08:00
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
714f2e1ced
fix node id dtype ( #674 )
2019-06-20 10:53:26 +08:00
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
f3d3fdf8cd
add reddit example. ( #658 )
2019-06-14 22:53:59 +08:00
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
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
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
08d4900fd1
Update README.md for the demo of distributed sampler ( #577 )
2019-05-28 17:24:33 +08:00
9aa5ffcab1
[DEMO] Update demo of distributed sampler ( #564 )
...
* update
* update
* update demo
2019-05-26 12:10:53 +08:00
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
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
f99725adbc
all demo use python-3 ( #555 )
2019-05-23 10:38:05 +08:00
d8c69d53e2
Fixing typo in JTNN after interface change ( #536 )
2019-05-16 09:43:03 +08:00
e89519150f
disable pytorch reddit ( #512 )
2019-04-28 11:24:14 +08:00
3b96299a45
sampling ( #505 )
2019-04-23 12:14:38 +08:00
ad9da36add
[Model] Add DGI Model ( #501 )
2019-04-22 11:37:06 +08:00
3f4645918d
[Model] Add edge dropout to APPNP ( #493 )
...
* [Model] Add edge dropout to APPNP
[Model] Add edge dropout to APPNP
* [Model] Refactor the sampling examples (#498 )
* reorganize sampling code.
* speedup gcn_ns.
* speed up gcn_cv
* fix graphsage_cv.
* undo the modification.
* accel training.
* update readme.
* [Model] Add edge dropout to APPNP
[Model] Add edge dropout to APPNP
update
2019-04-18 20:58:28 +08:00
a3febc061b
[Model] Add GIN Model ( #471 )
...
* add gin model
* convert dataset.py to data_ont_the_fly way and put it into dgl.data module
* convert dataset.py to data_ont_the_fly way and put it into dgl.data module
python code checked
* modified document and reference TUDataset; checked python part and bypass cpp part due to error
* change tensor to numpy in dataset and transform in collate@Dataloader
* Change minor format issue
Change minor format issue
* moved logging; adjusted tqdm etc
2019-04-17 14:57:07 +08:00
e2e04329a6
[BUG] Update URL for JTNN dataset
2019-04-11 22:23:34 +08:00
039a711da1
[Model] fix self-edge bug in GCN and GAT. ( #482 )
...
* Update gcn_mp.py
* Update train.py
* Update train.py
* Update train.py
* Update gat_batch.py
* Update gat_batch.py
* Update gcn_mp.py
2019-04-10 09:04:19 -04:00
fa887f69f4
[Model] Add APPNP model ( #480 )
...
* [Model] Add APPNP model
* update
* Revert "update"
This reverts commit a8e42d19c8b421fb382305a8f04d7426e78ea53c.
* update
* Update appnp.py
2019-04-09 19:54:09 +08:00
4ea42e3e08
fix gcn. ( #469 )
2019-04-05 10:56:09 -07:00
6066fee935
[Model][Feature] PinSage & Random Walk with Restart ( #453 )
...
* random walk traces generation
* remove outdated comments
* oops put in the wrong place
* explicit inline
* moving rand_r to util
* pinsage-like model on movielens
* the code runs now
* support cuda
* using readonly graph
* moving random walk to public function
* per-thread seed and openmp support
* pinsage-like model on movielens
* the code runs now
* support cuda
* using readonly graph
* using C random walk
* removing profile decorators
* param initialization
* no grad
* leaky relu fixes everything
* train and save
* WIP
* WIP
* WIP
* seems to work
* evaluation output
* swapping order of val/test and train
* debug
* hyperparam tuning
* prior/training dataset split changes
* random walk reorg
* random walk with restart
* signed comparison fix
* migrating random walk to nodeflow
* Revert "migrating random walk to nodeflow"
This reverts commit f2565347cced7c912a58a529b257c033d9f375b7.
* add README and remove dataset
* new endpoint
* lint
* lint x2
* oops forgot test
* including bpr - better for baseline
* addressing fixes
* throwing random walks out from SamplerOp class
* forgot to move RandomWalk; why did this even work?
* removing legacy garbage
* add todo
* address comments
* stupid bug fix
* call ndarrayvector converter to handle traces
2019-03-29 13:01:18 +08:00
2dff1aba83
[Bugfix] Fix shape of test_relation tensor in RGCN ( #464 )
2019-03-28 14:34:07 -04:00
d3c24cc25b
[BugFix] Fix bug in RGCN data processing and use index_select to improve speed ( #429 )
...
* use index_select instead of __getitem__
* fix bug in dataset processing
* fix edge_type shape bug
* comments
2019-03-04 10:52:21 -05:00
565f0c88fc
[WIP] [NN] Refactor NN package ( #406 )
...
* refactor graph conv
* docs & tests
* fix lint
* fix lint
* fix lint
* fix lint script
* fix lint
* Update
* Style fix
* Fix style
* Fix style
* Fix gpu case
* Fix for gpu case
* Hotfix edgesoftmax docs
* Handle repeated features
* Add docstring
* Set default arguments
* Remove dropout from nn.conv
* Fix
* add util fn for renaming
* revert gcn_spmv.py
* mx folder
* fix wierd bug
* fix mx
* fix lint
2019-02-25 18:41:21 -05:00
8c750170e8
[Model][Pytorch] GraphSAGE ( #403 )
...
* simple implemention of GraphSAGE
* update the abstract method and `torch.nn` modules are more utilized
2019-02-22 15:48:53 -05:00
cdbeb17f2d
[Model] Simplifying Graph Convolutional Networks. ( #405 )
...
* [Model] Simplifying Graph Convolutional Networks.
* - improve code stype (use nn.Linear)
- use `bias` option
- update hyperparameter and corresponding performance
2019-02-22 15:17:26 +08:00
788d8dd4ad
[Model]PPI dataloader and inductive learning script. ( #395 )
...
* Create ppi.py
* Create train_ppi.py
* Update train_ppi.py
* Update train_ppi.py
* Create gat.py
* Update train.py
* Update train_ppi.py
* Update ppi.py
* Update train_ppi.py
* Update ppi.py
* Update train_ppi.py
* Update train_ppi.py
* Update ppi.py
* Update train_ppi.py
* update docs and readme
2019-02-17 16:41:30 -05:00
1ea0bcf486
[Model] fix link & beam search ( #394 )
2019-02-17 13:32:36 -05:00
ae1806f651
[Bugfix] RGCN tutorial ( #388 )
...
* fix rgcn tutorial
* small fix
* upd
2019-02-15 11:14:56 +08:00
91b7382378
[Model] update gat ( #390 )
...
* update gat: add minus max for softmax
* small fix
2019-02-13 14:53:41 -05:00
6c3dba867a
[Tutorial] Update prerequisites of README ( #380 )
...
* Update prerequisites of README
* dependencies for pytorch models
* dependencies for mxnet models
* minor
2019-02-11 22:58:17 -05:00
29dd22e666
[Model] Support Multi-GPU for Transformer model ( #356 )
...
* multi-process version of transformer
* lots of fix
* fix bugs and accum gradients for multiple batches
* many fixes
* minor
* upd
* set torch device
* fix bugs
* fix and minor
* comments and clean up
* uncomment viz code
2019-02-11 20:17:45 -05:00
12d706300c
[Hotfix] fixing zero shaped tensor problems for PyTorch 1.0.0 in JTNN example ( #371 )
2019-01-31 21:07:09 -05:00
ddc2faa547
[Bugfix] Fix gat residual bug ( #355 )
...
* fix gat residual bug
* fix the residual addition; output heads; add some shape notations;
* minor
* fix the output head average
* add requests package in requirement
2019-01-12 08:37:01 -05:00
efae0f97d8
[Model] Improve GAT models ( #348 )
...
* two better GAT implementations
* update numbers
* use version switch for spmm
* add missing dropout and output heads
2019-01-11 02:02:06 -05:00
707334ce85
minor spelling tweaks ( #349 )
...
* minor spelling tweaks
* Update CONTRIBUTORS.md
2019-01-10 12:02:31 -05:00
1e50cd2eda
[Model][MXNet] MXNet Tree LSTM example ( #279 )
...
* TreeLSTM MXNet example
* hybridize
* add glove download
* usability
* Update README.md
2018-12-27 10:57:23 +08:00
896dc50e70
[Model] fix bugs in pygat and mxgat. ( #323 )
...
* mxgat
* pygat
* Update gat_batch.py
* add train
* Update gat_batch.py
* Update gat.py
* Update gat_batch.py
2018-12-23 20:45:25 -05:00
f35ac5443e
[Graph][Model] Cache adj & inc; MX GCN ( #307 )
...
* add cache to adj and incmat
* Fix bug in cached adj/inc
* mx gcn spmv runnable; acc debugging...
* fix bug in mx gcn that loss is not correctly calculated
* fix mx utest
* fix as requested
* use raw parameter tensors rather than dense layer
* fix dropout
* Add numbers in readme
2018-12-16 22:51:01 -05:00
cffa4034f5
[Model] fix GCN ( #305 )
...
* mxnet gcn spmv
* update readme
* fix gcn
* pytorch gcn
* update readme
2018-12-16 12:44:41 -05:00
9f32554296
[Model]Transformer ( #186 )
...
* change the signature of node/edge filter
* upd filter
* Support multi-dimension node feature in SPMV
* push transformer
* remove some experimental settings
* stable version
* hotfix
* upd tutorial
* upd README
* merge
* remove redundency
* remove tqdm
* several changes
* Refactor
* Refactor
* tutorial train
* fixed a bug
* fixed perf issue
* upd
* change dir
* move un-related to contrib
* tutuorial code
* remove redundency
* upd
* upd
* upd
* upd
* improve viz
* universal done
* halt norm
* fixed a bug
* add draw graph
* fixed several bugs
* remove dependency on core
* upd format of README
* trigger
* trigger
* upd viz
* trigger
* add transformer tutorial
* fix tutorial
* fix readme
* small fix on tutorials
* url fix in readme
* fixed func link
* upd
2018-12-07 15:22:46 +08:00
3564fdc5f1
[Bugfix][Model] fix treelstm model ( #274 )
...
* fix bug after moving batcher out of dgl.data
* disable mx utest
2018-12-06 23:52:50 -05:00
ddf96ff9e7
[Doc] Fix DGMG part in README ( #270 )
...
* Fix DGMG epoch time
* Fix README for DGMG
* update
2018-12-06 10:54:07 -05:00
1bbc885b40
move batcher to examples. ( #269 )
...
* move pytorch code to examples.
* fix.
* fix tutorial
2018-12-06 06:45:27 -08:00
ac660f45ef
[Doc] Readme ( #265 )
...
* [Doc] updated readme
* contribution guide change
* small edits
* add perf #
* Update README.md
* Update README.md
* add others...
* rewrite scalability
* model & code links
* some note for treelstm
* rewrite scalability
2018-12-06 00:31:04 -05:00
Xiagkun Hu