提交

提交图

2937 次代码提交

作者 SHA1 备注 提交日期
Da Zheng c99f423718 [Sampler] extend sampler (#238) 2018-12-04 16:46:26 -05:00
Lingfan Yu 455ea48570 [Bugfix] Conversion between networkx and DGLGraph (#244)
* copy feature to networkx

* fix to_network for multi-graph

* test case for nx conversion

* iterate over Index returns plain int instead of numpy.int64

* fix from_network multi-edge bug
2018-12-04 16:19:11 -05:00
VoVAllen 31800e7170 [Doc] Fix typo at doc index (#241)
Fix typo at doc index
2018-12-04 15:05:37 -05:00
Da Zheng a5a35d1f7a [Sampler] improve random shuffle performance in sampler (#228)
* fix.

* make it generic.

* add the API.

* fix.

* remove mxnet.
2018-12-04 09:24:32 -08:00
Da Zheng 78269cedfd [Doc] Update GCN README.md (#224)
* update readme.

* update readme
2018-12-04 12:07:05 -05:00
VoVAllen 16cc528757 [Doc] Improve Capsule with Jinyang & Fix wrong tutorial level layout (#236)
* improve capsule tutorial with jinyang

* fix wrong layout of second-level tutorial

* delete transformer
2018-12-04 10:46:49 -05:00
Da Zheng dafe46710b [Bug][MXNet] fix an API in immutable graph index. (#233) 2018-12-04 00:37:18 -05:00
HQ b1e8d95e99 [Doc] LGNN tutorial (#210)
* tutorial notebook added

* lg tutorial cleaned up

* dataset scaffold

* move dataloader to data

* fix model

* remove todo

* utils seperated

* [model]line graph new implementation + tutorial + binary sub graph dataset

* [tutorial] line graph sphinx scaffold

* [tutorial] lgnn tutorial improved

* [tutorial] remove notebook

* [tutorial] fix lg and gcn links

* [tutorial] fix random seed

* [tutorial]fix

* WIP

* code refactor done

* new mini dataset; remove utils code

* fix

* word fix

* fix link

* minor fix

* minor fix

* minor fix
2018-12-03 17:59:50 -05:00
Minjie Wang 10e18ed982 [Doc] overview and others (#222)
* model tutorials overview

* update the 1_first

* revised overview and summarized "what is DGL" in glance
2018-12-03 17:31:35 -05:00
Da Zheng 2c170a8c17 [Graph][Bugfix] Fix the API of map_to_subgraph_nid (#226)
* correct vid mapping API.

* fix sse.
2018-12-03 17:16:35 -05:00
Gan Quan 419ffbded9 [Doc] Installation guide (#223)
* docstrings for graph mutation & queries

* [Doc] contrib.sampler. (#200)

* subgraph docs

* updates

* added installation guide

* style fix

* package name fixes

* adding it back...

* fix as requested

* adding link

* mac os build instructions

* OS -> macOS
2018-12-03 14:33:21 -05:00
Lingfan Yu b0c76d48be [Bug] fix __slots__ in view.py (#230) 2018-12-03 13:28:22 -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
Da Zheng 8918cce0b5 [MXNet] add unit tests for mxnet (#227)
[MXNet] add unit tests for mxnet
2018-12-03 11:04:11 -05:00
yifeim 6f4898a128 [Model][MXNet] gcn normalization and compare with mlp baselines (#196)
* clean up pr-188 and resubmit

* address Da comments
2018-12-02 19:17:37 -08:00
Lingfan Yu 52ed09a392 [Bug] Fix inplace update (#221)
* inplace write row op and executor

* update scheduler and graph to use inplace write

* fix

* fix bug

* test case for inplace

* fix bugs for inplace apply node/edge

* fix comments

* th.allclose -> U.allclose
2018-12-02 19:14:39 -05:00
Da Zheng dd26ff10a8 [CI] change the mxnet installation in docker. (#218) 2018-12-02 14:47:50 -05:00
Gan Quan 5680f8dc13 [DOC] Documentation strings for graph mutation, queries, and subgraphing (#207)
* docstrings for graph mutation & queries

* [Doc] contrib.sampler. (#200)

* subgraph docs

* updates
2018-12-02 14:41:25 -05:00
Minjie Wang 57daf9c9a6 [Graph] small fix on the all index (#219) 2018-12-02 13:39:04 -05:00
Da Zheng 57b07fce92 [Model] Accelerate SSE (#172)
* lazy eval edge ids.

* parallelize node id lookup.

* fix a bug.

* use mxnet for index.

* use update_all in the subgraph training.

* sample neighbors.

* Revert "parallelize node id lookup."

This reverts commit e661f69bb06cb5a7c246f0e106f245e27800e220.

* update README.

* cache subgraphs.

* support all degrees.

* cache adj in CPU.

* fix a bug in sse.

* print.

* raise error on mutable graphs.

* measure train time per epoch.

* fix a bug in graph_index.

* remove readonly in DGLSubGraph.

* cache subgraph properly.

* accelerate getting adjacency.

* split infer.

* lazy eval edges.

* specify inference mode.

* update for new sampler.

* use new mxnet sampling api.

* fix indent.

* remove profiling code.

* remove mxnet from sampler.

* return a lambda function for graph edges.

* add docs for immutable subgraph.

* Revert "return a lambda function for graph edges."

This reverts commit 0de5d7f100e230c518a3fb8976a6227f474d09ee.

* get parent_eid.
2018-12-02 12:43:25 -05:00
Haibin Lin 5278220f85 update s3 links (#215) 2018-12-02 20:48:29 +08:00
Gan Quan ac932c665b [Model] Junction Tree VAE update (#157)
* cherry picking optimization from jtnn

* adding official code.  TODO: fix DGLMolTree

* updating to current api.  vae test still failing

* reverting to list stacking

* reverting to list stacking

* cleaning x flags (stupid windows)

* cleaning x flags (stupid windows)

* adding stats

* optimization

* updating dgl stats

* update again

* more optimization

* looks like computation is faster

* removing profiling code

* cleaning obsolete code

* remove comparison warning

* readme update

* official implementation got a lot faster

* minor fixes

* unbatch by slicing frames

* working around unbatch

* reduce pack

* oops

* support frame read/write with slices

* reverting back to readout as unbatch-by-slicing slows down backward

* reverting to unbatch by splitting; slicing is unfriendly to backward

* replacing lru cache with static object factory

* cherry picking optimization from jtnn

* unbatch by slicing frames

* reduce pack

* oops

* support frame read/write with slices

* reverting to unbatch by splitting; slicing is unfriendly to backward

* replacing lru cache with static object factory

* replacing Scheme object with namedtuple

* forgot the find edges interface

* subclassing namedtuple

* updating to the latest api spec

* bugfix

* bfs with edges

* dfs toy test case

* clean up

* style fix

* bugfix

* update to latest api; include traversal

* replacing with readout

* simplify decoder

* oops

* cleanup

* reducing number of sets

* more speed up

* profile results

* random fixes

* fixing tvmarray handling incontiguous dlpack input

* fancier dataloader

* fix a potential context mismatch

* todo: support pickling or using scipy in multiprocessing load

* pickling support

* resorting to suggested way of pickling

* custom attribute pickling check

* working around a weird pytorch pickling bug

* including partial frame case

* enabling multiprocessing dataloader

* pickling everything now

* really works

* oops

* updated profiling results

* cleanup

* fix as requested

* cleaning random blank lines

* removing profiler outputs

* starting decoding

* testing, WIP

* tree decoding

* graph decoding, WIP

* graph decoding works

* oops

* fixing legacy apis

* trimming number of candidate structures

* sampling cleanups

* removing comparison test

* updated description
2018-12-02 03:32:01 -05:00
Gan Quan 4682b76e60 fixing stupid segfault in edge_ids (#216) 2018-12-02 03:12:45 -05:00
Lingfan Yu 168794cdbc remove text using code line numbers (#217) 2018-12-01 21:04:39 -05:00
Minjie Wang 7156c7163b [Doc] new at glance doc draft (#214)
* WIP

* new at glance
2018-12-01 19:03:57 -05:00
Lingfan Yu 53ea03ad82 [Doc] Fix doc bug and warning (#213)
* fix dgmg tutorial indentation

* bug in incidence_matrix docstring

* remove print
2018-12-01 18:58:56 -05:00
Lingfan Yu 0f121eeb76 [Doc] RGCN tutorial (#206)
* rgcn tutorial

* data processing for rgcn

* many fix

* requirements for rgcn tutorial

* fix comments

* description of rgcn dataset

* author

* bug fix

* move all dataset to s3
2018-12-01 14:58:13 -05:00
Haibin Lin cccc2339c1 [Dataset] update s3 links (#209) 2018-12-01 14:45:30 -05:00
Mufei Li e02acc4c01 [Doc] Tutorial production for DGMG (#204) 2018-12-01 00:32:43 -05:00
Haibin Lin 70dc7cd184 [Doc] Dgl glance tutorial improvement (#197)
* update dgl at a glance

* update title

* Update 1_first.py
2018-11-30 23:53:23 -05:00
Minjie Wang bcf8e1194b [Bugfix][CI] force download data everything (#208)
* poke ci

* poke again

* force download data everytime
2018-11-30 23:44:40 -05:00
VoVAllen 91141a9826 [Dataset] Move file from dropbox to s3 (and fix ci fail due to sst dataset) (#205)
* move file from dropbox to s3

* fix ci and move sst to s3

* remove blank line
2018-11-30 19:44:55 -05:00
Zihao Ye c80dc5e012 [model] Tree-LSTM update & hotfix(dropbox link) (#202)
* change the signature of node/edge filter

* upd filter

* Support multi-dimension node feature in SPMV

* stable version

* hotfix

* upd tutorial

* upd README
2018-11-30 19:39:49 -05:00
Mufei Li 3a0f86a663 [Doc] data (#198)
* Fix dataset

* Track data APIs
2018-11-30 11:40:22 -05:00
Mufei Li 485f6d3abd [Doc] udf (#195)
* Docstring for udf

* Track udf docs

* Improve

* Improved

* Delete udf.py

* Improve
2018-11-30 00:38:58 -05:00
Haibin Lin 9e8261735f [Doc] Minor doc update for mxnet/immutable_graph_index (#194) 2018-11-29 20:59:13 -05:00
Mufei Li 41c9c3b94b [Doc] Fix batched_graph doc (#193) 2018-11-29 12:23:10 -05:00
Mufei Li 8dc6784b12 [Doc] docstring for BatchedDGLGraph + some fix (#184)
* Update doc

* Update batched_graph.py

* Fix

* Fix

* Fix

* Fix
2018-11-29 09:53:16 -05:00
VoVAllen e17c41c0da [Model][Tutorial] Fix capsule memory leak (#185)
* fix memory leak & Remove unnecessary initializer

* change confused name

* fix name

* Move func outside loop

* fix name inconsistency
2018-11-29 09:40:21 -05:00
Gan Quan 79fe09d3e0 forgot to remove conda branch meta (#190) 2018-11-29 02:26:39 -05:00
Gan Quan a8fa85c7d5 Conda builds (#187)
* conda recipe

* switching to source build

* trying windows build

* wip

* windows build works
2018-11-29 02:21:53 -05:00
Mufei Li a0d0b1ea00 [Model] Fix + batched DGMG (#175)
* DGMG with batch size 1

* Fix

* Adjustment

* Fix

* Fix

* Fix

* Fix

* Fix has_node and __contains__

* Batched implementation for DGMG

* Remove redundant dependency

* Adjustment

* Fix

* Add comments
2018-11-29 00:13:30 +08:00
HQ 5cda368d79 [Model] SBM hotfix (#137)
* [Model]SBM hotfix

* [Model] remove backend in data
2018-11-27 11:58:03 -05:00
yifeim 02eb463ab1 report accuracy and fix training mask (#181) 2018-11-27 15:15:57 +08:00
Minjie Wang 2c2b747808 [Install] pip install (#180)
* add bdist_wheel build

* fix wheel build

* fix setup

* fix ci

* fix setup

* fix

* fix setup

* poke ci

* fix

* poke ci

* clean build

* fix ci

* addback the env
2018-11-27 00:47:01 -05:00
Minjie Wang b82b8e2d2a [Schedule] get rid of unnecessary graph.edges() call for v2v-spmv update_all (#179) 2018-11-27 00:37:03 -05:00
Minjie Wang 524e656dd6 [Bugfix] enable SPMV for multi-dimension node feature (#178)
* Support multi-dimension node feature in SPMV

* fix as requested
2018-11-26 11:12:28 -05:00
Minjie Wang b1eeb93449 [Bugfix][Runtime] Zero degree behaviors (#177)
* fix recv nodes are all 0deg; fix hybriddict does not through keyerror properly

* fallback to apply_nodes when all nodes are 0deg; WIP on pull spmv 0deg

* new 0deg behavior

* new 0deg behavior

* update mx utest for pull-0deg

* fix mx

* fix mx

* get rid of unnecessary sort-n-unique
2018-11-25 22:33:08 -05:00
Minjie Wang 4dfe7547e8 Fix num_rows bug in batched_graph (#169) 2018-11-23 02:49:59 +08:00
Zihao Ye 23e2e83b5e [API] change the signature of node/edge filter (#166)
* change the signature of node/edge filter

* upd filter
2018-11-22 13:00:36 -05:00