7c7113f660
add use_ddp to dataloaders ( #2911 )
2021-05-17 15:46:02 +08:00
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
85231dc93b
[Example] Label Propagation and Correct&Smooth ( #2852 )
...
* [example] label propagation and correct&smooth
* update
* update
* update
* update
* [docs] add speed for c&s
* update
* [fix] remove gat&consistent with the author's code
* [feat] multi-type adj norm supported
* update
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
2021-05-12 18:45:46 +08:00
d70ca6eb7c
[Example] graphsaint ( #2792 )
...
* graphsaint
* graphsaint
* graphsaint
* graphsaint
* fixed the model
* fixed some bugs
* fixed the computing of normalization and updated the results
* fixed some bugs and updated the results
* Update utils.py
* Update train_sampling.py
* Update train_sampling.py
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com >
2021-05-11 16:25:56 +08:00
70695ff8f8
[Model] Training GraphSAGE with PyTorch Lightning ( #2878 )
...
* pytorch lightning initial examples
* revert most changes in dataloader to favor #2886 .
* address comments
2021-05-11 11:28:13 +08:00
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
b2b531e041
[Example] add implementation of grace ( #2828 )
...
* [Example] add implementation of grace
* [Doc] add grace in the index file
* fix
* fix typos
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
2021-04-27 12:43:03 +08:00
2ae891a78c
Update main.py ( #2872 )
2021-04-26 17:12:19 +08:00
756fdd8e90
[Example] DeeperGCN ( #2831 )
...
* [example] deepergcn
* update
* update
* update
* update
* update
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
2021-04-25 18:37:41 +08:00
3c387988d7
[Example] DCRNN and GaAN ( #2858 )
...
* Ready for PR
* refractor code
Co-authored-by: Ubuntu <ubuntu@ip-172-31-45-47.ap-northeast-1.compute.internal >
Co-authored-by: Tianjun Xiao <xiaotj1990327@gmail.com >
2021-04-23 22:33:39 +08:00
ebc6a85a80
[Bugfix] TGN model bugfix and improvement ( #2860 )
...
* Add DCRNN Example
* Bugfix DCRNN
* Bugfix DCRNN
* Bugfix Train/eval have different flow
* Performance Matched
* modified tgn for research
* kind of fixed 2hop issue
* remove data
* bugfix and improve the performance
* Refractor Code
Co-authored-by: Ubuntu <ubuntu@ip-172-31-45-47.ap-northeast-1.compute.internal >
Co-authored-by: Tianjun Xiao <xiaotj1990327@gmail.com >
2021-04-23 15:44:22 +08:00
4579bbf7b7
Update README.md ( #2853 )
...
Change the format of README.md
2021-04-21 13:05:25 +08:00
6232ee8d44
Update README.md ( #2820 )
...
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com >
2021-04-20 13:17:30 +08:00
e58eeebfb6
[Model] Add RECT example ( #2813 )
...
* Update README.md
Add description of RECT
* [Example] Add implementation of RECT
[Example] Add implementation of RECT
* Update classify.py
Modify the class names and the function names mentioned above
* Update main.py
Modify the function names mentioned above
* Update label_utils.py
Adjust the comments
* Update README.md
Add the github information
Co-authored-by: Tianjun Xiao <xiaotj1990327@gmail.com >
2021-04-19 21:48:38 +08:00
e70138bbe4
[Performance][GPU] Enable GPU uniform edge sampling ( #2716 )
...
* Start on uniform GPU sampling
* Save more work
* Get cu file compiling
* Update sampling
* More changes
* Get GPU sampling for uniform probabilities solved
* Fix batch tensor migration
* Fix
* update kernels
* expand blocking
* Undo testing change
* Cut down on sampling overhead
* Fix replacement
* Update unit tests
* Add option to gpu sample in graphsage
* Copy only csc to gpu
* Add ogbn support
* Fix linting
* Remove nvtx from sample
* Improve documentation and error checking
* Expand documentation
* Update assert checking
* delete extra space
* Use standard dataloader when dataset is a dictionary
* ogb -> ogbn
* Fix edge selection determinism
* Fix typos
* Remove nvtx
* Add comment for self.fanout_arrays and assert
* Fix linting
* Migrate to scalarbatcher
* Fix indentation
* Fix batcher
* Fix indexing
* Only use databatcher for GPU
* Convert to DGL NDArray to PyTorch Tensor
* Add optimization for PyTorch's F.tensor() for list of GPU tensors
Co-authored-by: Da Zheng <zhengda1936@gmail.com >
2021-04-14 18:55:09 -07:00
195f99362d
[example] multi gpus training for mag240m ( #2835 )
...
* multi gpus training for mag240m
* remove unused function
* small changes
* more fix
* use multi gpus to eval and more fix
Co-authored-by: Quan Gan <coin2028@hotmail.com >
2021-04-13 20:29:43 +08:00
afc83aa216
Graphsim ( #2794 )
...
* 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
* Finish Coding
* working code cpu
* pyg compare
* Accelerate with batching
* FastMode Enabled
* update readme
* Update README.md
* refractor code
* add graphsim code
* modified code
* few fix
* Modified graphsim
* Simple Model Added
* Clean up code
* Refractor the code for Merge
* Bugfix enable gradient when train
* update readme and format
Co-authored-by: Chen <chesirui@3c22fbe5458c.ant.amazon.com >
Co-authored-by: Tianjun Xiao <xiaotj1990327@gmail.com >
Co-authored-by: Ubuntu <ubuntu@ip-172-31-4-63.ap-northeast-1.compute.internal >
Co-authored-by: Ubuntu <ubuntu@ip-172-31-45-47.ap-northeast-1.compute.internal >
2021-04-12 19:02:36 +08:00
d70eade4f2
bug fix type mismatch ( #2830 )
...
int has no attribute of item
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com >
2021-04-11 20:33:03 +08:00
bfbbefa793
[Distributed] Fix a bug in multiprocessing sampling. ( #2826 )
...
Co-authored-by: Ubuntu <ubuntu@ip-172-31-73-81.ec2.internal >
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com >
2021-04-08 01:29:33 -07:00
2afa3598ba
[Example] JKNet ( #2795 )
...
* [example] jknet
* update
* update
* update
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
2021-04-07 14:29:33 +08:00
2980137719
[Doc] Typo in command example. ( #2816 )
...
* Typo in command example.
process_nowplaying_rs.py is the right name under this folder.
* Update README.md
* Update README.md
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com >
2021-04-07 09:43:37 +08:00
672e322719
[Model] Add BGNN example ( #2740 )
...
* Add BGNN example
* Update bgnn example
* Remove uncessary files
* update model index
* style fix
* Add Readme for BGNN example
* Update documentation; remove preprocessing
Co-authored-by: s.ivanov <s.ivanov@mesos-slave080-am6.central.criteo.prod >
Co-authored-by: s.ivanov <s.ivanov@mesos-slave150-am6.central.criteo.prod >
Co-authored-by: Quan Gan <coin2028@hotmail.com >
Co-authored-by: s.ivanov <s.ivanov@mesos-slave148-am6.central.criteo.prod >
Co-authored-by: s.ivanov <s.ivanov@mesos-slave078-am6.central.criteo.prod >
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2021-04-07 00:16:09 +08:00
e61138be19
[Example] CompGCN ( #2768 )
...
* compgcn
* readme
* readme
* update
* readme
Co-authored-by: zhjwy9343 <6593865@qq.com >
2021-04-05 20:45:38 +08:00
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
cba5af22eb
[bugfix] Fix bugs in vgae ( #2727 )
...
* [Example]Variational Graph Auto-Encoders
* change dgl dataset to single directional graph
* clean code
* refresh
* fix bug
* fix bug
* fix bug
* add gpu
Co-authored-by: Tianjun Xiao <xiaotj1990327@gmail.com >
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2021-04-03 20:02:04 +08:00
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
66f7fe8bce
Update model.py ( #2796 )
2021-03-29 16:30:32 +08:00
a655123b4e
[Example] Baselines for OGB-LSC-PCQM4M ( #2778 )
...
* Update
* Add files via upload
* Add files via upload
* Add files via upload
2021-03-29 14:07:07 +08:00
a1b5b42af4
remove unused and undefined function ( #2791 )
...
Co-authored-by: messi-k <messi.k@kakaocorp.com >
2021-03-28 00:17:11 +08:00
d04d59eef9
[Example] Temporal Graph Neural Network ( #2636 )
...
* 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
* Finish Coding
* working code cpu
* pyg compare
* Accelerate with batching
* FastMode Enabled
* update readme
* Update README.md
* refractor code
* Fix Bug
* add a simple temporal sampling method
* test results
* add train speed result
* Update README.md
* Update README.md
* Update README.md
* Update README.md
* Update README.md
* fix bug
* Fixed Grammar and Format Issue
Co-authored-by: Chen <chesirui@3c22fbe5458c.ant.amazon.com >
Co-authored-by: Tianjun Xiao <xiaotj1990327@gmail.com >
Co-authored-by: Ubuntu <ubuntu@ip-172-31-4-63.ap-northeast-1.compute.internal >
Co-authored-by: WangXuhongCN <wangxuhongcn@163.com >
Co-authored-by: WangXuhongCN <34096632+WangXuhongCN@users.noreply.github.com >
2021-03-23 08:57:21 +08:00
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
e6f6c2eb31
[Bugfix] Update deprecated method name in load_graph.py ( #2769 )
...
Method `dataset.num_labels` has been deprecated and replaced by `dataset.num_classes`.
Updating the method name to avoid runtime warning.
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2021-03-22 20:07:32 +08:00
9aac93ff21
Gnnexplainer ( #2717 )
...
* gnn-explainer
* gnn-explainer
* gnn-explainer
* gnn-explainer
* fix
* fix
* fix
* readme
* readme
Co-authored-by: zhjwy9343 <6593865@qq.com >
2021-03-22 18:13:08 +08:00
bb54206620
remove pyinstrument. ( #2772 )
...
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com >
2021-03-21 20:27:12 -07:00
93074dd358
Bug fix for RGCN Link Prediction Example Exceeding Specified Epochs ( #2757 )
...
* Address bug where epochs exceed specified limit
* fix spacing
2021-03-17 14:29:45 +08:00
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
91cb347719
[Example][Bug Fix] Improve DiffPool ( #2730 )
...
* change DiffPoolBatchedGraphLayer
* fix bug and add benchmark
* upt
* upt
* upt
* upt
Co-authored-by: Tong He <hetong007@gmail.com >
2021-03-09 16:03:00 +08:00
3317522622
[Example] Add implementation of InfoGraph ( #2644 )
...
* Add implementation of unsupervised model
* [Doc] Update Implementor's information
* [doc] add index of infograph
* [Feature] QM9_v2 Dataset Support
* fix a typo
* move qm9dataset from data to examples
* Update qm9_v2.py
* Infograph -> InfoGraph
* add implementation and results of semi-supervised model
* Update README.md
* Update README.md
* fix a typo
* Remove the duplicated links.
* fix some typos
* fix typos
* update model.py
* update collate fn
* remove unused functions
* Update model.py
* add device option
* Update evaluate_embedding.py
* Update evaluate_embedding.py
* Update unsupervised.py
* Fix typos
* fix bugs
* Update README.md
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
2021-03-09 15:26:11 +08:00
f7938649d1
[Example] Variational Graph Auto-Encoders ( #2587 )
...
* [Example]Variational Graph Auto-Encoders
* change dgl dataset to single directional graph
* clean code
* refresh
Co-authored-by: Tianjun Xiao <xiaotj1990327@gmail.com >
2021-03-02 10:24:34 +08:00
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
28b0a76618
[Example] Provide instructions of setting up passwordless SSH login. ( #2707 )
2021-02-27 18:34:46 -08:00
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
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
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
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
ff64bd0de8
add warning for a bug in README ( #2665 )
2021-02-16 00:16:40 -08:00
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
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
48d357b8f6
Fix numbers ( #2641 )
...
Co-authored-by: Ubuntu <ubuntu@ip-172-31-56-220.ec2.internal >
2021-02-08 14:05:30 +08:00
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
Quan (Andy) Gan