a00636a02b
[Tensorflow] Several nn & example ( #1191 )
...
* several nn example
* appnp
* fix lint
* lint
* add dgi
* fix
* fix
* fix
* fff
* docs
* 111
* fix
* change init
* change result
* tiaocan+1
* fix
* fix lint
* fix
* fix
2020-01-19 22:48:40 +08:00
3d6646288f
Update README.md ( #1184 )
2020-01-09 12:06:48 +08:00
bfc2d3030d
[BUG] Fix MovieLens feature assignment ( #1183 )
...
Fixes #1182 .
This is embarrassing.
2020-01-09 11:21:18 +08:00
e705118bcc
some comments ( #1132 )
2020-01-06 10:36:11 +08:00
6731ea3af7
[KVStore] Reduce memory cost of kvstore ( #1156 )
...
* API change of kvstore
* add demo for kvstore
* update
* remove duplicated log
* change queue size
* update
* update
* update
* update
* update
* update
* update
* update
* update
* fix lint
* change name
* update
* fix lint
* update
* update
* update
* update
* change message queue size to a python argument
* change default queue size to 2GB
* OMP_NUM_THREADS=1
* add multiple NICs support for kvstore
* test
* fix lint
* update
* update
* update
* update
* update
* update
* update
* fix lint
* fix lint
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* fix lint
2019-12-30 20:25:32 +08:00
4f02bb751b
[KVStore] Made kvstore can support multiple NICs ( #1150 )
...
* API change of kvstore
* add demo for kvstore
* update
* remove duplicated log
* change queue size
* update
* update
* update
* update
* update
* update
* update
* update
* update
* fix lint
* change name
* update
* fix lint
* update
* update
* update
* update
* change message queue size to a python argument
* change default queue size to 2GB
* OMP_NUM_THREADS=1
* add multiple NICs support for kvstore
* test
* fix lint
* update
* update
* update
* update
* update
* update
* update
* fix lint
* fix lint
* update
* update
* update
* update
2019-12-30 15:17:31 +08:00
d57ff78da1
[Small Fix] Change default message queue size of communicator to 2GB ( #1140 )
...
* API change of kvstore
* add demo for kvstore
* update
* remove duplicated log
* change queue size
* update
* update
* update
* update
* update
* update
* update
* update
* update
* fix lint
* change name
* update
* fix lint
* update
* update
* update
* update
* change message queue size to a python argument
* change default queue size to 2GB
* OMP_NUM_THREADS=1
2019-12-25 17:03:52 +08:00
e4ef8d1a80
[KVStore] API change of kvstore ( #1058 )
...
* API change of kvstore
* add demo for kvstore
* update
* remove duplicated log
* change queue size
* update
* update
* update
* update
* update
* update
* update
* update
* update
* fix lint
* change name
* update
* fix lint
* update
* update
* update
* update
* change message queue size to a python argument
2019-12-24 17:05:15 +08:00
2ab4d0e10f
[Doc] Hotfix for README ( #1119 )
2019-12-23 19:22:38 +08:00
d30a69bf27
[Backend] TF backend ( #978 )
...
* tf
* add builtin support
* fiix
* pytest
* fix
* fix
* fix some bugs
* fix selecting
* fix todo
* fix test
* fix test fail in tf
* fix
* fix
* fix gather row
* fix gather row
* log backend
* fix gather row
* fix gather row
* fix for pytorch
* fix
* fix
* fix
* fix
* fix
* fix tests
* fix
* fix
* fix
* fix
* fix
* fix
* fix convert
* fix
* fix
* fix
* fix inplace
* add alignment setting
* add debug option
* Revert "add alignment setting"
This reverts commit ec63fb3506ea84fff7d447a1fbdfd1d5d1fb6110.
* tf ci
* fix lint
* fix lint
* add tfdlpack
* fix type
* add env
* fix backend
* fix
* fix tests
* remove one_hot
* remove comment
* remove comment
* fix
* use pip to install all
* fix test
* fix base
* fix
* fix
* add skip
* upgrade cmake
* change version
* change ci
* fix
* fix
* fix
* fix
* fix seg fault
* fix
* fix python version
* fix
* try fix
* fix
* fix
* tf takes longer time in ci
* change py version
* fix
* fix
* fix oom
* change kg env
* change kg env
* 啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊
* 我再也不搞各种乱七八糟环境了……
* use pytest
* Chang image
2019-12-20 15:56:51 +08:00
cf9ba90fd4
[Chem] ACNN and various utilities ( #1117 )
...
* Add several splitting methods
* Update
* Update
* Update
* Update
* Update
* Fix
* Update
* Update
* Update
* Update
* Fix
* Fix
* Fix
* Fix
* Fix
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Finally
* CI
2019-12-19 11:54:40 +08:00
2448751c8e
Fixed the training file ( #1109 )
2019-12-14 18:20:05 +08:00
d16fb3ff9d
[Example] GCMC in PyTorch ( #1101 )
...
* upd
* fix
* upd
* upd
* upd
* fix
* upd
* upd
* upd
* upd
* upd
* upd
* upd
* upd
2019-12-14 08:18:48 +08:00
bc4f4352d1
[Chem Model Zoo] Hot fix for README ( #1071 )
...
* Fix outdated example
* Hot fix
2019-12-05 14:37:06 +08:00
fff3dd9593
[Example] add graphwriter pytorch example ( #1068 )
...
* upd
* fig edgebatch edges
* add test
* trigger
* add graphwriter pytorch example
* fix line break in graphwriter README
* upd
* fix
2019-12-05 02:44:15 +08:00
fdc58a8996
[Hotfix] Fix GAT example and clarify the usage of early stop ( #1065 )
...
* upd
* rm redundancy:
* upd
2019-11-30 23:43:15 +08:00
dca2580b84
fix minor error: GCNLayerSAGE->GraphSAGELayer ( #1010 )
...
fix minor error: GCNLayerSAGE->GraphSAGELayer
2019-11-27 01:54:56 +08:00
85c9ff01c9
[Example] Finish adaptive sampling ( #998 )
...
* [Example] Finish adaptive sampling
* refactor code and use argparse for command-line usage
* Update README.md
* add node_per_layer command-line argument
2019-11-26 13:47:26 +08:00
1915fdb984
Make print of GCN example of mxnet and pytorch exact same. ( #994 )
2019-11-13 12:47:06 +08:00
a0193fd539
Small change for kvstore api ( #981 )
...
* Small change for kvstore api
* fix ci
* fix ci
2019-11-08 16:25:12 +08:00
7897fa3b9c
add init api on kvserver ( #975 )
2019-11-05 16:21:35 +08:00
cccde032f4
[kvstore] Performance improvement for distributed kvstore ( #972 )
...
* Performance improvment for distributed kvstore
* update
* update
2019-11-04 18:27:00 +08:00
fdd0fe651d
hotfix ( #971 )
2019-11-04 12:11:53 +08:00
9a0511c8e9
[NN] nn modules & examples update ( #890 )
...
* upd
* damn it
* fuck
* fuck pylint
* fudge
* remove some comments about MXNet
* upd
* upd
* damn it
* damn it
* fuck
* fuck
* upd
* upd
* pylint bastard
* upd
* upd
* upd
* upd
* upd
* upd
* upd
* upd
* upd
2019-11-03 17:29:42 -05:00
82499e602b
[Bug Fix] Fix package reliability bug of networkx ( #949 )
...
* upd
* fig edgebatch edges
* add test
* trigger
* Update README.md for pytorch PinSage example.
Add noting that the PinSage model example under
example/pytorch/recommendation only work with Python 3.6+
as its dataset loader depends on stanfordnlp package
which work only with Python 3.6+.
* Provid a frame agnostic API to test nn modules on both CPU and CUDA side.
1. make dgl.nn.xxx frame agnostic
2. make test.backend include dgl.nn modules
3. modify test_edge_softmax of test/mxnet/test_nn.py and
test/pytorch/test_nn.py work on both CPU and GPU
* Fix style
* Delete unused code
* Make agnostic test only related to tests/backend
1. clear all agnostic related code in dgl.nn
2. make test_graph_conv agnostic to cpu/gpu
* Fix code style
* fix
* doc
* Make all test code under tests.mxnet/pytorch.test_nn.py
work on both CPU and GPU.
* Fix syntex
* Remove rand
* Add TAGCN nn.module and example
* Now tagcn can run on CPU.
* Add unitest for TGConv
* Fix style
* For pubmed dataset, using --lr=0.005 can achieve better acc
* Fix style
* Fix some descriptions
* trigger
* Fix doc
* Add nn.TGConv and example
* Fix bug
* Update data in mxnet.tagcn test acc.
* Fix some comments and code
* delete useless code
* Fix namming
* Fix bug
* Fix bug
* Add test for mxnet TAGCov
* Add test code for mxnet TAGCov
* Update some docs
* Fix some code
* Update docs dgl.nn.mxnet
* Update weight init
* Fix
* reproduce the bug
* Fix concurrency bug reported at #755 .
Also make test_shared_mem_store.py more deterministic.
* Update test_shared_mem_store.py
* Update dmlc/core
* networkx >= 2.4 will break our examples
* Update tutorials/requirements
* fix selfloop edges
* upd version
2019-10-29 23:34:31 -04:00
98c1448bfc
[Cleanup] Change Byte to Bool for training masks ( #954 )
...
* Change Byte to Bool for training masks
* Check if module has Bool, otherwise use Byte
2019-10-29 17:14:08 -04:00
86cf154be9
[Misc] Update README(s) ( #962 )
...
* Update
* Update
2019-10-29 15:37:25 +08:00
421b05e7b8
[Model] Clean up for Cluster GCN ( #940 )
...
* Update
* Update
* Update
* Update
* Update
* Update
* Fix
* Fix
2019-10-26 21:02:47 +08:00
52c7ef49fc
[Model Zoo] AttentiveFP ( #955 )
...
* Update
* Fix style
* Update
* Update
* Fix
* Update
* Update
2019-10-26 01:57:34 +08:00
577cf2e63b
[Model Zoo] Refactor and Add Utils for Chemistry ( #928 )
...
* Refactor
* Add note
* Update
* CI
2019-10-22 00:47:45 +08:00
ff9f67ee0d
Send any shape of tensor rather than matrix ( #942 )
2019-10-21 17:27:03 +08:00
ae3102d3a2
[Dataset] Migration for Chemistry Datasets ( #926 )
...
* Update
* Update
2019-10-21 16:38:28 +08:00
c37076dfe6
[Model] Gated Graph Neural Network for bAbI tasks ( #880 )
...
* rrn model and sudoku
* add README
* refine the code, add doc strings
* add sudoku solver
* add example for sudoku_solver
* ggnn example
* Rewrite README file
* fix typos
2019-10-21 16:07:32 +08:00
9c790b1143
[Dataset] Add CoraFull, Amazon, KarateClub, Coauthor Dataset ( #855 )
...
* convert np.ndarray to backend tensor
* add datasets
* add qm7
* add dataset
* add dataset
* fix
* change ppi
* tu dataset
* add datasets
* fix
* fix
* fix
* fix
* add docstring
* docs
* doc
2019-10-01 01:19:59 +08:00
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
51a73504fe
[Model] Add Pytorch example for Cluster GCN ( #877 )
...
* initial commit of cluster GCN
* update readme
* fix small bugs running ppi
* nearly sota ppi training script&update readme
* rm unused imports&add shebang line to scripts
* minor comments&readme appended
* add rnd seed control&update readme
2019-09-28 10:10:42 +08:00
c03046a08f
[Model][Hetero] HAN ( #868 )
...
* Add HAN
* Fix
* WIP; load raw ACM dataset
* DGL's own preprocessing with metapath coalescer
* various fixes
* comparison against simple logistic regression
* rename
* fix test
2019-09-27 12:53:16 +08:00
9737b2700f
[Model][Hetero] GCMC using new hetero APIs ( #860 )
...
* init
* rm data
* README
* fix rating values that are not decimal
* rm stale codes
* small fix
* upd
* rewrite decoder
* fix many
* many fix; performance matched
* upd; handle sparse input
* upd
* address comments
* more docstring; download data automatically
* shared param mode
2019-09-26 17:50:38 -04:00
e3921d5dec
[Sampler] Metapath sampler for metapath2vec ( #861 )
...
* metapath sampler
* lint & fixes
* lint x2
* lint x3
* fix windows
* remove max_cycle argument
* add todo note
2019-09-26 16:42:33 -04:00
7782276962
[KVStore] Distributed kvstore ( #851 )
...
* update
* speedup
* add some comments
2019-09-11 14:54:50 +08:00
9df8cd3242
[Model Zoo] Fix JTNN ( #843 )
...
* Update
* Update
* Update
* Update
* Update
2019-09-10 16:14:37 +08:00
6a4b5ae98a
[Feature] Edge softmax on a subset of edges in the graph. ( #842 )
...
* upd
* add test
* fix
* upd
* merge
* hotfix
* upd
* fix
2019-09-09 13:44:56 +08:00
189c2c0907
[Model Zoo] Refactor Model Zoo for Chemistry ( #839 )
...
* Update
* Update
* Update
* Update fix
* Update
* Update
* Refactor
* Update
* Update
* Update
* Update
* Update
* Update
* Fix style
2019-09-07 20:01:09 +08:00
5b417683f6
[docfix] Update the README of metapath2vec ( #832 )
...
* upd
* fig edgebatch edges
* add test
* trigger
* upd
2019-09-05 16:43:12 +08:00
8731050394
[Example] Fix GIN Example ( #826 )
...
* Add serialization
* fix gin
* Add serialization
* fix gin
* fix
2019-09-03 17:55:27 +08:00
ddeb86f91c
[Example] metapath2vec ( #764 )
...
* Prepare for metapath sampler
This file is just for reviewing metapath sampling algorithm (Python version).
* Delete metapath_sampler
* Prepare for metapath sampler
This file is just for reviewing metapath sampling algorithm (Python code).
* Add files via upload
* Create metapath2vec.md
* Add files via upload
* Delete data_handler.py
* Delete word2vec.py
* Delete word_train.py
* Add files via upload
Metapath2vec implementations. Metapath2vec++ needs negative sampler optimization.
* Delete shuffle_training.py
* Delete test.py
* Add files via upload
* Delete sampler.py
* Delete metapath_sampler.md
* Add files via upload
* Update and rename shuffle_training.py to metapath2vec.py
* Update reading_data.py
* Update metapath2vec.md
* Update metapath2vec.md
* Update metapath2vec.md
* Update metapath2vec.md
* Update metapath2vec.md
* Create label 2
* Delete label 2
* Create testing.md
* Add files via upload
* Create sample.md
* Add files via upload
* Delete sampler.py
* Add files via upload
* Delete googlescholar.8area.author.label.txt
* Delete googlescholar.8area.venue.label.txt
* Delete testing.md
* Delete id_author.txt
* Delete id_conf.txt
* Delete paper.txt
* Delete paper_author.txt
* Delete paper_conf.txt
* Delete sample.md
* Delete sampler.py
* Add files via upload
* Add files via upload
* Add files via upload
* Delete reading_data.py
* Add files via upload
* Add files via upload
* Delete metapath2vec.py
* Add files via upload
* Rename shuffle_training.py to metapath2vec.py
* Update metapath2vec.md
* Delete reading_data.py
* add comments and remov e commented codes
2019-09-02 00:03:21 -07:00
a9a0592c3e
Hot fix ( #816 )
2019-09-01 02:11:03 +08:00
1b5820a5d6
Fix ( #806 )
2019-08-29 01:18:04 +08:00
e17add5602
[NN] Add MXNet impl for TAGCN module. ( #799 )
...
* upd
* fig edgebatch edges
* add test
* trigger
* Update README.md for pytorch PinSage example.
Add noting that the PinSage model example under
example/pytorch/recommendation only work with Python 3.6+
as its dataset loader depends on stanfordnlp package
which work only with Python 3.6+.
* Provid a frame agnostic API to test nn modules on both CPU and CUDA side.
1. make dgl.nn.xxx frame agnostic
2. make test.backend include dgl.nn modules
3. modify test_edge_softmax of test/mxnet/test_nn.py and
test/pytorch/test_nn.py work on both CPU and GPU
* Fix style
* Delete unused code
* Make agnostic test only related to tests/backend
1. clear all agnostic related code in dgl.nn
2. make test_graph_conv agnostic to cpu/gpu
* Fix code style
* fix
* doc
* Make all test code under tests.mxnet/pytorch.test_nn.py
work on both CPU and GPU.
* Fix syntex
* Remove rand
* Add TAGCN nn.module and example
* Now tagcn can run on CPU.
* Add unitest for TGConv
* Fix style
* For pubmed dataset, using --lr=0.005 can achieve better acc
* Fix style
* Fix some descriptions
* trigger
* Fix doc
* Add nn.TGConv and example
* Fix bug
* Update data in mxnet.tagcn test acc.
* Fix some comments and code
* delete useless code
* Fix namming
* Fix bug
* Fix bug
* Add test code for mxnet TAGCov
* Update some docs
* Fix some code
* Update docs dgl.nn.mxnet
* Update weight init
* Fix
2019-08-28 13:19:17 +08:00
14bffe9728
[NN] Renaming NearestNeighborGraph to KNNGraph ( #802 )
...
* initial commit
* second commit
* another commit
* change docstring
* migrating to dgl.nn
* fixes
* docs
* lint
* multiple fixes
* doc
* renaming nearest neighbor graph
2019-08-28 12:20:35 +08:00
VoVAllen