文件历史

提交图

261 次代码提交

作者 SHA1 备注 提交日期
Mufei Li 5be937a7fb [Kernel] Slicing Batched Graphs (#2965)
* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Add files via upload

* Add files via upload

* Add files via upload

* Add files via upload

* Update

* Update

* Add files via upload

* Add files via upload

* Update

* Lint

* Add files via upload

* Lint

* Update

* Update

* Update

* Update

* Update

* Lint Fix

* Lint

Co-authored-by: Ubuntu <ubuntu@ip-172-31-12-161.us-west-2.compute.internal>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
2021-06-10 12:07:44 +08:00
Quan (Andy) Gan a7fe461c1c [Doc] Single-machine Multi-GPU node classification tutorial (#2976)
* multi-GPU node classification tutorial

* Update 2_node_classification.py

* fixes

* elaborate a bit more

* address comments

* address comments
2021-06-08 10:28:02 +08:00
Tong He 972a9f1323 [Doc] Re-organize the code for dgl.geometry, and expose it in the doc (#2982)
* reorg and expose dgl.geometry

* fix lint

* fix test

* fix
2021-06-07 11:19:39 +08:00
Kay Liu 895d6fff72 [Doc] changing of output dimension (#2981)
Co-authored-by: zhjwy9343 <6593865@qq.com>
2021-06-05 22:12:41 +08:00
Da Zheng b1628f2398 [Tutorial] Distributed node classification. (#2969)
* add init version.

* fix build.

* fix format.

* fix.

* fix.

* fix format.

* update README.

* avoid running CI on distributed training tutorials.

* Update tutorials/dist/1_node_classification.py

Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>

* fix.

Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
2021-06-04 21:50:16 +08:00
Kay Liu 6383e64937 [Doc] correct the matrix multiplication symbols (#2961)
* correct the matrix multiplication symbols

* modified dimensionality of linear

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-06-03 08:51:23 +08:00
Quan (Andy) Gan 2ae190d690 [Doc] fix shape mismatch problem in user guide (#2938)
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-06-02 12:18:10 +08:00
Mufei Li 09642d7c21 [Tutorial] Single Machine Multi-GPU Minibatch Graph Classification (#2940)
* Update

* Update

* Update tutorials/multi/1_graph_classification.py

Co-authored-by: Tong He <hetong007@gmail.com>

* Update

* Update

* Update

Co-authored-by: Ubuntu <ubuntu@ip-172-31-4-21.us-west-2.compute.internal>
Co-authored-by: Tong He <hetong007@gmail.com>
2021-05-31 18:36:34 +08:00
Da Zheng 6e7f19f27b [Distributed] add distributed in-degree and out-degree. (#2918)
* add distributed in-degree and out-degree.

* update comments.

* fix a bug.

* add tests.

* add tests.

* fix a bug.

* fix docstring.

* update doc.

* fix

* fix.

Co-authored-by: Zheng <dzzhen@3c22fba32af5.ant.amazon.com>
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
2021-05-18 19:46:09 +08:00
Quan (Andy) Gan 657c220de7 [Feature] Python interface for adjacency matrix summation and multiplication (#2893)
* test commit

* fixes

* oops

* add docs

* lint

* why does it say I have a trailing whitespace

* oh ok

* fixes

* why there's an invalid argument error

* address comments

* fix

* address comments
2021-05-17 19:47:02 +08:00
kyawlinoo b03077b60d [NN] GCNII model added <https://arxiv.org/pdf/2007.02133.pdf> (#2874)
* Auto stash before rebase of "origin/pytorch-nn-working"

GCNII model added

* linting

* linting

* lint

* Frequency Adaptive gcn init comit

* Revert "Frequency Adaptive gcn init comit"

This reverts commit 86a80586ac0040497c1edfa0e80df719992dcc4a.

* Update python/dgl/nn/pytorch/conv/gcn2conv.py

modified docstring

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>

* added beta formula and changed param name

* fix docstring

* lint

* white space lint

* update docstring

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>

* docstring formula update

* added gcn2

* added GCN2Conv

* Update nn.pytorch.rst

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-05-17 10:25:59 +08:00
Quan (Andy) Gan caa6d6072c [Feature] Replacing thread_wrapped_func with minimal mp.Process wrapper (#2905)
* standardizing thread_wrapped_func

* lints

* Update __init__.py
2021-05-14 11:34:49 +08:00
Da Zheng a90296aa9c fix. (#2907)
Co-authored-by: Zheng <dzzhen@3c22fba32af5.ant.amazon.com>
2021-05-13 16:55:31 +08:00
YYY c18f957d0e [Model] add implementation of twirls (#2770)
* add implementation of twirls

* format the code

* fix some format error, and ignore others

* fix format errors

* fix format errors

* expose unfolding & attention

* Update nn.pytorch.rst

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-05-11 09:58:48 +08:00
Theheavens 5498cf051e [Doc]Fix examples on negative graph sampler (#2863)
* Fix examples on negative graph sampler

In heterogeneous graph, the old example may build a negative graph without edges. In dgl.dataloading.dataloader line637, it shows the key of neg_edges dict should be item in g.canonical_etypes.

* poke ci

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-05-06 21:34:31 +08:00
xiang song(charlie.song) 975eb8fc5d [Distributed] Distributed node embedding and sparse optimizer (#2733)
* Draft for sparse emb

* add some notes

* Fix

* Add sparse optim for dist pytorch

* Update test

* Fix

* upd

* upd

* Fix

* Fix

* Fix bug

* add transductive exmpale

* Fix example

* Some fix

* Upd

* Fix lint

* lint

* lint

* lint

* upd

* Fix lint

* lint

* upd

* remove dead import

* update

* lint

* update unitest

* update example

* Add adam optimizer

* Add unitest and update data

* upd

* upd

* upd

* Fix docstring and fix some bug in example code

* Update rgcn readme

Co-authored-by: Ubuntu <ubuntu@ip-172-31-57-25.ec2.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-24-210.ec2.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-2-66.ec2.internal>
2021-05-03 08:37:40 +08:00
Da Zheng 45ec21b033 [Distributed] fix docs for distributed training. (#2840)
* update distributed training doc.

* explain data split.

* fix message passing.

* id mapping.

* fix.

* test data reshuffling.

* fix a bug.

* fix test.

* Revert "fix test."

This reverts commit 907f1dd88994894db94ffcd77b5995279f76b489.

* Revert "fix a bug."

This reverts commit ff0a4d893bbe4911a75ecb04bcbd9bc65315166b.

* Revert "test data reshuffling."

This reverts commit 99bb2f6405de67fc73652220b02e7184461c3e5e.

Co-authored-by: Zheng <dzzhen@3c22fba32af5.ant.amazon.com>
2021-04-29 14:38:41 +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
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
Da Zheng 17f863569e fix (#2744) 2021-03-21 22:38:55 -07: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
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
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
Da Zheng 48a1794f85 [Doc] Fix the user guide for distributed partitioning. (#2684)
* fix doc.

* explain the schema file.

* fix.
2021-02-22 18:59:15 -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 9e04a52abd [Doc] Reorganize tutorial (#2678) 2021-02-19 13:49:13 +08:00
Da Zheng e4ff484494 [Distributed] Distributed METIS partition (#2576)
* add convert.

* fix.

* add write_mag.

* fix convert_partition.py

* write data.

* use pyarrow to read.

* update write_mag.py

* fix convert_partition.py.

* load node/edge features when necessary.

* reshuffle nodes.

* write mag correctly.

* fix a bug: inner nodes in a partition might be empty.

* fix bugs.

* add verify code.

* insert reverse edges.

* fix a bug.

* add get node/edge data.

* add instructions.

* remove unnecessary argument.

* update distributed preprocessing.

* fix readme.

* fix.

* fix.

* fix.

* fix readme.

* fix doc.

* fix.

* update readme

* update doc.

* update readme.

Co-authored-by: Ubuntu <ubuntu@ip-172-31-9-132.us-west-1.compute.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-2-202.us-west-1.compute.internal>
2021-02-08 17:19:18 -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
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
Mufei Li 3ca52b1e59 [Fix] Various Fix after Bug Bash (#2620)
* update

* Misc fix

* Update
2021-02-05 17:34:16 +08:00
Mufei Li b1840f49fa [Dataset] Change the Data Type of the Node Features in GINDataset/TUDataset from Float64 to Float32 (#2592)
* Update

* Update

* Update

* update

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Tong He <hetong007@gmail.com>
2021-02-03 21:00:45 +08:00
Quan (Andy) Gan 3a19bbc508 fix doc (#2604) 2021-02-02 11:04:43 +08:00
Minjie Wang 4ca706e1b4 [Doc] Update doc of dgl.function (#2585)
* update doc of dgl.function

* small fix
2021-01-28 16:32:18 +08:00
Zihao Ye 7bab1365e2 [feature] Supporting half precision floating data type (fp16). (#2552)
* add tvm as submodule

* compilation is ok but calling fails

* can call now

* pack multiple modules, change names

* upd

* upd

* upd

* fix cmake

* upd

* upd

* upd

* upd

* fix

* relative path

* upd

* upd

* upd

* singleton

* upd

* trigger

* fix

* upd

* count reducible

* upd

* upd

* upd

* upd

* upd

* upd

* upd

* upd

* upd

* only keep related files

* upd

* upd

* upd

* upd

* lint

* lint

* lint

* lint

* pylint

* upd

* upd

* compilation

* fix

* upd

* upd

* upd

* upd

* upd

* upd

* upd doc

* refactor

* fix

* upd number

Co-authored-by: Zhi Lin <linzhilynn@gmail.com>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-42-78.us-east-2.compute.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-21-156.us-east-2.compute.internal>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-01-28 11:21:58 +08:00
xiang song(charlie.song) a7e941c379 [Feature] Add support for sparse embedding (#2451)
* Add sparse embedding for dgl and update rgcn example

* upd

* Fix

* Revert "Fix"

This reverts commit 4da87cdfb8b8c3506b7fc7376cd2385ba8045c2a.

* Fix

* upd

* upd

* Fix

* Add unitest and update impl

* fix

* Clean up rgcn example code

* upd

* upd

* update

* Fix

* update score

* sparse for sage

* remove model sparse

* upd

* upd

* remove global norm

* revert delete model_sparse.py

* update according to comments

* Fix doc

* upd

* Fix test

* upd

* lint

* lint

* lint

* upd

* upd

* clean up

Co-authored-by: Ubuntu <ubuntu@ip-172-31-56-220.ec2.internal>
2021-01-28 00:26:49 +08:00
Tong He 0855d25501 [NN] Support scalar edge weight for GraphConv, SAGEConv and GINConv (#2557)
* add edge weight in forward

* fix lint

* fix

* fix

* address comments

* add utils

* add util to normalize in gcn way

* fix lint

* add unittest

* fix lint

* fix docstring

* fix docstring

* address comments

* improve notation consistence

* use preferred fn
2021-01-26 23:34:12 +08:00
Quan (Andy) Gan 6c23fba8fa [Doc][Org] move edge_softmax to functional (#2442)
* move edge_softmax to functional

* change examples

* fixes

* revert

* fix

* fix

* remove

* fix

* test

* add init

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-01-26 12:46:37 +08:00
Quan (Andy) Gan 2288098ba6 [Feature] Converting a block to a graph and allow block construction (#2555)
* first commit

* more commit

* fix and docstring

* add to api reference

* address comments

* Update convert.py
2021-01-24 12:44:54 +08:00
xnouhz 2c6d071617 [Feature] QM9 Dataset Support (#2521)
* [Model] MixHop for Node Classification task

* [docs] update

* [docs] update

* [fix] remove seed option

* [fix] update readme

* [feature] support qm9 dataset

* [style] update

* [docs] fix the details

* [fix] indexing only support int

* [style] update

* [fix] multiple backends support

* [docs] add qm9

* [fix] Z type: float32 -> int32

* [fix] Z type: int32 -> long

* [docs] add ref

* [docs] fix

* [docs] update

* [docs] update

* [fix] test eval

* [docs] fix example

Co-authored-by: xnuohz@126.com <ubuntu@ip-172-31-44-184.us-east-2.compute.internal>
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-01-21 01:33:30 +08:00
Jinjing Zhou 917a755291 [Fix] fix version check (#2544) 2021-01-19 19:35:43 +08:00
Mufei Li a6ec8480ef [Doc] Update FAQ (#2536) 2021-01-14 14:26:02 +08:00
Da Zheng 1db7588739 fix. (#2534)
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-01-14 14:23:36 +08:00
zhjwy9343 c448e61b55 [Doc] CN User Guide Chapter 7 (#2481)
* [Doc] Add Chapter 7

* [Doc] Add CN part 7, 7.1, 7.2, and 7.3

* [Doc] Add CN part 7, 7.1, 7.2, and 7.3

* [Doc] 1st round copyediting, 7, 7.1, 7.2, and 7.3, removing EN parts.

* [Doc] 2nd round copyediting.

* [Doc] 3rd round copyediting.

* [Doc] Copyedit with reviewers' comments.

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2021-01-13 19:28:20 +08:00
zhjwy9343 31f65f036a [Doc] user guide CN chapter 6 (#2452)
* [Doc] Add 1st CN in chapter 6

* [Doc] Add 1st CN in chapter 6

* [Doc] Add 1st CN in chapter 6.1

* [Doc] Add 1st CN in chapter 6.1

* [Doc] Add 1st CN in chapter 6.2

* [Doc] Add CN in chapter 6.2

* [Doc] Add CN in chapter 6.3

* [Doc] Fix a code bug in chapter 6.3

* [Doc] Add CN in chapter 6.4, 6.5, and 6.6.

* [Doc] Add CN in chapter 6.4, 6.5, and 6.6,then remove EN parts.

* [Doc] revised 6.4, 6.5, and 6.6.

* [Doc] 2nd round of copyediting.

* Delete syn1.bin

Remove this unrelated file to this branch

* [Doc] Update guide_cn index.rst.

* [Doc] Update guide_cn index.rst.

* [Doc] Update guide_cn index.rst.

* [Doc] Update guide_cn index.rst.

* [Doc] Add chapter 7 in index page

* [Doc] edit chapter 6 in PR

* [Doc] edit chapter 6 in PR

* [Doc] Copyediting with Murphy and Minjie's comments.

* [Doc] Fix doc line in the previous chapters.

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-01-13 17:12:33 +08:00
Tong He a9f8f2589c [Doc] Update GraphDataLoader in our docs (#2504)
* update graphdataloader in docs

* fix

* update examples

* fix sagpool

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-01-11 13:11:19 +08:00
Quan (Andy) Gan e018939729 [Tutorial] New tutorials for small graphs (#2482)
* new tutorials for small graphs

* address changes and use GraphDataLoader

* fix and add data

* fix load_data

* style fixes

* Update 5_graph_classification.py
2021-01-10 22:24:21 +08:00
Tong He e2e524df41 [Feature] Add GraphDataLoader implementation (#2496)
* add graph dataloader

* add to doc

* fix

* fix

* fix docstring

* update according to torch default_collate

* add unittest

* fix

* fix lint

* fix
2021-01-07 13:38:21 +08:00
Zihao Ye fa5f4d6aa7 [Doc] Hotfix on docstring of set_batch_num_edges (#2498) 2021-01-06 21:45:04 +08:00