* Add weight based edge sampler
* Can run, edge weight work.
TODO: test node weight
* Fix node weight sample
* Fix y
* Update doc
* Fix syntex
* Fix
* Fix GPU test for sampler
* Fix test
* Fix
* Refactor EdgeSampler to act as class object not function that it
can record its own private states.
* clean
* Fix
* Fix
* Fix run bug on kg app
* update
* update test
* test
* Simply python API and fix some C code
* Fix
* Fix
* Fix syntex
* Fix
* Update API description
* add replacement for edge sampler
* Now edge sampler support replacement and no-replacement
* Fix
* Fix
* change kg/app to use edge sampler with replacement config
* Update replacement algo
* Fix syntax
* Update
* Update
Co-authored-by: Da Zheng <zhengda1936@gmail.com>
* Add weight based edge sampler
* Can run, edge weight work.
TODO: test node weight
* Fix node weight sample
* Fix y
* Update doc
* Fix syntex
* Fix
* Fix GPU test for sampler
* Fix test
* Fix
* Refactor EdgeSampler to act as class object not function that it
can record its own private states.
* clean
* Fix
* Fix
* Fix run bug on kg app
* update
* update test
* test
* Simply python API and fix some C code
* Fix
* Fix
* Fix syntex
* Fix
* Update API description
* 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
* Update Knowledge Graph CI with new Docker image
* Remove unused line_profierx
* Poke Jenkins
* Update test with exit code check and simplify docker
* Update Jenkinsfile to make app test a standalone stage
* Update kg_test
* Update Jenkinsfile
* Make some KG test parallel
* Update
* KG MXNet does not support ComplEx
* Update Jenkinsfile
* Update Jenkins file
* Change torch-1.2 to torch-1.2-cu92
* ci
* Update ubuntu_install_mxnet_cpu.sh
* Update ubuntu_install_mxnet_gpu.sh
* We only need to test train and eval script.
Delete some test code
* 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
* 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
* init version.
* change default value of regularization.
* avoid specifying adversarial_temperature
* use default eval_interval.
* remove original model.
* remove optimizer.
* set default value of num_proc
* set default value of log_interval.
* don't need to set neg_sample_size_valid.
* remove unused code.
* use uni_weight by default.
* unify model.
* rename model.
* remove unnecessary data sampler.
* remove the code for checkpoint.
* fix eval.
* raise exception in invalid arguments.
* remove RowAdagrad.
* remove unsupported score function for now.
* Fix bugs of kg
Update README
* Update Readme for mxnet distmult
* Update README.md
* Update README.md
* revert changes on dmlc
* add tests.
* update CI.
* add tests script.
* reorder tests in CI.
* measure performance.
* add results on wn18
* remove some code.
* rename the training script.
* new results on TransE.
* remove --train.
* add format.
* fix.
* use EdgeSubgraph.
* create PBGNegEdgeSubgraph to simplify the code.
* fix test
* fix CI.
* run nose for unit tests.
* remove unused code in dataset.
* change argument to save embeddings.
* test training and eval scripts in CI.
* check Pytorch version.
* fix a minor problem in config.
* fix a minor bug.
* fix readme.
* Update README.md
* Update README.md
* Update README.md
* 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
* 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
* Start implementing masked-mm kernel.
Add base control flow code.
* Add masked dot declare
* Update func/variable name
* Skeleton compile OK
* Update Implement. Unify BinaryDot with BinaryReduce
* New Impl of x_dot_x, reuse binary reduce template
* Compile OK.
TODO:
1. make sure x_add_x, x_sub_x, x_mul_x, x_div_x work
2. let x_dot_x work
3. make sure backward of x_add_x, x_sub_x, x_mul_x, x_div_x work
4. let x_dot_x backward work
* Fix code style
* Now we can pass the tests/compute/test_kernel.py for add/sub/mul/div forward and backward
* Fix mxnet test code
* Add u_dot_v, u_dot_e, v_dot_e unitest.
* Update doc
* Now also support v_dot_u, e_dot_u, e_dot_v
* Add unroll for some loop
* Add some Opt for cuda backward of dot builtin.
Backward is still slow for dot
* Apply UnravelRavel opt for broadcast backward
* update docstring
* PBG negative edge sampler.
* add a positive edge to make it regular, handle last batch.
* exclude all positive edges in the parent graph.
* just uniformly sample negative nodes.
* fix lint.
* shuffle one-side nodes of positive edges.
* just uniformly sample negative nodes.
* change the data type.
* address comment.
* remove commented code.
Quan (Andy) Gan