文件历史

842 次代码提交

作者 SHA1 备注 提交日期
Quan Gan 10009cc1d5 update version 2021-04-07 08:35:50 +00:00
Quan (Andy) Gan 48b9ecd201 Fix discrepancy in dgl.dataloading.EdgeDataLoader (#2819)
* fix discrepancy

* address comments
2021-04-07 16:08:14 +08:00
Quan (Andy) Gan 3301bd0bcc bump nightly to 0.7 (#2814) 2021-04-06 13:51:26 +08:00
Quan (Andy) Gan cfe6e70b7a RGAT baseline for OGB-LSC node classification challenge (#2810)
* RGAT baseline

* update links (data.dgl.ai hasn't refreshed yet)

* nvm still use dgl.ai domain

* descriptions

* actually it took less

* address comments

* stop worrying about cache

* update links

* oops
2021-04-05 18:21:18 +08:00
Xiangkun Hu b49cd0952f [Bugfix] Fix feature type of 'ENZYMES' in TUDataset (#2800)
* PPIDataset

* Revert "PPIDataset"

This reverts commit 264bd0c960cfa698a7bb946dad132bf52c2d0c8a.

* Hide implementations that may cause confusions to the user

* fix number of classes in Amazon Cobuy datasets

* add four papers

* fix feature type of TU dataset

* fix readme

* remove commented codes

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-04-01 14:36:55 +08:00
milesial 1f4c0b7193 Fix processing of QM9EdgeDataset (#2801) 2021-04-01 10:13:51 +08:00
Da Zheng e36c5db614 [Distributed] Simplify distributed API (#2775)
* remove num_workers.

* remove num_workers.

* remove num_workers.

* remove num-servers.

* update error message.

* update docstring.

* fix docs.

* fix tests.

* fix test.

* fix.

* print messages in test.

* fix.

* fix test.

* fix.

Co-authored-by: Ubuntu <ubuntu@ip-172-31-9-132.us-west-1.compute.internal>
2021-03-29 19:19:23 -07:00
Da Zheng 97863ab85b [Distributed] Cleanup the process pool correctly when the process exits. (#2781)
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
2021-03-29 14:53:51 -07:00
Mufei Li 2952f3c473 Update subgraph.py (#2784)
Co-authored-by: Zihao Ye <expye@outlook.com>
2021-03-29 14:58:39 +08:00
Jinjing Zhou 0576a33b7f Update convert.py (#2789)
Co-authored-by: Zihao Ye <expye@outlook.com>
2021-03-28 20:59:52 +08:00
Konstantin Sofeikov e22087aa36 [BUG] fix for zero degree nodes (#2786)
* fix for zero degree nodes

* Update python/dgl/ops/spmm.py

linting fix

Co-authored-by: Zihao Ye <zihaoye.cs@gmail.com>

Co-authored-by: Konstantin Sofeikov <konstantin.sofeikov@complyadvantage.com>
Co-authored-by: Zihao Ye <expye@outlook.com>
Co-authored-by: Zihao Ye <zihaoye.cs@gmail.com>
2021-03-25 22:09:53 +08:00
kyawlinoo 03482f0a4c [NN] tf nn for edgeConv (#2741)
* tf nn for edgeConv

* Auto stash before merge of "tf_working" and "origin/tf_working"

* clean up

* added test for edge_conv

* fix

* fix

* fix

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Quan Gan <coin2028@hotmail.com>
2021-03-25 13:44:07 +08:00
Quan (Andy) Gan 929d863447 [Feature] Sparse-sparse matrix multiplication, addition, and masking (#2753)
* test

* more stuff

* add test

* fixes

* optimize algo

* replace unordered_map with arrays

* lint

* lint x2

* oops

* disable gpu csrmm tests

* remove gpu invocation

* optimize with openmp

* remove python functions

* add back with docstrings

* lint

* lint

* update python interface

* functionize

* functionize

* lint

* lint
2021-03-24 17:24:41 +08:00
Tianqi Zhang (张天启) 6999f88f3f [Model] Add random_start option for Farthest Point Sampler (#2755)
* add random_start option for FPS

* change doc

* change from random_start to start_idx

* change error condition

* change error msg

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-03-22 21:53:46 +08:00
Hengrui Zhang bcffdb82c9 [Example] Add implementation of mvgrl (#2739)
* [Example ]add mvgrl

* [Doc] add mvgrl to readme

* add more comments

* fix typos

* replace tab with space

* [doc] replace tab with space

* [Doc] fix a typo

* fix minor typos

* fix typos

* fix typos

* fix typos

* fix typos

* fix

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2021-03-22 21:20:21 +08:00
Hengrui Zhang 1789972c4b [Bugfix] add default value for BlockSampler (#2771)
* [Bugfix] add default value for BlockSampler

* [Doc] modify user_guide description about MultiLayerDropoutSample

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-03-22 14:02:27 +08:00
Zihao Ye edf644633d [performance] Optimize the association order of AXW in GraphSAGE. (#2747)
* upd

* lint

* upd

* upd

* compatibility

* upd

* upd

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-03-19 01:41:33 +08:00
Jinjing Zhou 366cc7ebf3 [Pickle] Fix HeteroGraphConv pickle problem (#2761)
* fix pickle problem

* lint

* add pickle tests

* fix

* fix

* fix

* fix

* fix for windows
2021-03-18 17:56:28 +08:00
Mufei Li 09ef2c2c3d [Doc] Add Pseudo Code for HeteroGraphConv (#2729)
* Update

* Update

* Update

* Update

Co-authored-by: Ubuntu <ubuntu@ip-172-31-15-8.us-west-2.compute.internal>
2021-03-17 20:00:15 +08:00
Mufei Li b34a1e63d4 Update subgraph.py (#2754) 2021-03-17 18:15:30 +08:00
鄢振宇Michael Yan 3c8382f6b6 Fix typos in zero-in-degree (#2750) 2021-03-14 19:13:59 +08:00
xnouhz 0b47e86803 [Example] Add DimeNet(++) for Molecular Graph Property Prediction (#2706)
* [example] arma

* update

* update

* update

* update

* update

* [example] dimenet

* [docs] update dimenet

* [docs] update tf results

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2021-03-09 17:08:59 +08:00
Tianqi Zhang (张天启) c88fca5055 [Feature] Add edge coarsening for homogeneous undirected graphs (#2691)
* finish graph matching gpu version

* use C++ shuffle

* finish graph matching

* fix bug

* fix bug

* change name and use swap

* upt

* fix format problem

* fix format problem

* stronger test

* upt

* upt

* change python api

* upt

* upt

* format check

* upt

* upt

* fix bug

Co-authored-by: Tong He <hetong007@gmail.com>
2021-03-09 16:35:27 +08:00
Hengrui Zhang a1f59c3b68 [Feature] QM9Edge Dataset Support (#2704)
* [Feature] Support QM9Edge Datset

* Update qm9_edge.py

* disable tqdm

* Update qm9_edge.py

* Update qm9_edge.py

* Update qm9_edge.py

* Update qm9_edge.py

* Update qm9_edge.py

* Update qm9_edge.py

* Update qm9_edge.py

* remove preprocessing part

* add comparisons in qm9.py

* [docs] add qm9edge dataset

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2021-03-09 14:51:21 +08:00
Mufei Li 7380d61eb9 [NN] Support Unidirectional Bipartite Graphs in CFConv (#2674)
* Update

* update

* Update

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-03-03 16:35:07 +08:00
Da Zheng d0638b1ea9 [Example] Update the example of distributed GNN training of RGCN (#2709)
* support gpu training.

* remove unnecessary arguments.

* update README.

* update time measurement.

* add zero_grad.

Co-authored-by: Ubuntu <ubuntu@ip-172-31-9-132.us-west-1.compute.internal>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-03-01 12:00:15 -08:00
Padarn Wilson 062f59a8f6 fix small typo bug (#2708)
Co-authored-by: Zihao Ye <expye@outlook.com>
2021-02-28 17:37:13 +08:00
Quan (Andy) Gan 99751d4960 [Doc] Rename block to message flow graph (#2702)
* rename block to mfg

* revert

* rename
2021-02-25 18:26:19 +08:00
Ligeng Zhu 7ab3d993a4 Change the absolute path to relative one in part_config.json (#2690)
* Update partition.py

* fix lint.

Co-authored-by: Da Zheng <zhengda1936@gmail.com>
2021-02-24 16:12:19 -08:00
Quan (Andy) Gan 1686106390 change __repr__ (#2696)
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-02-25 00:21:00 +08:00
Quan (Andy) Gan 71d53d9eef add thumbnails for tutorials (#2695)
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-02-23 20:33:35 +08:00
Tong He 7298173ac1 [Doc]Improve docstring for GINDataset (#2676)
* add docstring for gin

* update doc

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2021-02-23 15:14:10 +08:00
Zihao Ye 174dd18905 [hotfix] A bunch of fix for SetTransformer (#2658)
* upd

* lint

* upd

* fix

* fix

* upd

* upd

* fix

* warning

* upd

* upd
2021-02-23 14:19:51 +08:00
xiang song(charlie.song) 80c26877de [Bug Fix] Fix munmap bug in sparse optimizer (#2675)
* Fix munmap bug

* lint

* update

Co-authored-by: Ubuntu <ubuntu@ip-172-31-56-220.ec2.internal>
2021-02-22 11:33:44 +08:00
xiang song(charlie.song) 16e324daf3 [Bug Fix] Add warning for invalid input graph of edge UDF (#2682)
* Add warning for invalid graph of edge UDF

* lint

Co-authored-by: Ubuntu <ubuntu@ip-172-31-56-220.ec2.internal>
2021-02-22 10:12:40 +08:00
Zihao Ye 72a16b22e4 [bugfix] Fix the behavior of select_topk (#2688)
* fix

* lint

* lint
2021-02-21 18:52:35 +08:00
Zihao Ye e6bf54cdcb [Feature] Support aggregate multiple edge features in to_simple. (#2623)
* upd

* fix

* upd

* upd

* upd

* upd

* upd

* upd

* upd

* upd

* upd

* fix

* refactor

* upd test

* large feat_len or n in segment reduce

* lint
2021-02-21 10:05:42 +08:00
Minjie Wang 8a07ab7737 [Doc] Tutorials re-organization (#2683)
* reorg

* change titles

* rm some stale API doc; minor fix

* fix docs

* add warning

* rm new-tutorial run in ci

* lint
2021-02-20 17:42:45 +08:00
Da Zheng 0fc649523b [Doc] Distributed heterogeneous graph training (#2654)
* update DistGraph docstrings.

* add user guide.

* add doc string.

* fix.

* fix.

* fix.

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-02-19 10:48:05 -08:00
xiang song(charlie.song) 453d358d2c [Doc] Fix docs for sparse optimizer (#2680)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-56-220.ec2.internal>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-02-19 14:07:35 +08:00
Quan (Andy) Gan dda103d9f6 Fix bug in EdgeConv implementation (#2669)
* Update edgeconv.py

* update edgeconv to builtin

* add explanation on why ReLU is not there

* remove redundant code
2021-02-19 13:07:21 +08:00
Quan (Andy) Gan d93a975997 revert (#2672) 2021-02-19 12:05:42 +08:00
Zihao Ye 0cbaa1fbcb [Doc] Make the docstring of nn.glob more informative. (#2649)
* upd

* fix weight and sum

* upd sortpool
2021-02-12 12:29:22 +08:00
Quan (Andy) Gan 9e630101a1 fix and lots of tests (#2650) 2021-02-12 11:40:05 +08:00
Zihao Ye cf8a3fb305 [hotfix] Address the performance issue of topk. (#2628)
* upd

* fix

* upd

* upd

* udp

* upd

* upd

* upd

* upd

* upd

* upd

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-02-11 00:00:39 +08:00
Tong He 8ccb8206bf [Doc] Update docs to motivate the use of set_batch_num_nodes and set_batch_num_edges (#2630)
* fix doc

* update docstring for transformation on batch graph

* update cn guide

* fix format

* update rst
2021-02-08 14:43:24 +08:00
nv-dlasalle bc3a532f5e [Sampling] Implement dgl.to_block() for the GPU (#2339)
* Add start of to_block gpu implementation

* Pull in more changes from 0.4.2 cuda_to_block

* Move more code to IdArray

* Refactor DeviceNodeMapMaker

* Updates

* get compiling

* Integrate to_block

* Fix ID allocation

* Minor fixes

* Cleanup cuda calls to use cuda_common

* Reduce kernel calls

* Lint cleanup

* Expand documentation

* Remove unused function

* Rename variables for consistency

* Add doxygen comments

* Fix file extension

* Remove raw asynccopy for deviceapi

* Remove unused function

* Fix block/tile configuration

* Add cuda_device_common.cuh

* Add basic hashtable

* Migrate part of hashtable

* Refactor to use external hashtable

* Make functions members

* Format hash table functions

* Migrate duplicate filling

* Move last function over

* Refactor with cu file

* lint c++ code

* Move context check to C++ code

* Use macro switch

* Add missing files

* Update docstring

* update docs

* Move atomic functions

* Refactor hashtable

* Fix linting

* Expand docs

* Fix mismatched argument names

* Switch doxygen comments from using @param to \param

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-02-08 13:00:43 +08:00
Quan (Andy) Gan 98ac391ed9 [Doc] Large graph training tutorials (#2595)
* large graph training tutorials

* add full graph link prediction update

* fix

* fix

* fix

* addressed comments
2021-02-07 17:55:49 +08:00
Jinjing Zhou 0346b0aa80 [Test] Add test for Cython and force rebuild when run setup.py (#2635) 2021-02-05 22:38:46 +08:00
Mufei Li 3ca52b1e59 [Fix] Various Fix after Bug Bash (#2620)
* update

* Misc fix

* Update
2021-02-05 17:34:16 +08:00