文件历史

14 次代码提交

作者 SHA1 备注 提交日期
Da Zheng e317f715e0 deprecate DGL-KE. (#1432)
* deprecate DGL-KE.

* Update README.md

* remove from CI.

* remove tests.

Co-authored-by: Zheng <dzzhen@3c22fb0ed81b.ant.amazon.com>
2020-04-09 01:08:33 -07:00
xiang song(charlie.song) 1b9bc16b1a [KG]Update README and config (#1375)
* Update README and config

* upd

Co-authored-by: Ubuntu <ubuntu@ip-172-31-60-78.ec2.internal>
2020-03-19 17:06:02 +08:00
xiang song(charlie.song) f9ba3cdc81 [KGE] Kg loader (#1300)
* Now dataset accepts user define datasets

* UPdate README

* Fix eval

* Fix

* Fix Freebase

* Fix

* Fix

* upd

* upd

* Update README

* Update some docstrings.

* upd

Co-authored-by: Ubuntu <ubuntu@ip-172-31-8-26.us-east-2.compute.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-19-78.us-east-2.compute.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-60-78.ec2.internal>
2020-03-08 23:57:40 -07:00
Jinjing Zhou c23a61bd5c fix s3 link (#1310) 2020-03-04 21:01:29 +08:00
xiang song(charlie.song) dca0e376e0 [KG][Score Func] Update TransE with L2 distance support. (#1059)
* Add L2 distance score for TransE

* Update README.md

* Use linalg.gemm to speedup mx l2 dist

* Fix
2019-12-01 14:03:36 -08:00
MilkshakeForReal 8b17a5c1d5 [Model] add RotatE to dgl-kg (#964)
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
2019-11-14 10:39:05 +08:00
xiang song(charlie.song) 7f65199a86 [NN]Supporting TransR in app/kg score_func (#945)
* 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
2019-11-01 10:50:44 -07:00
Da Zheng 0bf3b6ddb3 add KG statistics. (#931)
* add KG statistics.

* add Freebase.

* fix link.
2019-10-21 22:49:46 +08:00
Chao Ma 02fb05817b [KG] Add RESCAL model to DGL-KGE (#923)
* Add RESCAL model

* update

* update

* match acc

* update

* add README.md

* fix
2019-10-11 23:45:25 -07:00
Da Zheng c26b1bae89 [KG] a little update on readme. (#914)
* update readme.

* Update README.md
2019-10-09 13:44:06 -07:00
Minjie Wang 5eae8e0543 [Fix] Fix typo in KGE readme 2019-10-08 12:14:47 -04:00
Da Zheng 3a0bbb3edd [Doc] Update kg readme (#901)
* add Freebase results.

* update.

* explain

* add DistMult.

* reproduce.
2019-10-04 15:05:06 -07:00
Da Zheng 190cdbd25f [Doc] Fix KG readme (#897)
* fix.

* update readme.

* shorten format desc.
2019-10-02 17:49:09 -07:00
Da Zheng 15b951d4c4 [KG][Model] Knowledge graph embeddings (#888)
* 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
2019-10-02 12:57:29 -07:00