- 04 Jan, 2020 1 commit
-
-
Da Zheng authored
* remove parallel sampling for multiprocessing. * avoid memory copy in eval. * remove print.
-
- 03 Jan, 2020 1 commit
-
-
Da Zheng authored
* add no_eval_filter * fix eval.
-
- 02 Jan, 2020 1 commit
-
-
Da Zheng authored
-
- 31 Dec, 2019 1 commit
-
-
Da Zheng authored
* accel edge sampling. * measure sampling time. * attach positive. * add neg_deg_sample. * fix a minor bug. * remove unnecessary code. * Revert "add neg_deg_sample." This reverts commit 45f3c08c87e33589563ca7e6a3ec91cb1239910e. * Revert "attach positive." This reverts commit 503d7821555b69b7ccfed55ce8b87ee24fff852d. Co-authored-by:xiang song(charlie.song) <classicxsong@gmail.com>
-
- 29 Dec, 2019 1 commit
-
-
Da Zheng authored
* reduce memory consumption. * fix a bug. * fix a bug. * fix.
-
- 24 Dec, 2019 1 commit
-
-
xiang song(charlie.song) authored
* 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>
-
- 20 Dec, 2019 1 commit
-
-
VoVAllen authored
* 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
-
- 13 Dec, 2019 1 commit
-
-
Da Zheng authored
-
- 01 Dec, 2019 2 commits
-
-
Da Zheng authored
-
xiang song(charlie.song) authored
* Add L2 distance score for TransE * Update README.md * Use linalg.gemm to speedup mx l2 dist * Fix
-
- 14 Nov, 2019 1 commit
-
-
MilkshakeForReal authored
Add RotatE support for KGE (apps/kg) Performance Result: Dataset FB15k: Result from Paper: MR: 40 MRR: 0.797 HIT@1: 74.6 HIT@3: 83.0 HIT@10: 88.4 Our Impl: MR: 39.6 MRR: 0.725 HIT@1: 62.8 HIT@3: 80.2 HIT@10: 87.5
-
- 01 Nov, 2019 1 commit
-
-
xiang song(charlie.song) authored
* Add TransR for kge * Now Pytorch TransR can run * Add MXNet TransR * Now mxnet can work with small dim size * Add test * Pass simple test_score * Update test with transR score func * Update RESCAL MXNet * Add missing funcs * Update init func for transR score * Revert "Update init func for transR score" This reverts commit 0798bb886095e7581f6675da5343376844ce45b9. * Update score func of TransR MXNet Make it more memory friendly and faster, thourgh it is still very slow and memory consuming * Update best config * Fix ramdom seed for test * Init score-func specific var * Update Readme
-
- 21 Oct, 2019 1 commit
-
-
Da Zheng authored
* add KG statistics. * add Freebase. * fix link.
-
- 12 Oct, 2019 1 commit
-
-
Chao Ma authored
* Add RESCAL model * update * update * match acc * update * add README.md * fix
-
- 11 Oct, 2019 3 commits
-
-
xiang song(charlie.song) authored
It cannot work when only mxnet backend is installed.
-
xiang song(charlie.song) authored
* 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 -
xiang song(charlie.song) authored
* 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 * Add complEx for mxnet * ComplEx is ready for MXNet
-
- 09 Oct, 2019 1 commit
-
-
Da Zheng authored
* update readme. * Update README.md
-
- 08 Oct, 2019 1 commit
-
-
Minjie Wang authored
-
- 04 Oct, 2019 2 commits
- 03 Oct, 2019 2 commits
- 02 Oct, 2019 1 commit
-
-
Da Zheng authored
* 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
-