提交

提交图

3215 次代码提交

作者 SHA1 备注 提交日期
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
Lingfan Yu deb653f8dc [Runtime] Scheduler and Executor (#140)
* executor api

* draft executor interface

* WIP

* revert changes to avoid conflict with api change

* core scheduling logic

* WIP: build graph adj

* incidence matrix for in edges

* support incidence matrix for partial recv nodes

* improve

* build adjmat in scheduler

* graph store

* get degree bucketing schedule

* connect to c++ degree bucketing

* conceptual executor creation code

* executor comments

* fix

* more executor comments

* WIP: full send_and_recv schedule

* most schedulers

* simplify scheduler

* executors

* runtime

* builtin function base class

* adj indices and shape

* completely refactor scheduler

* rename and move bundled out to function.py

* use_edge_feature in msg func

* rewrite scheduler

* node edge executor

* connect with graph api

* handle zero degree

* misc

* fix test cases

* fix a good many bugs...

* remove old scheduler

* push and pull

* fix send recv

* c++ lint

* fix batched send recv

* hot fix for mxnet

* typo

* write back executor

* apply node edge

* clean up, doc string

* fix as requested

* refactor

* fix

* WIP

* WIP

* ir draft

* more on ir

* WIP: spmv schedule

* WIP

* recv schedule

* refactor

* WIP

* snr degree bucketing

* snr scheduler

* move prog to graph.py; rename

* unittest for send/recv

* remove some legacy codes

* WIP: update_all

* pass test_basics

* passed all current utests

* more utests; fix mx utest

* WIP: fixing zero deg initial value

* some tests

* fix 0deg problem

* fix mx

* fix mx

* some notes

* fix as requested
2018-11-22 12:15:50 -05:00
Mufei Li 3e8b63ecd3 [Model] DGMG Training with Batch Size 1 (#161)
* DGMG with batch size 1

* Fix

* Adjustment

* Fix

* Fix

* Fix

* Fix
2018-11-22 10:29:25 -05:00
VoVAllen bf6d0025aa [Bugfix] fix gpu manual choice for capsule example (#164) 2018-11-21 22:01:24 -05:00
Minjie Wang 0ec1a492b3 [Frame] change num rows behavior (#163)
* move initializer to column scheme; change num rows behavior

* poke mx ci

* fix mx utest in append

* fix bug in add edges

* utest for updating partial rows

* fix bug in from_networkx and from_scipy_matrix

* revert per-col initializer change

* fix pickle utest
2018-11-21 11:50:46 -05:00
Da Zheng 440aecee8f [Test] add tests for built-in functions (#159)
* add spmv tests on mxnet.

* test backward.

* add tests for pull.

* use mxnet backend.

* address comment.

* add comments.
2018-11-20 23:39:54 -05:00
Da Zheng 5f89cd662e Fix the convergence problem in SSE. (#162)
* add a test.

* move file.

* debug

* add degree normalization.

* it can converge now.

* remove hidden_data.

* use readonly graph index.

* use subgraph loader.

* refactor code and remove intermediate data.

* split inference.

* mix cpu and gpu training.

* fix mxnet sparse_matrix constructor.

* convert tensor context.

* set up the prediction model.

* load mxnet csr directly.

* add timing.

* move test_sse.py to sse_batch.py

* fix the tensor version of SSE.

* update README.
2018-11-20 15:57:52 +08:00
Gan Quan e515f5311b [Frame] supporting partial row on first column and individual column initializers (#160)
addresses issue #156
2018-11-19 18:18:36 -05:00
Minjie Wang cab1fdf2ec [Doc] doc update (#158)
* minor fix

* jake's change
2018-11-17 01:53:53 -05:00
Gan Quan 79a510252b Pickling support (#155)
* pickling support

* resorting to suggested way of pickling

* custom attribute pickling check

* working around a weird pytorch pickling bug

* including partial frame case

* pickling everything now

* fix as requested
2018-11-16 17:19:20 -05:00
Gan Quan 05f464f824 [Bugfix] fix args with incontiguous node and edge tensor (#153) 2018-11-15 10:55:15 -05:00
Gan Quan ed1854e800 [Bugfix] fix a potential context mismatch (#154) 2018-11-15 09:47:20 -05:00
Zihao Ye 53b9a4bdbc [DOC][Model] Update Tree-LSTM (#152)
* update tree lstm

* tree_lstm (new interface)

* simplify pop

* merge qipeng(root)

* upd tree-lstm & tutorial

* upd model

* new capsule tutorial

* capsule for new API

* fix deprecated API

* New tutorial and example

* investigate gc problem

* add viz code

* new capsule tutorial

* remove ipynb

* move u_hat

* add link

* add requirements.txt

* remove ani.save

* update ci to install requirements

* utf-8

* change seed

* graphviz requirement

* accelerate

* little format

* update some markup
2018-11-14 22:29:59 -05:00
Da Zheng 2389df8155 [API] Subgraph sampling API. (#138)
* add neighbor sampler.

* fix sampler.

* handle non-existing nodes differently in MapParentIdToSubgraphId.

* fix sampling tests.

* parallelize sampling.

* add more tests.

* fix a bug in subgraph loader.

* get number of sampled nodes directly.

* load mxnet CSR to DGLGraph directly.

* create graph from mxnet csr array directly.

* use subgraph loader in SSE.

* fix a bug in sampling.

* output framework tensor.

* fix.

* remove unnecessary code.

* fix tests.

* add SSE README.

* move to contrib.

* address comments.

* add tests.
2018-11-14 13:51:27 -08:00
Minjie Wang 048f6d7a30 [GraphIndex] refactor graph caching (#150)
* refactor graph caching

* fix mx test

* fix typo
2018-11-13 22:12:38 -05:00
Zihao Ye a9ffb59e66 [DOC][Model] Tree-LSTM example and tutorial (#146)
[DOC][Model] Tree-LSTM example and tutorial
2018-11-13 18:09:14 -05:00
Gan Quan 0452cc3c0c [Bug] Transpose adj is not correctly cached (#148) 2018-11-12 22:15:45 -05:00
Minjie Wang cd3e25a064 [CI] add graphviz (#147) 2018-11-12 15:18:22 -05:00
VoVAllen 7cb50072a4 [Doc][Model] New Capsule Tutorial & Example (#143)
* new capsule tutorial

* capsule for new API

* fix deprecated API

* New tutorial and example

* investigate gc problem

* add viz code

* new capsule tutorial

* remove ipynb

* move u_hat

* add link

* add requirements.txt

* remove ani.save

* update ci to install requirements

* add graphviz
2018-11-12 12:43:13 -05:00
Minjie Wang a95459e3a2 [CI] Improved CI (#141)
* change ci

* update ci

* update ci

* update ci

* update ci

* update ci

* update ci

* update ci

* update ci

* update ci

* update ci

* update ci

* update ci

* update ci

* update ci

* update ci

* nx package

* update ci

* update ci

* update ci

* fix

* mx dockerfile by zhengda

* python3.6->3.5

* update ci image

* add tutorial test

* fix ci

* fix ssl problem

* minor change

* small fix on traversal utest

* fix syntax

* add matplotlib in image

* fix

* update ci

* update ci
2018-11-12 01:06:05 -05:00
Gan Quan 7cbb83db9f better bfs edge test (#142) 2018-11-11 23:09:39 -05:00
Gan Quan d6c80e0cf7 [API] bfs with edges (#132)
* bfs with edges

* dfs toy test case

* clean up

* style fix

* bugfix

* updating docs for bfs_edges and readouts
2018-11-11 17:19:55 -05:00
Gan Quan a36d0841b5 removing 3.6 f-string (#139) 2018-11-11 14:43:15 -05:00