文件历史

提交图

634 次代码提交

作者 SHA1 备注 提交日期
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
Da Zheng 28b0a76618 [Example] Provide instructions of setting up passwordless SSH login. (#2707) 2021-02-27 18:34:46 -08:00
Ligeng Zhu 6568d0aaf6 [Example] Recommend NFS over copy files (#2689)
* Update README to use NFS, rather than copying files

* Update README.md

* Update README.md

* Update README.md

* Add ParMetis

* Update README.md

* fix directory.

* fix

* redo.

* fix.

* update.

* add more descriptions

Co-authored-by: Da Zheng <zhengda1936@gmail.com>
Co-authored-by: Quan Gan <coin2028@hotmail.com>
2021-02-25 20:20:52 +08:00
xnouhz 491d908b55 [Example] Add ARMA for Node Classification (#2694)
* [example] arma

* update

* update

* update

* update

* update

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2021-02-25 17:38:43 +08:00
Smile 583aa76621 [Example] Pytorch Seal example (#2638)
* add seal example

* 1. add paper infomation in examples/README
2. adjust codes
3. option test

* use latest `to_simple` to replace coalesce graph function

* remove outdated codes

* remove useless comment
2021-02-25 15:53:35 +08:00
Da Zheng 0526b885ee [Distributed] Fix the path for distributed partitioning. (#2700)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-9-132.us-west-1.compute.internal>
2021-02-24 20:41:41 -08:00
Da Zheng ff64bd0de8 add warning for a bug in README (#2665) 2021-02-16 00:16:40 -08:00
Tianqi Zhang (张天启) e46983f662 [Bug Fix][Example] fix dataloader bug in gxn (#2631)
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-02-10 11:21:34 +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
xiang song(charlie.song) 48d357b8f6 Fix numbers (#2641)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-56-220.ec2.internal>
2021-02-08 14:05:30 +08:00
Tianjun Xiao ae18d7f04e [Example]Add dependencies for GNN-Film and dagnn (#2633)
* add dependencies for GNN-Film and dagnn

* fix

* fix

* fix
2021-02-08 10:42:32 +08:00
Tianqi Zhang (张天启) 3cf42fd199 [Bug Fix][Example] hgp-sl: More info for running (#2634)
* [Bug Fix][Example] hgp-sl: More info for running

* change list to table

* change dataset summary in gxn and sagpool

* remove redundent

Co-authored-by: Tong He <hetong007@gmail.com>
2021-02-05 18:26:33 +08:00
Mufei Li 3ca52b1e59 [Fix] Various Fix after Bug Bash (#2620)
* update

* Misc fix

* Update
2021-02-05 17:34:16 +08:00
Chen Sirui f8c9d58aed [Bug] Hardgat enable gpu training (#2632)
* add multihead in DotGatConv

* Fix spacing issue

* Add Unit test for dotgat

* Modified Unit test for dotgat

* Add transformer like divisor

* Update dotgatconv.py

* Update hgao.py

* Update train.py

Co-authored-by: Chen <chesirui@3c22fbe5458c.ant.amazon.com>
Co-authored-by: Zihao Ye <expye@outlook.com>
2021-02-05 16:52:41 +08:00
xiang song(charlie.song) 23afe911b2 [Bug Fix] Fix several sparse optimizer bugs (#2596)
* Fix pytorch TCP kvstore bug

* lint

* Fix

* upd

* Fix lint

* Fix

* trigger

* fix

Co-authored-by: Ubuntu <ubuntu@ip-172-31-56-220.ec2.internal>
2021-02-05 11:49:41 +08:00
KounianhuaDu 2f71bc507a [Example] Neural Graph Collaborative Filtering (NGCF). (#2612)
* ngcf

* ngcf

* update

* ngcf

* ngcf

* remove data

* update

* data

Co-authored-by: zhjwy9343 <6593865@qq.com>
Co-authored-by: Zihao Ye <expye@outlook.com>
2021-02-05 00:32:21 +08:00
张恒瑞 94ab9709a7 [Bugfix] fix a typo in consis_loss function (#2616) 2021-02-04 11:59:41 +08:00
wcyjames ff345c2e22 [Example] update PointNet and PointNet++ examples for Part Segmentation (#2547)
* [Model] update PointNet example for Part Segmentation

* Fixed issues with pointnet examples

* update the README

* Added image

* Fixed README and tensorboard arguments

* clean

* Add timing

* Update README.md

* Update README.md

Fixed a typo

Co-authored-by: Tong He <hetong007@gmail.com>
2021-02-03 22:15:46 +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
S22 b04cf65b2b [Bugfix] Fix sparseadam bug in metapath2vec(pytorch) (#2607)
It seems that there is a bug in Sparseadam in PyTorch, and it can be temporarily fixed by: 
optimizer = optim.SparseAdam(list(self.skip_gram_model.parameters()), lr=self.initial_lr)

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-02-03 17:27:20 +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
Tianqi Zhang (张天启) 8e981db960 [Example] Add Graph Cross Net (GXN) example for pytorch backend (#2559)
* add sagpool example for pytorch backend

* polish sagpool example for pytorch backend

* [Example] SAGPool: use std variance

* [Example] SAGPool: change to std

* add sagpool example to index page

* add graph property prediction tag to sagpool

* [Example] add graph classification example HGP-SL

* [Example] fix sagpool

* fix bug

* [Example] change tab to space in README of hgp-sl

* remove redundant files

* remote redundant network

* [Example]: change link from code to doc in HGP-SL

* [Example] in HGP-SL, change to meaningful name

* [Example] Fix path mistake for 'hardgat'

* [Bug Fix] Fix undefined var bug in LegacyTUDataset

* upt

* [Bug Fix] Fix cache file name bug in TUDataset

* [Example] Add GXN example for pytorch backend

* modify readme

* add more exp result

Co-authored-by: zhangtianqi <tianqizh@amazon.com>
Co-authored-by: Tong He <hetong007@gmail.com>
2021-01-27 18:14:27 +08:00
lt610 8900450d9e [Example] DAGNN (#2545)
* dagnn

* dagnn

* Update README.md

* Update README.md

* fixed some details

* fixed some details

* Update README.md

* Update README.md

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2021-01-26 17:06:32 +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
Da Zheng 25ac334403 [Distributed] Heterogeneous graph support (#2457)
* Distributed heterograph (#3)

* heterogeneous graph partition.

* fix graph partition book for heterograph.

* load heterograph partitions.

* update DistGraphServer to support heterograph.

* make DistGraph runnable for heterograph.

* partition a graph and store parts with homogeneous graph structure.

* update DistGraph server&client to use homogeneous graph.

* shuffle node Ids based on node types.

* load mag in heterograph.

* fix per-node-type mapping.

* balance node types.

* fix for homogeneous graph

* store etype for now.

* fix data name.

* fix a bug in example.

* add profiler in rgcn.

* heterogeneous RGCN.

* map homogeneous node ids to hetero node ids.

* fix graph partition book.

* fix DistGraph.

* shuffle eids.

* verify eids and their mappings when loading a partition.

* Id map from homogneous Ids to per-type Ids.

* verify partitioned results.

* add test for distributed sampler.

* add mapping from per-type Ids to homogeneous Ids.

* update example.

* fix DistGraph.

* Revert "add profiler in rgcn."

This reverts commit 36daaed8b660933dac8f61a39faec3da2467d676.

* add tests for homogeneous graphs.

* fix a bug.

* fix test.

* fix for one partition.

* fix for standalone training and evaluation.

* small fix.

* fix two bugs.

* initialize projection matrix.

* small fix on RGCN.

* Fix rgcn performance (#17)

Co-authored-by: Ubuntu <ubuntu@ip-172-31-62-171.ec2.internal>

* fix lint.

* fix lint.

* fix lint.

* fix lint.

* fix lint.

* fix lint.

* fix.

* fix test.

* fix lint.

* test partitions.

* remove redundant test for partitioning.

* remove commented code.

* fix partition.

* fix tests.

* fix RGCN.

* fix test.

* fix test.

* fix test.

* fix.

* fix a bug.

* update dmlc-core.

* fix.

* fix rgcn.

* update readme.

* add comments.

Co-authored-by: Ubuntu <ubuntu@ip-172-31-2-202.us-west-1.compute.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-9-132.us-west-1.compute.internal>
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-62-171.ec2.internal>

* fix.

* fix.

* add div_int.

* fix.

* fix.

* fix lint.

* fix.

* fix.

* fix.

* adjust.

* move code.

* handle heterograph.

* return pytorch tensor in GPB.

* remove some tests in example.

* add to_block for distributed training.

* use distributed to_block.

* remove unnecessary function in DistGraph.

* remove distributed to_block.

* use pytorch tensor.

* fix a bug in ntypes and etypes.

* enable norm.

* make the data loader compatible with the old format.

* fix.

* add comments.

* fix a bug.

* add test for heterograph.

* support partition without reshuffle.

* add test.

* support partition without reshuffle.

* fix.

* add test.

* fix bugs.

* fix lint.

* fix dataset.

* fix for mxnet.

* update docstring.

* rename to floor_div

* avoid exposing NodePartitionPolicy and EdgePartitionPolicy.

* fix docstring.

* fix error.

* fixes.

* fix comments.

* rename.

* rename.

* explain IdMap.

* fix docstring.

* fix docstring.

* update docstring.

* remove the code of returning heterograph.

* remove argument.

* fix example.

* make GraphPartitionBook an abstract class.

* fix.

* fix.

* fix a bug.

* fix a bug in example

* fix a bug

* reverse heterograph sampling.

* temp fix.

* fix lint.

* Revert "temp fix."

This reverts commit c450717b9f578b8c48769c675f2a19d6c1e64381.

* compute norm.

* Revert "reverse heterograph sampling."

This reverts commit bd6deb7f52998de76508f800441ff518e2fadcb9.

* fix.

* move id_map.py

* remove check

* add more comments.

* update docstring.

Co-authored-by: Ubuntu <ubuntu@ip-172-31-2-202.us-west-1.compute.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-9-132.us-west-1.compute.internal>
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-62-171.ec2.internal>
2021-01-24 19:18:00 -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
张天启 b36b6c268e [Example] Add HGP-SL example for pytorch backend (#2515)
* add sagpool example for pytorch backend

* polish sagpool example for pytorch backend

* [Example] SAGPool: use std variance

* [Example] SAGPool: change to std

* add sagpool example to index page

* add graph property prediction tag to sagpool

* [Example] add graph classification example HGP-SL

* [Example] fix sagpool

* fix bug

* [Example] change tab to space in README of hgp-sl

* remove redundant files

* remote redundant network

* [Example]: change link from code to doc in HGP-SL

* [Example] in HGP-SL, change to meaningful name

* [Example] Fix path mistake for 'hardgat'

Co-authored-by: zhangtianqi <tianqizh@amazon.com>
2021-01-18 17:01:52 +08:00
Chen Sirui 673cc64c8c [Example] Add hardGAT example (#2486)
* Add hgat example

* Add experiment

* Clean code

* clear the code

* Add index in README

* Add index in README

* Add index in README

* Add index in README

* Add index in README

* Add index in README

* Change the code title and folder name

* Ready to merge

* Prepare for rebase and change message passing function

* use git ignore to handle empty file

* change file permission to resolve empty file

* Change permission

* change file mode

Co-authored-by: Chen <chesirui@3c22fbe5458c.ant.amazon.com>
Co-authored-by: Tianjun Xiao <xiaotj1990327@gmail.com>
2021-01-15 16:58:19 +08:00
KounianhuaDu a9e16c177c [Example] GNN-FiLM (#2476)
* gnnfilm

* update

* readme

Co-authored-by: zhjwy9343 <6593865@qq.com>
2021-01-13 22:34:14 +08:00
张恒瑞 80fdcfdf83 [Example] Graph Random Neural Network (#2502)
* Add Implementation of GRAND

* Performance updated

* Update README.md

add indexing entries

* add indexing entries for grand

* change grand from 2021 to 2020

* [Example] fix some bugs

* [Examples] fix bug

Co-authored-by: 张恒瑞 <hengrui@macbook.local>
Co-authored-by: 张恒瑞 <hengrui@192.168.124.35>
Co-authored-by: zhjwy9343 <6593865@qq.com>
2021-01-13 14:06:47 +08:00
Hongyu Cai 1502c56d51 [Doc] Fix typo of datasets (#2517)
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-01-13 12:34:47 +08:00
Jinjing Zhou 71283997ce Integrate Regression Test with Jenkins (#2448)
* add bench jenkins

* instance type

* fix

* fix

* fix

* 111

* test

* 111

* 111

* fix

* test

* run

* fix

* fix

* fix

* fix

* fix

* publish results

* 111

* regression

* launch ec2 script

* fix

* add

* run on master

* change

* rrr

* run gpu

* fix

* fix

* try fix

* fix

* ff

* fix

* fix

* fix

* refactor

* fix

* fix

* update

* fix

* fix

* fix

* fix

* remove import torchtext

* add shm size

* update

* fix

* fix

* fix

* fix

* fix this!!!!

* 111

* fix

* remove verbose

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* update readme

* fix

* fix

* fix

* change asv default to head

* commit sage and rgcn

* fix

* update
2021-01-11 16:01:51 +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
Pein 400687d7a8 add venue label filename; delete redundant code (#2491)
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-01-08 17:05:33 +08:00
Quan (Andy) Gan 79940695de [Examples][Bug] fix a silly typo in JTNN (#2505) 2021-01-07 20:14:45 +08:00
Pein a8ce095724 update the code of examples/pytorch/metapath2vec.py (#2463)
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-01-02 01:02:01 +08:00
Mufei Li 927d2b319b [Example] Move Data to GPU before Minibatch Training (#2453)
* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update
2020-12-28 21:26:32 +08:00
张天启 72ef642f29 [Example] Add sagpool example for pytorch backend (#2429)
* add sagpool example for pytorch backend

* polish sagpool example for pytorch backend

* [Example] SAGPool: use std variance

* [Example] SAGPool: change to std

* add sagpool example to index page

* add graph property prediction tag to sagpool

Co-authored-by: zhangtianqi <tianqizh@amazon.com>
2020-12-28 17:44:55 +08:00
Mufei Li 0018e90c5a [Doc] Implementation Index Page (#2443)
* Update

* Update

* Update

* Update README.md

* Update README.md

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update
2020-12-26 14:37:04 +08:00
Mufei Li 492ad9be81 [Fix] Avoid Overloading Nonzero for PyTorch Backend (#2434)
* Update gatedgraphconv.py

* Update entity_classify.py

* Update data-process.rst

* Update reading_data.py

* Update data-process.rst

* Update utils.py

* Update knowledge_graph.py

* Update entity_classify.py

* Update rdf.py

* Update entity_classify_mb.py

* Update test_classify.py

* Update tensor.py

* Update sparse.py

* Update entity_classify_mp.py

* Update 6_line_graph.py
2020-12-20 13:12:17 +08:00
xnouhz 5c77b61184 [Model] MixHop for Node Classification task (#2426)
* [Model] MixHop for Node Classification task

* [docs] update

* [docs] update

* [fix] remove seed option

* [fix] update readme

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>
2020-12-18 19:28:46 +08:00
Maybewuss 0ceb6751a0 Correct error when using enisum (#2417)
* correct error when using enisum

* Correct error when using enisum

Co-authored-by: Zihao Ye <expye@outlook.com>
2020-12-13 01:00:03 +08:00
Maybewuss ee8a0e6628 correct error when using enisum (#2406)
Co-authored-by: Zihao Ye <expye@outlook.com>
2020-12-08 23:16:10 +08:00
Maybewuss 6c0cc1fb76 modify code with buildin function (#2394)
* modify code with buildin function

* use enisum

Co-authored-by: Zihao Ye <expye@outlook.com>
2020-12-08 12:46:09 +08:00
mszarma 62b4bbb494 [Fix] Enable mini-batch rgcn for CPU (#2345) 2020-11-16 17:27:54 +08:00
Guo Qipeng cb2bc69b55 Fix a bug in the data loading of GraphWriter (#2343)
In the bucket sampler, the length array should be recovered after random picking samples.
2020-11-13 22:33:12 +08:00
Quan (Andy) Gan 88a042d34a fix reddit self loop (#2330) 2020-11-09 14:40:21 +08:00
Lingfan Yu b8cc26e34a [Model] SIGN for OGB dataset (#2316)
* sign for ogbn products, arxiv, mag

* texts

* fix

* update ogb folder readme

* use dgl nightly build

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2020-11-03 10:28:31 +08:00
chwan-rice 269983dbcd [Fix] typo (#2265)
* upd

* upd

* upd

* upd

* upd

* fix OpenMP compatibility issues

* typo

* partition

* misc

* fix typo

* num_parts=1

* import torch

* long

* print info

* print info

* print info

* upd

* remove debug code

* revert partition.py

* fix cut count

* fix cut count

* Revert "fix cut count"

This reverts commit 10926b4fd48f45c8f1ddb58be7db6c22e653effd.

* Revert "fix cut count"

This reverts commit 76465283bef093a2b4209ad70dd15d2437b2ec8a.

* type of deprecate

* typo in deprecate info

* fix typo

* use cv for partitioning

* CE

* no message

* revert

* typo

* no message

* no message

* no message

* practice

* dmlc-core

* DGLGraph

Co-authored-by: yzh119 <expye@outlook.com>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2020-10-29 23:05:36 +08:00
maqy1995 d06c540336 [Example] fix print info (#2305) 2020-10-29 09:20:27 +08:00