Chang Liu
72cfb9342f
[Example][Refactor] RGCN link prediction example refactor ( #4548 )
2022-09-30 12:57:50 -07:00
Chang Liu
40a2f3c760
[Example][Refactor] Refactor RGCN example ( #4327 )
...
* Refactor full graph entity classification
* Refactor rgcn with sampling
* README update
* Update
* Results update
* Respect default setting of self_loop=false in entity.py
* Update
* Update README
* Update for multi-gpu
* Update
2022-08-25 13:19:32 +08:00
Minjie Wang
0227ddfb66
[NN] Rework RelGraphConv and HGTConv ( #3742 )
...
* WIP: TypedLinear and new RelGraphConv
* wip
* further simplify RGCN
* a bunch of tweak for performance; add basic cpu support
* update on segmm
* wip: segment.cu
* new backward kernel works
* fix a bunch of bugs in kernel; leave idx_a for future
* add nn test for typed_linear
* rgcn nn test
* bugfix in corner case; update RGCN README
* doc
* fix cpp lint
* fix lint
* fix ut
* wip: hgtconv; presorted flag for rgcn
* hgt code and ut; WIP: some fix on reorder graph
* better typed linear init
* fix ut
* fix lint; add docstring
2022-02-23 12:10:20 +08:00
Mufei Li
e9c3c0e8ad
[Model] Simplify RGCN
...
* Update (#5 )
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* FIx
* Try
* Update
* Update
* Update
* Fix
* Update
* Fix
* Fix
* Fix
* Fix
* Update
* Fix
* Update
* Update
* Update
* Fix
* Fix
* Update
* Update
* Update
* Update
* Fix
* Fix
* Fix
* Update
* Update
* Update
* Update
* Update
* Update README.md
* Update
* Fix
* Update
* Update
* Fix
* Fix
* Fix
* Update
* Update
* Update
Co-authored-by: Ubuntu <ubuntu@ip-172-31-6-240.us-west-2.compute.internal >
* Update
* Update
* Fix
* Update
* Update
* Update
* Fix
* Update
* Update
* Update
* Update
* Update
* Update
* CI
Co-authored-by: Ubuntu <ubuntu@ip-172-31-6-240.us-west-2.compute.internal >
Co-authored-by: Ubuntu <ubuntu@ip-172-31-57-123.us-west-2.compute.internal >
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com >
2022-02-17 15:33:27 +08:00
Quan (Andy) Gan
b4cd60a952
fix relgraphconv bug ( #3256 )
2021-08-23 17:29:14 +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
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
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
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
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
xiang song(charlie.song)
dbb028acc8
[Dataset] Builtin knowledge graph dataset ( #1881 )
...
* buildin knowledge graph dataset
* upd
* docstring
* Fix
Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal >
Co-authored-by: Tong He <hetong007@gmail.com >
2020-07-30 23:37:33 +08:00
xiang song(charlie.song)
aee1067941
[Dataset] RDF dataset with DGL-Dataset template ( #1869 )
...
* update rdf builtin dataset
* Fix
* use new dataset
* fix
* rdf dataset using new framework
* tf work
* Fix mxnet
* Fix tensorflow
* Fix mxnet
* Update
* upd
* update some docstring
* clean some dead code
Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal >
2020-07-30 09:12:00 +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
xiang song(charlie.song)
d6b4e286b6
RGCN minibatch using homogeneous graph. ( #1745 )
...
* minibatch rgcn
* minibatch based rgcn
* remove ogb
* lint
* Add sparse emb
* Fix
* Fix
Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal >
Co-authored-by: Ubuntu <ubuntu@ip-172-31-68-185.ec2.internal >
Co-authored-by: Ubuntu <ubuntu@ip-172-31-87-240.ec2.internal >
2020-07-07 00:26:10 +08:00
hgalioulline
6c7c403914
[Feature] Filtered MRR metrics for R-GCN example ( #1298 )
...
* Add filtered metrics for R-GCN example
* Add new line to end of file
* Add evaluation protocol argument option for R-GCN example
* Update README
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com >
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com >
Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal >
2020-03-13 16:25:03 +08:00
Minjie Wang
1c91f460d3
[Hetero][Model] RGCN for heterogeneous input ( #885 )
...
* new hetero RGCN
* bgs running
* fix gpu
* am dataset
* fix bug in label preparation
* Fix AM training; add result
* rm sym link
* new embed layer; mutag
* mutag matched; other fix
* minor fix
* dataset refactor
* new data loading
* rm old files
* refactor
* docstring
* include literal nodes in AIFB dataset
* address comments
* docstring
2019-09-29 00:21:17 -04:00
Giuseppe Futia
6c3dba867a
[Tutorial] Update prerequisites of README ( #380 )
...
* Update prerequisites of README
* dependencies for pytorch models
* dependencies for mxnet models
* minor
2019-02-11 22:58:17 -05:00
Gan Quan
ac660f45ef
[Doc] Readme ( #265 )
...
* [Doc] updated readme
* contribution guide change
* small edits
* add perf #
* Update README.md
* Update README.md
* add others...
* rewrite scalability
* model & code links
* some note for treelstm
* rewrite scalability
2018-12-06 00:31:04 -05:00
Minjie Wang
e557ed89ff
[Doc][Model] example folder doc and model summary ( #259 )
...
* update example readme
* mx example readme
* add results in readme
* mx rgcn readme
2018-12-05 14:35:14 -05:00
Lingfan Yu
b98dc92c59
[Model] Relational GCN ( #55 )
...
* data preprocessing for rgcn
* edge subgraph
* WIP: RGCN
* use edge feature in spmv
* fix bugs
* match AIFB accuracy
* match mutag accuracy
* avoid materializing in featureless case
* remove untouched nodes and relabel nodes
* fix python list concatenate overhead
* sparsely store edge types
* refactor entity classify code for clean link prediction implementation
* further refactor code
* refactoring
* rgcn block decompose layers
* link predict dataset
* link predict model and eval code
* dropout, self-loop, regularization, etc, plus bug fixes
* update to new api
* dataset update
* bugs, WIP, need to impl early stopping and filtered metrics
* instruction to run, and minor
* group conv and early stop
* clean slow code
* some code comments
* use new api in model code
* change data preprocessing
* entity classify model
* WIP
* move dgl graph out of model
* hot fix for extract zip
* fix link predict model
* use latest dgl apis
* still have memory issue...
* bug fix and move inference to cpu
* move rgcn data processing to contrib
* th.allclose -> U.allclose
* minor change in readme
* fix memory issue in entity classify
* fix and testing code for link predict
* fix entity classify
* clean up
* fix comments
* revert erroneous git merge changes
* code clean up and more comments
* minor
* dependent package version
2018-12-03 13:01:27 -05:00