文件历史

435 次代码提交

作者 SHA1 备注 提交日期
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
Josh Carty 8a2b54d1d4 Add a TensorFlow example for SGC (#2480)
This is a TensorFlow implementation of Simple Graph Convolution. For
simplicity, we use the implementation of AdamW in TensorFlow Addons
that differs from Adam in PyTorch. This results in worse performance
on the evaluation datasets.

Co-authored-by: Josh Carty <joshuac@spotify.com>
Co-authored-by: Zihao Ye <expye@outlook.com>
2021-01-20 15:35:32 +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
Minjie Wang b1fb3c15ec Update example README.md 2021-01-14 17:19:13 +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
Mufei Li 80de340d86 Update README.md (#2485) 2021-01-04 15:29:24 +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