文件历史

110 次代码提交

作者 SHA1 备注 提交日期
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
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
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
鄢振宇Michael Yan 3c8382f6b6 Fix typos in zero-in-degree (#2750) 2021-03-14 19:13:59 +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
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
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) 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
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
Mufei Li 3ca52b1e59 [Fix] Various Fix after Bug Bash (#2620)
* update

* Misc fix

* Update
2021-02-05 17:34:16 +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
Quan (Andy) Gan 1f6eba9ed3 [Bug] Heterogeneous graph convolution bugfix (#2578)
* fix heterograph conv

* remove test cases

* fix test

* fix test

* fix test

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-01-29 09:30:19 +08:00
Chen Sirui e4ddafe951 [NN] add multihead in DotGatConv (#2549)
* add multihead in DotGatConv

* Fix spacing issue

* Add Unit test for dotgat

* Modified Unit test for dotgat

* Add transformer like divisor

* Update dotgatconv.py

Co-authored-by: Chen <chesirui@3c22fbe5458c.ant.amazon.com>
Co-authored-by: Zihao Ye <expye@outlook.com>
2021-01-28 21:05:10 +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
Tong He bedccc2392 fix 2532 (#2533) 2021-01-14 13:40:25 +08:00
xiang song(charlie.song) 54f186bba0 Fix (#2524)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-59-204.ec2.internal>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-01-13 15:09:21 +08:00
Quan (Andy) Gan 498505bafb [Doc] fix docserver build for tutorials (#2518)
* fix doc build

* hide GPU code
2021-01-12 17:55:58 +08:00
Minjie Wang d3c1645550 [Performance][Test] New low memory implementation for RGCN and related regression tests (#2468)
* WIP

* finish lowmem impl; benchmarking

* wip

* wip

* fix benchmarks

* fix bug in searchsorted

* update readme

* update

* lint and ogb dependency

* add flags to to_homogeneous

* fix docstring

* address comments
2021-01-04 22:23:59 +08:00
Zihao Ye baace710cf [documentation] Improve the docstring of global pooling layer (#2437)
* upd

* fix-link

* lint
2020-12-21 10:12:54 +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
Mufei Li 013d145663 [NN] Attention Retrieval for NN Modules (#2397)
* Update

* Update
2020-12-07 23:42:25 +08:00
Mufei Li 45e3e9af2c [WIP] Misc Fix (#2359)
* Update

* Update

* Update
2020-11-19 23:54:55 +08:00
Mufei Li 812f597096 Update appnpconv.py (#2324) 2020-11-10 15:37:30 +08:00
Junki Ishikawa 6b0d42db15 Fix typo (#2235)
Co-authored-by: Chao Ma <mctt90@gmail.com>
2020-09-25 16:40:22 +08:00
xiang song(charlie.song) 0435b74c48 [Hotfix] Fix layer norm (#2119)
* hotfix

* Fix Layer Norm

Co-authored-by: Ubuntu <ubuntu@ip-172-31-87-240.ec2.internal>
2020-08-28 11:20:33 +08:00
Tianjun Xiao 7ead28c315 [Doc] [NN] Fix message and nn user guide link (#2090)
* fix user guide link

* some fix on tensorflow
2020-08-22 00:18:22 +08:00
Minjie Wang f13b9b6212 [Doc] Scan the API docs and make many changes (#2080)
* WIP: api

* dgl.sampling, dgl.data

* dgl.sampling; dgl.dataloading

* sampling packages

* convert

* subgraph

* deprecate

* subgraph APIs

* All docstrings for convert/subgraph/transform

* almost all funcs under dgl namespace

* WIP: DGLGraph

* done graph query

* message passing functions

* lint

* fix merge error

* fix test

* lint

* fix

Co-authored-by: Quan Gan <coin2028@hotmail.com>
2020-08-20 23:26:49 +08:00
Tianjun Xiao 65866989c0 [NN] [Doc] Fix nn api doc (#2047)
* fix mxnet nn doc and no note on chebconv

* change notes to note to highlight
2020-08-18 23:47:16 +08:00
Mufei Li be444e52d9 [Doc/Feature] Refactor, doc update and behavior fix for graphs (#1983)
* Update graph

* Fix for dgl.graph

* from_scipy

* Replace canonical_etypes with relations

* from_networkx

* Update for hetero_from_relations

* Roll back the change of canonical_etypes to relations

* heterograph

* bipartite

* Update doc

* Fix lint

* Fix lint

* Fix test cases

* Fix

* Fix

* Fix

* Fix

* Fix

* Fix

* Update

* Fix test

* Fix

* Update

* Use DGLError

* Update

* Update

* Update

* Update

* Fix

* Fix

* Fix

* Fix

* Fix

* Fix

* Fix

* Fix

* Update

* Fix

* Update

* Fix

* Fix

* Fix

* Update

* Fix

* Update

* Fix

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Fix

* Fix

* Update

* Update

* Update

* Update

* Update

* Update

* rewrite sanity checks

* delete unnecessary checks

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Fix

* Update

* Update

* Update

* Fix

* Fix

* Fix

* Update

* Fix

* Update

* Fix

* Fix

* Update

* Fix

* Update

* Fix

Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
Co-authored-by: Quan Gan <coin2028@hotmail.com>
2020-08-18 04:26:29 +08:00
Jinjing Zhou 0afc3cf8e1 [NN] Add ChebConv for Tensorflow (#2038)
* add chebconv

* fix

* lint

* fix lint

* docs
2020-08-17 22:12:27 +08:00
Tianjun Xiao b1e69105d4 [Doc] API Doc update for mxnet and tf, remove some degree check (#2028)
* mx tf relconv

* use method instead of private attr

* src and dst have different fc for gat

* update edgeconv

* change sage and sgconv

* no degree check on gin

* add remainding API doc

* fix pylint

* infer fc_src and fc_dst, only one tensor for block

* fix pytest
2020-08-17 13:23:44 +08:00
zhjwy9343 2fa7f71ad5 [Doc] new nn api doc (#2019)
* Add dotproduct attention

* [Feature] Add dotproduct attention

* [Feature] Add dotproduct attention

* [Feature] Add dotproduct attention

* [New] Update landing page

* [New] Update landing page

* [New] Update landing page

* [Doc] Update landing page

* [Doc] Update landing page

* [Doc] Update landing page

* [Doc] Update landing page

* [Doc] Update landing page

* [Doc] Update landing page

* [Doc] Update landing page

* [Doc] Update landing page

* [Improvement] use dgl build-in in dotgatconv

* [Doc] review API doc string bottom up

* [Doc] Add doc of input and output features

* [Doc] Update doc string for pooling and transformer.

* [Doc] Reformat doc string and change some wordings.

* [Doc] Doc string refactoring.

* [Doc] Doc string refactoring.

* [Doc] Doc string refactoring.

* [Doc] Doc string refactoring.

* [Doc] Doc string refactoring.

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2020-08-14 18:00:38 +08:00
Tianjun Xiao 4672c022a8 [hotfix] no in-degree check on sage (#2014)
* no degre check on sage

* fix pytest

* no in degree check for heteroconv
2020-08-13 13:11:55 +08:00
Tianjun Xiao 69f5869f3b [Doc] NN doc refactor Conv Layers (#1672)
* go through gcn, relgcn

* fix tagconv formula

* fix doc in sageconv

* fix sgconv doc

* replace hat with tilde

* more comments on gmmconv

* fix agnnconv chebconv doc

* modify nnconv doc

* remove &

* add nn conv examples

* Rebase master

* More merge conflicts

* check homo

* add back self loop for some convs, check homo in tranform

* add example for denseconv

* add example and doc for dotgat and cfconv

* check in-degree for graphconv

* add language fix

* gconv address all comments

* another round of change based on api template

* change agnn

* go through agnn, appnp, atomic, cf, cheb, dense, gat, sage modules

* finish pytorch part of nn conv

* mxnet graphconv done

* tensorflow graphconv works

* add new modules into doc

* add comments to not split code

* refine doc

* resr

* more comments

* more fix

* finish conv and dense conv part api

* pylint fix

* fix pylink

* fix pylint

* more fix

* fix

* fix test fail because zere in degree

* fix test fail

* sage is not update for mxnet tf

Co-authored-by: Ubuntu <ubuntu@ip-172-31-0-81.us-east-2.compute.internal>
2020-08-12 18:47:19 +08:00
xiang song(charlie.song) 4ef01dbb05 [Example] Rgcn support ogbn-mag dataset. (#1812)
* rgcn support ogbn-mag dataset

* upd

* multi-gpu val and test

* Fix

* fix

* Add support for ogbn-mag

* Fix

* Fix

* Fix

* Fix

* Add layer_norm

* update

* Fix merge

* Clean some code

* update Readme

* upd

Co-authored-by: Ubuntu <ubuntu@ip-172-31-68-185.ec2.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-87-240.ec2.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal>
2020-08-09 15:56:15 +08:00
Quan (Andy) Gan f7ce48bcd4 [NN] Fix error message of GatedGraphConv (#1964)
Co-authored-by: Zihao Ye <expye@outlook.com>
2020-08-08 23:01:07 +08:00
Zihao Ye 18bfec24cf [hotfix] Refactor edge softmax module (#1967)
* upd

* upd

* upd

* upd

* upd

* upd

* upd

* upd
2020-08-07 23:50:12 +08:00
Zihao Ye 56339b34bf [Fix] Update ogb examples according to new API. (#1922)
* upd

* upd

Co-authored-by: Ubuntu <ubuntu@ip-172-31-29-3.us-east-2.compute.internal>
2020-08-03 23:08:50 +08:00
Zihao Ye 34a067eaa0 [bugfix] Fix the memory leak issue of Cluster GAT under 0.5 kernel and simplify the bipartite GAT. (#1908)
* uipd

* upd

* upd

* upd

* upd
2020-08-02 00:59:27 +08:00
Minjie Wang f4608c2227 [CUDA][Kernel] A bunch of int64 kernels for COO and CSR (#1883)
* COO sort

* COOToCSR

* CSR2COO

* CSRSort; CSRTranspose

* pass all CSR tests

* lint

* remove int32 conversion

* fix tensorflow nn tests

* turn on CI

* fix

* addreess comments
2020-07-30 21:02:07 +08:00
Zihao Ye 2b8eb5be6f [refactor] Move spmm and sddmm operators to operator module (#1873)
* upd

* upd

* upd

* upd
2020-07-28 18:39:12 +08:00
Minjie Wang 44089c8b4d [Refactor][Graph] Merge DGLGraph and DGLHeteroGraph (#1862)
* Merge

* [Graph][CUDA] Graph on GPU and many refactoring (#1791)

* change edge_ids behavior and C++ impl

* fix unittests; remove utils.Index in edge_id

* pass mx and th tests

* pass tf test

* add aten::Scatter_

* Add nonzero; impl CSRGetDataAndIndices/CSRSliceMatrix

* CSRGetData and CSRGetDataAndIndices passed tests

* CSRSliceMatrix basic tests

* fix bug in empty slice

* CUDA CSRHasDuplicate

* has_node; has_edge_between

* predecessors, successors

* deprecate send/recv; fix send_and_recv

* deprecate send/recv; fix send_and_recv

* in_edges; out_edges; all_edges; apply_edges

* in deg/out deg

* subgraph/edge_subgraph

* adj

* in_subgraph/out_subgraph

* sample neighbors

* set/get_n/e_repr

* wip: working on refactoring all idtypes

* pass ndata/edata tests on gpu

* fix

* stash

* workaround nonzero issue

* stash

* nx conversion

* test_hetero_basics except update routines

* test_update_routines

* test_hetero_basics for pytorch

* more fixes

* WIP: flatten graph

* wip: flatten

* test_flatten

* test_to_device

* fix bug in to_homo

* fix bug in CSRSliceMatrix

* pass subgraph test

* fix send_and_recv

* fix filter

* test_heterograph

* passed all pytorch tests

* fix mx unittest

* fix pytorch test_nn

* fix all unittests for PyTorch

* passed all mxnet tests

* lint

* fix tf nn test

* pass all tf tests

* lint

* lint

* change deprecation

* try fix compile

* lint

* update METIDS

* fix utest

* fix

* fix utests

* try debug

* revert

* small fix

* fix utests

* upd

* upd

* upd

* fix

* upd

* upd

* upd

* upd

* upd

* trigger

* +1s

* [kernel] Use heterograph index instead of unitgraph index (#1813)

* upd

* upd

* upd

* fix

* upd

* upd

* upd

* upd

* upd

* trigger

* +1s

* [Graph] Mutation for Heterograph (#1818)

* mutation add_nodes and add_edges

* Add support for remove_edges, remove_nodes, add_selfloop, remove_selfloop

* Fix

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

* upd

* upd

* upd

* fix

* [Transfom] Mutable transform (#1833)

* add nodesy

* All three

* Fix

* lint

* Add some test case

* Fix

* Fix

* Fix

* Fix

* Fix

* Fix

* fix

* triger

* Fix

* fix

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

* [Graph] Migrate Batch & Readout module to heterograph (#1836)

* dgl.batch

* unbatch

* fix to device

* reduce readout; segment reduce

* change batch_num_nodes|edges to function

* reduce readout/ softmax

* broadcast

* topk

* fix

* fix tf and mx

* fix some ci

* fix batch but unbatch differently

* new checkk

* upd

* upd

* upd

* idtype behavior; code reorg

* idtype behavior; code reorg

* wip: test_basics

* pass test_basics

* WIP: from nx/ to nx

* missing files

* upd

* pass test_basics:test_nx_conversion

* Fix test

* Fix inplace update

* WIP: fixing tests

* upd

* pass test_transform cpu

* pass gpu test_transform

* pass test_batched_graph

* GPU graph auto cast to int32

* missing file

* stash

* WIP: rgcn-hetero

* Fix two datasety

* upd

* weird

* Fix capsuley

* fuck you

* fuck matthias

* Fix dgmg

* fix bug in block degrees; pass rgcn-hetero

* rgcn

* gat and diffpool fix
also fix ppi and tu dataset

* Tree LSTM

* pointcloud

* rrn; wip: sgc

* resolve conflicts

* upd

* sgc and reddit dataset

* upd

* Fix deepwalk, gindt and gcn

* fix datasets and sign

* optimization

* optimization

* upd

* upd

* Fix GIN

* fix bug in add_nodes add_edges; tagcn

* adaptive sampling and gcmc

* upd

* upd

* fix geometric

* fix

* metapath2vec

* fix agnn

* fix pickling problem of block

* fix utests

* miss file

* linegraph

* upd

* upd

* upd

* graphsage

* stgcn_wave

* fix hgt

* on unittests

* Fix transformer

* Fix HAN

* passed pytorch unittests

* lint

* fix

* Fix cluster gcn

* cluster-gcn is ready

* on fixing block related codes

* 2nd order derivative

* Revert "2nd order derivative"

This reverts commit 523bf6c249bee61b51b1ad1babf42aad4167f206.

* passed torch utests again

* fix all mxnet unittests

* delete some useless tests

* pass all tf cpu tests

* disable

* disable distributed unittest

* fix

* fix

* lint

* fix

* fix

* fix script

* fix tutorial

* fix apply edges bug

* fix 2 basics

* fix tutorial

Co-authored-by: yzh119 <expye@outlook.com>
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-7-42.us-west-2.compute.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-1-5.us-west-2.compute.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-68-185.ec2.internal>
2020-07-28 14:30:41 +08:00
Quan (Andy) Gan ec2e24be6d [Feature] Slice dstnode features in NN modules (#1838)
* slice dstdata from srcdata within nn module

* a bunch of fixes

* add comment

* fix gcmc layer

* repr for blocks

* fix

* fix context

* fix

* do not copy internal columns

* docstring
2020-07-25 21:57:49 +08:00
Zihao Ye f25b1a064d [Feature] Autograd of gspmm and gsddmm on PyTorch/MXNet/Tensorflow (#1680)
* init

* reverse(by minjie

* upd

* upd

* upd

* upd

* upd

* upd

* upd

* gpu

* upd

* upd

* upd

* upd

* udp

* upd

* upd

* imidiot

* fix

* upd

* upd

* upd

* udp

* upd

* upd

* fix

* udp

* upd

* upd

* upd

* upd

* upd

* fix

* remove redundency

* upd

* upd

* upd

* cache

* upd

* upd

* upd

* upd

* upd

* udp

* upd

* trigger

* upd

* fix

* upd

* unused import

* upd

* upd
2020-07-13 23:21:22 +08:00