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
6124667f42
[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.
2019-04-17 01:31:14 -07: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
b89dcce16c
[RPC] Refactoring networking APIs ( #496 )
...
* Refactoring network API
* update demo
* update
* update demo
* update demo
* add num_sender
* update
* fix lint
* fix lint
* fix lint
* update
2019-04-16 16:33:29 +08:00
688a9228a8
fix. ( #491 )
2019-04-11 15:38:35 -07: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
da3ab84c3c
Add demo for distributed sampler ( #474 )
...
* add C++ rpc infrastructure and distributed sampler
* update
* update lint
* update lint
* update lint
* update
* update
* update
* updare
* update
* update
* update
* update serialize and unittest
* update serialize
* lint
* update
* update
* update
* update
* update
* update
* update unittest
* put Finalize() to __del__
* update unittest
* update
* delete buffer in Finalize
* update unittest
* update unittest
* update unittest
* update unittest
* update
* update
* fix small bug
* windows socket impl
* update API
* fix bug in serialize
* fix bug in serialzie
* set parent graph
* update
* update
* update
* update
* update
* update
* fix lint
* fix lint
* fix
* fix windows compilation error
* fix windows error
* change API to lower-case
* update test
* fix typo
* update
* add SamplerPool
* add SamplerPool
* update
* update test
* update
* update
* update
* update
* add example
* update
* update
* add distributed sampler demo
* add index
* update demo of distributed sampler
* fix lower-case
* print subg index
* update README.md
* update
* remove --gpu args
2019-04-03 16:09:59 -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
ca2a7e1ca1
[Refactor] Nodeflow, sampling, CAPI ( #430 )
...
* enable cython
* add helper function and data structure for void_p vector return
* move sampler from graph index to contrib.sampling
* WIP
* WIP
* refactor layer sampling
* pass tests
* fix lint
* fix graphsage
* remove comments
* pickle test
* fix comments
* update dev guide for cython build
2019-03-05 14:07:22 -05: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
bea07b41b3
[Sampler] fix the API of neighbor sampler. ( #407 )
...
* don't return aux_info.
* fix sampler test.
* fix sse.
* fix.
* add comment.
2019-02-28 13:45:21 -08:00
7e30382e4f
[Model][MXNet] neighbor sampling & skip connection & control variate & graphsage ( #322 )
...
* neighbor sampling draft
* val/test acc
* control variate draft
* control variate
* update
* fix new_history
* maintain aggregated history while updating new history
* preprocess the first layer, change push to pull
* update
* fix subg_degree
* nodeflow
* clear
* readme
* doc and unittest for self loop
* address comments
* rename
* update
* fix
* Update node_flow.py
* Update node_flow.py
2019-02-28 13:29:51 -08: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
f370e628cd
[Feature] add NodeFlow API ( #361 )
...
* sample layer subgraphs.
* fix.
* fix.
* add layered subgraph.
* fix lint.
* fix.
* fix tutorial.
* fix.
* remove copy_to_parent.
* add num_layers
* move sampling code to sampler.cc
* fix.
* move subgraph construction out.
* Revert "move subgraph construction out."
This reverts commit 24b3d13b0d8ed5f91847ea75a7674ee8f7d45cff.
* change to NodeFlow.
* use NodeFlow in Python.
* use NodeFlowIndex.
* add node_mapping and edge_mapping.
* remove unnecessary code in SSE tutorial.
* Revert "remove unnecessary code in SSE tutorial."
This reverts commit 093f0413d5fa2e63ca5f80c46c80a126a9fb720c.
* fix tutorial.
* move to node_flow.
* update gcn cv updater.
* import NodeFlow.
* update.
* add demo code for vanilla control variate sampler.
* update.
* update.
* add neighbor sampling.
* return flow offsets.
* update node_flow.
* add test.
* fix sampler.
* fix graph index.
* fix a bug in sampler.
* fix map_to_layer_nid and map_to_flow_eid.
* fix apply_flow.
* remove model code.
* implement flow_compute.
* fix a bug.
* reverse the csr physically.
* add mini-batch test.
* add mini batch test.
* update flow_compute.
* add prop_flows
* run on specific nodes.
* test copy
* fix a bug in creating frame in NodeFlow.
* add init gcn_cv_updater.
* fix a minor bug.
* fix gcn_cv_updater.
* fix a bug.
* fix a bug in NodeFlow.
* use new h in gcn_cv_updater.
* add layer_in_degree and layer_out_degree.
* fix gcn_cv_updater for gpu.
* temp fix in NodeFlow for diff context.
* allow enabling/disabling copy back.
* add with-updater option.
* fix a bug in computing degree.
* add with-cv option.
* rename and add comments.
* fix lint complain.
* fix lint.
* avoid assert.
* remove assert.
* fix.
* fix.
* fix.
* fix.
* fix the methods in NodeFlow.
* fix lint.
* update SSE.
* remove gcn_cv_updater.
* correct comments for the schedulers.
* update comment.
* add map_to_nodeflow_nid
* address comment.
* remove duplicated test.
* fix int.
* fix comments.
* fix lint
* fix.
* replace subgraph with NodeFlow.
* move view.
* address comments.
* fix lint.
* fix lint.
* remove static_cast.
* fix docstring.
* fix comments.
* break SampleSubgraph.
* move neighbor sampling to sampler.cc
* fix comments.
* rename.
* split neighbor_list.
* address comments.
* fix.
* remove TODO.
2019-02-19 15:49:28 -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
9cd0df7dde
[Doc] Update README in MXNet GCN. ( #373 )
...
* update readme.
* Revert "update readme."
This reverts commit cd5ea512059adadfcb22a615a2c42bba12f334a8.
* update.
2019-02-02 21:59:07 -08: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
632d598c77
[Doc] Add readme to mxnet gat ( #300 )
...
* readme
* Update and rename README to README.md
* Update README.md
* Update README.md
* Update README.md
* Update README.md
* Update README.md
* Update README.md
* Update README.md
* Update README.md
2018-12-17 04:36:26 -08:00
f8811c7d3c
[BUGFIX] fix some minor problems in GAT ( #308 )
...
* fix gat.
* fix context.
2018-12-16 22:20:27 -08:00
dba36c87af
Fix some minor problems in SSE ( #309 )
...
* fix SSE.
* fix.
* fix.
2018-12-16 21:21:35 -08: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
f4a9a455f5
[Doc] Fix hyperlink in tutorial of tutorals ( #260 )
...
* fix all code tutorial links and typos in texts
* sse readme format
* fix
* sse paper link
* gat readme
* fix
2018-12-05 15:04:27 -05:00
e557ed89ff
[Doc][Model] example folder doc and model summary ( #259 )
...
* update example readme
* mx example readme
* add results in readme
* mx rgcn readme
2018-12-05 14:35:14 -05:00
2c5b48ab60
[Model][MXNet] RGCN Entity Classification ( #246 )
...
* entity classify work for examples
* add loop_msg
* remove wrong assert
* remove one reshape
* add readme
* add MRR
* remove mrr from entity task
2018-12-05 10:27:59 -08:00
7c7cc7e0c2
[sampler] Adjust the sampler API for the future extension. ( #243 )
...
* return seed ids.
* fix tests.
* implement.
2018-12-05 10:05:10 -08:00
378c264561
[Model] Fix GCN Normalization ( #249 )
...
* WIP
* lr -> 0.01
* new cora dataset
* normalization code
* minor format change
* normalization factor for deg bucket
2018-12-05 12:32:10 -05:00
21255b6528
[Bugfix] tolist and dependencies in dgl.data ( #239 )
...
* change Index.tolist -> Index.tonumpy; fix bug in traversal; remove dependencies in data
* fix import
* fix __all__ and some docstring
2018-12-05 12:20:59 -05:00
a5a35d1f7a
[Sampler] improve random shuffle performance in sampler ( #228 )
...
* fix.
* make it generic.
* add the API.
* fix.
* remove mxnet.
2018-12-04 09:24:32 -08:00
Aymen Waheb