文件历史

提交图

655 次代码提交

作者 SHA1 备注 提交日期
Chao Ma 49c4a9e4cc [DEMO] Remove duplicate code for sampling (#557)
* update

* update

* re-use single-machine code

* update

* use relative path

* update

* update

* update

* add __init__.py

* add __init__.py

* import sys, os

* fix typo

* update
2019-05-23 16:57:35 +08:00
Chao Ma f99725adbc all demo use python-3 (#555) 2019-05-23 10:38:05 +08:00
Quan (Andy) Gan d8c69d53e2 Fixing typo in JTNN after interface change (#536) 2019-05-16 09:43:03 +08:00
Ziyue Huang e89519150f disable pytorch reddit (#512) 2019-04-28 11:24:14 +08:00
Ziyue Huang 3b96299a45 sampling (#505) 2019-04-23 12:14:38 +08:00
Zhengwei ad9da36add [Model] Add DGI Model (#501) 2019-04-22 11:37:06 +08:00
Aymen Waheb 3f4645918d [Model] Add edge dropout to APPNP (#493)
* [Model] Add edge dropout to APPNP

[Model] Add edge dropout to APPNP

* [Model] Refactor the sampling examples (#498)

* reorganize sampling code.

* speedup gcn_ns.

* speed up gcn_cv

* fix graphsage_cv.

* undo the modification.

* accel training.

* update readme.

* [Model] Add edge dropout to APPNP

[Model] Add edge dropout to APPNP

update
2019-04-18 20:58:28 +08:00
kitaev-chen a3febc061b [Model] Add GIN Model (#471)
* add gin model

* convert dataset.py to data_ont_the_fly way and put it into dgl.data module

* convert dataset.py to data_ont_the_fly way and put it into dgl.data module
python code checked

* modified document and reference TUDataset; checked python part and bypass cpp part due to error

* change tensor to numpy in dataset and transform in collate@Dataloader

* Change minor format issue

Change minor format issue

* moved logging; adjusted tqdm etc
2019-04-17 14:57:07 +08:00
Quan (Andy) Gan e2e04329a6 [BUG] Update URL for JTNN dataset 2019-04-11 22:23:34 +08:00
Hao Zhang 039a711da1 [Model] fix self-edge bug in GCN and GAT. (#482)
* Update gcn_mp.py

* Update train.py

* Update train.py

* Update train.py

* Update gat_batch.py

* Update gat_batch.py

* Update gcn_mp.py
2019-04-10 09:04:19 -04:00
Aymen Waheb fa887f69f4 [Model] Add APPNP model (#480)
* [Model] Add APPNP model

* update

* Revert "update"

This reverts commit a8e42d19c8b421fb382305a8f04d7426e78ea53c.

* update

* Update appnp.py
2019-04-09 19:54:09 +08:00
Da Zheng 4ea42e3e08 fix gcn. (#469) 2019-04-05 10:56:09 -07:00
Quan (Andy) Gan 6066fee935 [Model][Feature] PinSage & Random Walk with Restart (#453)
* random walk traces generation

* remove outdated comments

* oops put in the wrong place

* explicit inline

* moving rand_r to util

* pinsage-like model on movielens

* the code runs now

* support cuda

* using readonly graph

* moving random walk to public function

* per-thread seed and openmp support

* pinsage-like model on movielens

* the code runs now

* support cuda

* using readonly graph

* using C random walk

* removing profile decorators

* param initialization

* no grad

* leaky relu fixes everything

* train and save

* WIP

* WIP

* WIP

* seems to work

* evaluation output

* swapping order of val/test and train

* debug

* hyperparam tuning

* prior/training dataset split changes

* random walk reorg

* random walk with restart

* signed comparison fix

* migrating random walk to nodeflow

* Revert "migrating random walk to nodeflow"

This reverts commit f2565347cced7c912a58a529b257c033d9f375b7.

* add README and remove dataset

* new endpoint

* lint

* lint x2

* oops forgot test

* including bpr - better for baseline

* addressing fixes

* throwing random walks out from SamplerOp class

* forgot to move RandomWalk; why did this even work?

* removing legacy garbage

* add todo

* address comments

* stupid bug fix

* call ndarrayvector converter to handle traces
2019-03-29 13:01:18 +08:00
Giuseppe Futia 2dff1aba83 [Bugfix] Fix shape of test_relation tensor in RGCN (#464) 2019-03-28 14:34:07 -04:00
Lingfan Yu d3c24cc25b [BugFix] Fix bug in RGCN data processing and use index_select to improve speed (#429)
* use index_select instead of __getitem__

* fix bug in dataset processing

* fix edge_type shape bug

* comments
2019-03-04 10:52:21 -05:00
Minjie Wang 565f0c88fc [WIP] [NN] Refactor NN package (#406)
* refactor graph conv

* docs & tests

* fix lint

* fix lint

* fix lint

* fix lint script

* fix lint

* Update

* Style fix

* Fix style

* Fix style

* Fix gpu case

* Fix for gpu case

* Hotfix edgesoftmax docs

* Handle repeated features

* Add docstring

* Set default arguments

* Remove dropout from nn.conv

* Fix

* add util fn for renaming

* revert gcn_spmv.py

* mx folder

* fix wierd bug

* fix mx

* fix lint
2019-02-25 18:41:21 -05:00
hbsun2113 8c750170e8 [Model][Pytorch] GraphSAGE (#403)
* simple implemention of GraphSAGE

* update the abstract method and `torch.nn` modules are more utilized
2019-02-22 15:48:53 -05:00
Tianyi cdbeb17f2d [Model] Simplifying Graph Convolutional Networks. (#405)
* [Model] Simplifying Graph Convolutional Networks.

* - improve code stype (use nn.Linear)
- use `bias` option
- update hyperparameter and corresponding performance
2019-02-22 15:17:26 +08:00
Hao Zhang 788d8dd4ad [Model]PPI dataloader and inductive learning script. (#395)
* Create ppi.py

* Create train_ppi.py

* Update train_ppi.py

* Update train_ppi.py

* Create gat.py

* Update train.py

* Update train_ppi.py

* Update ppi.py

* Update train_ppi.py

* Update ppi.py

* Update train_ppi.py

* Update train_ppi.py

* Update ppi.py

* Update train_ppi.py

* update docs and readme
2019-02-17 16:41:30 -05:00
Zihao Ye 1ea0bcf486 [Model] fix link & beam search (#394) 2019-02-17 13:32:36 -05:00
Zihao Ye ae1806f651 [Bugfix] RGCN tutorial (#388)
* fix rgcn tutorial

* small fix

* upd
2019-02-15 11:14:56 +08:00
Minjie Wang 91b7382378 [Model] update gat (#390)
* update gat: add minus max for softmax

* small fix
2019-02-13 14:53:41 -05: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
Lingfan Yu 29dd22e666 [Model] Support Multi-GPU for Transformer model (#356)
* multi-process version of transformer

* lots of fix

* fix bugs and accum gradients for multiple batches

* many fixes

* minor

* upd

* set torch device

* fix bugs

* fix and minor

* comments and clean up

* uncomment viz code
2019-02-11 20:17:45 -05:00
Quan (Andy) Gan 12d706300c [Hotfix] fixing zero shaped tensor problems for PyTorch 1.0.0 in JTNN example (#371) 2019-01-31 21:07:09 -05:00
Minjie Wang ddc2faa547 [Bugfix] Fix gat residual bug (#355)
* fix gat residual bug

* fix the residual addition; output heads; add some shape notations;

* minor

* fix the output head average

* add requests package in requirement
2019-01-12 08:37:01 -05:00
Minjie Wang efae0f97d8 [Model] Improve GAT models (#348)
* two better GAT implementations

* update numbers

* use version switch for spmm

* add missing dropout and output heads
2019-01-11 02:02:06 -05:00
brett koonce 707334ce85 minor spelling tweaks (#349)
* minor spelling tweaks

* Update CONTRIBUTORS.md
2019-01-10 12:02:31 -05:00
Sheng Zha 1e50cd2eda [Model][MXNet] MXNet Tree LSTM example (#279)
* TreeLSTM MXNet example

* hybridize

* add glove download

* usability

* Update README.md
2018-12-27 10:57:23 +08:00
Hao Zhang 896dc50e70 [Model] fix bugs in pygat and mxgat. (#323)
* mxgat

* pygat

* Update gat_batch.py

* add train

* Update gat_batch.py

* Update gat.py

* Update gat_batch.py
2018-12-23 20:45:25 -05:00
Minjie Wang f35ac5443e [Graph][Model] Cache adj & inc; MX GCN (#307)
* add cache to adj and incmat

* Fix bug in cached adj/inc

* mx gcn spmv runnable; acc debugging...

* fix bug in mx gcn that loss is not correctly calculated

* fix mx utest

* fix as requested

* use raw parameter tensors rather than dense layer

* fix dropout

* Add numbers in readme
2018-12-16 22:51:01 -05:00
Ziyue Huang cffa4034f5 [Model] fix GCN (#305)
* mxnet gcn spmv

* update readme

* fix gcn

* pytorch gcn

* update readme
2018-12-16 12:44:41 -05:00
Zihao Ye 9f32554296 [Model]Transformer (#186)
* change the signature of node/edge filter

* upd filter

* Support multi-dimension node feature in SPMV

* push transformer

* remove some experimental settings

* stable version

* hotfix

* upd tutorial

* upd README

* merge

* remove redundency

* remove tqdm

* several changes

* Refactor

* Refactor

* tutorial train

* fixed a bug

* fixed perf issue

* upd

* change dir

* move un-related to contrib

* tutuorial code

* remove redundency

* upd

* upd

* upd

* upd

* improve viz

* universal done

* halt norm

* fixed a bug

* add draw graph

* fixed several bugs

* remove dependency on core

* upd format of README

* trigger

* trigger

* upd viz

* trigger

* add transformer tutorial

* fix tutorial

* fix readme

* small fix on tutorials

* url fix in readme

* fixed func link

* upd
2018-12-07 15:22:46 +08:00
Minjie Wang 3564fdc5f1 [Bugfix][Model] fix treelstm model (#274)
* fix bug after moving batcher out of dgl.data

* disable mx utest
2018-12-06 23:52:50 -05:00
Mufei Li ddf96ff9e7 [Doc] Fix DGMG part in README (#270)
* Fix DGMG epoch time

* Fix README for DGMG

* update
2018-12-06 10:54:07 -05:00
Da Zheng 1bbc885b40 move batcher to examples. (#269)
* move pytorch code to examples.

* fix.

* fix tutorial
2018-12-06 06:45:27 -08: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
Lingfan Yu f4a9a455f5 [Doc] Fix hyperlink in tutorial of tutorals (#260)
* fix all code tutorial links and typos in texts

* sse readme format

* fix

* sse paper link

* gat readme

* fix
2018-12-05 15:04:27 -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 378c264561 [Model] Fix GCN Normalization (#249)
* WIP

* lr -> 0.01

* new cora dataset

* normalization code

* minor format change

* normalization factor for deg bucket
2018-12-05 12:32:10 -05:00
Minjie Wang 21255b6528 [Bugfix] tolist and dependencies in dgl.data (#239)
* change Index.tolist -> Index.tonumpy; fix bug in traversal; remove dependencies in data

* fix import

* fix __all__ and some docstring
2018-12-05 12:20:59 -05:00
Da Zheng 78269cedfd [Doc] Update GCN README.md (#224)
* update readme.

* update readme
2018-12-04 12:07:05 -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
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
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
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
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
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