├── .gitignore ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── CONTRIBUTORS.md ├── Jenkinsfile ├── LICENSE ├── NOTICE ├── README.md ├── conda └── README.md ├── docker └── README.md ├── docs ├── .gitignore ├── Makefile ├── images │ ├── dgl_ke_arch.png │ ├── dist_train.png │ ├── distmult.png │ ├── kg_example.png │ ├── kge_scores.png │ ├── metis.png │ ├── multi-core.png │ ├── multi-gpu.png │ ├── multi-gpu.svg │ ├── rescal.png │ ├── rescal2.png │ ├── rescal3.png │ ├── rotate.png │ ├── transe.png │ ├── transr.png │ ├── vs-gv-fb15k.png │ └── vs-pbg-fb.png ├── make.bat └── source │ ├── benchmarks.rst │ ├── commands.rst │ ├── conf.py │ ├── dist_train.rst │ ├── emb_sim.rst │ ├── eval.rst │ ├── format_kg.rst │ ├── format_out.rst │ ├── index.rst │ ├── install.rst │ ├── kg.rst │ ├── partition.rst │ ├── predict.rst │ ├── profile.rst │ ├── train.rst │ └── train_user_data.rst ├── examples ├── README.md ├── biokg │ └── multi_gpu.sh ├── fb15k │ ├── dist_train.sh │ ├── multi_cpu.sh │ └── multi_gpu.sh ├── freebase │ ├── dist_train.sh │ ├── multi_cpu.sh │ └── multi_gpu.sh ├── wikikg2 │ └── multi_gpu.sh ├── wn18 │ ├── dist_train.sh │ ├── head.list │ ├── infer_emb_sim.sh │ ├── infer_score.sh │ ├── multi_cpu.sh │ ├── multi_gpu.sh │ ├── raw_head.list │ ├── raw_rel.list │ ├── raw_tail.list │ ├── rel.list │ └── tail.list └── wn18_weighted │ └── README.md ├── img ├── dgl_ke_arch.PNG ├── logo.png ├── vs-gv-fb15k.png ├── vs-gv-wn18.png └── vs-pbg-fb.png ├── notebook-examples └── kge_wikimedia.ipynb ├── python ├── dglke │ ├── VERSION.txt │ ├── __init__.py │ ├── convert.py │ ├── dataloader │ │ ├── KGDataset.py │ │ ├── __init__.py │ │ └── sampler.py │ ├── dist_train.py │ ├── eval.py │ ├── infer_emb_sim.py │ ├── infer_score.py │ ├── kvclient.py │ ├── kvserver.py │ ├── models │ │ ├── __init__.py │ │ ├── base_loss.py │ │ ├── general_models.py │ │ ├── infer.py │ │ ├── ke_model.py │ │ ├── mxnet │ │ │ ├── __init__.py │ │ │ ├── loss.py │ │ │ ├── score_fun.py │ │ │ └── tensor_models.py │ │ └── pytorch │ │ │ ├── __init__.py │ │ │ ├── ke_tensor.py │ │ │ ├── loss.py │ │ │ ├── score_fun.py │ │ │ └── tensor_models.py │ ├── partition.py │ ├── tests │ │ ├── test_dataset.py │ │ ├── test_infer.py │ │ ├── test_score.py │ │ └── test_topk.py │ ├── train.py │ ├── train_mxnet.py │ ├── train_pytorch.py │ └── utils.py ├── setup.cfg └── setup.py └── tests ├── fake_data ├── raw_udd │ ├── test.tsv │ ├── train.tsv │ └── valid.tsv ├── raw_udd_1 │ ├── test.tsv │ ├── train.tsv │ └── valid.tsv ├── raw_udd_2 │ ├── test.tsv │ ├── train.tsv │ └── valid.tsv ├── udd │ ├── entity.dict │ ├── relation.dict │ ├── test.tsv │ ├── train.tsv │ └── valid.tsv └── udd_1 │ ├── entity.dict │ ├── relation.dict │ ├── test.tsv │ ├── train.tsv │ └── valid.tsv ├── lint └── pylintrc └── scripts ├── task_kg_test.sh ├── task_lint.sh └── test_kg_predict.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/.gitignore -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /CONTRIBUTORS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/CONTRIBUTORS.md -------------------------------------------------------------------------------- /Jenkinsfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/Jenkinsfile -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/LICENSE -------------------------------------------------------------------------------- /NOTICE: -------------------------------------------------------------------------------- 1 | Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. 2 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/README.md -------------------------------------------------------------------------------- /conda/README.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /docker/README.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /docs/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/.gitignore -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/images/dgl_ke_arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/images/dgl_ke_arch.png -------------------------------------------------------------------------------- /docs/images/dist_train.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/images/dist_train.png -------------------------------------------------------------------------------- /docs/images/distmult.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/images/distmult.png -------------------------------------------------------------------------------- /docs/images/kg_example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/images/kg_example.png -------------------------------------------------------------------------------- /docs/images/kge_scores.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/images/kge_scores.png -------------------------------------------------------------------------------- /docs/images/metis.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/images/metis.png -------------------------------------------------------------------------------- /docs/images/multi-core.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/images/multi-core.png -------------------------------------------------------------------------------- /docs/images/multi-gpu.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/images/multi-gpu.png -------------------------------------------------------------------------------- /docs/images/multi-gpu.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/images/multi-gpu.svg -------------------------------------------------------------------------------- /docs/images/rescal.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/images/rescal.png -------------------------------------------------------------------------------- /docs/images/rescal2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/images/rescal2.png -------------------------------------------------------------------------------- /docs/images/rescal3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/images/rescal3.png -------------------------------------------------------------------------------- /docs/images/rotate.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/images/rotate.png -------------------------------------------------------------------------------- /docs/images/transe.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/images/transe.png -------------------------------------------------------------------------------- /docs/images/transr.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/images/transr.png -------------------------------------------------------------------------------- /docs/images/vs-gv-fb15k.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/images/vs-gv-fb15k.png -------------------------------------------------------------------------------- /docs/images/vs-pbg-fb.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/images/vs-pbg-fb.png -------------------------------------------------------------------------------- /docs/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/make.bat -------------------------------------------------------------------------------- /docs/source/benchmarks.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/source/benchmarks.rst -------------------------------------------------------------------------------- /docs/source/commands.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/source/commands.rst -------------------------------------------------------------------------------- /docs/source/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/source/conf.py -------------------------------------------------------------------------------- /docs/source/dist_train.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/source/dist_train.rst -------------------------------------------------------------------------------- /docs/source/emb_sim.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/source/emb_sim.rst -------------------------------------------------------------------------------- /docs/source/eval.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/source/eval.rst -------------------------------------------------------------------------------- /docs/source/format_kg.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/source/format_kg.rst -------------------------------------------------------------------------------- /docs/source/format_out.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/source/format_out.rst -------------------------------------------------------------------------------- /docs/source/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/source/index.rst -------------------------------------------------------------------------------- /docs/source/install.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/source/install.rst -------------------------------------------------------------------------------- /docs/source/kg.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/source/kg.rst -------------------------------------------------------------------------------- /docs/source/partition.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/source/partition.rst -------------------------------------------------------------------------------- /docs/source/predict.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/source/predict.rst -------------------------------------------------------------------------------- /docs/source/profile.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/source/profile.rst -------------------------------------------------------------------------------- /docs/source/train.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/source/train.rst -------------------------------------------------------------------------------- /docs/source/train_user_data.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/docs/source/train_user_data.rst -------------------------------------------------------------------------------- /examples/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/examples/README.md -------------------------------------------------------------------------------- /examples/biokg/multi_gpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/examples/biokg/multi_gpu.sh -------------------------------------------------------------------------------- /examples/fb15k/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/examples/fb15k/dist_train.sh -------------------------------------------------------------------------------- /examples/fb15k/multi_cpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/examples/fb15k/multi_cpu.sh -------------------------------------------------------------------------------- /examples/fb15k/multi_gpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/examples/fb15k/multi_gpu.sh -------------------------------------------------------------------------------- /examples/freebase/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/examples/freebase/dist_train.sh -------------------------------------------------------------------------------- /examples/freebase/multi_cpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/examples/freebase/multi_cpu.sh -------------------------------------------------------------------------------- /examples/freebase/multi_gpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/examples/freebase/multi_gpu.sh -------------------------------------------------------------------------------- /examples/wikikg2/multi_gpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/examples/wikikg2/multi_gpu.sh -------------------------------------------------------------------------------- /examples/wn18/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/examples/wn18/dist_train.sh -------------------------------------------------------------------------------- /examples/wn18/head.list: -------------------------------------------------------------------------------- 1 | 0 2 | 1 3 | 2 4 | 3 5 | 4 6 | 5 7 | 6 8 | 7 9 | 8 10 | 9 11 | -------------------------------------------------------------------------------- /examples/wn18/infer_emb_sim.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/examples/wn18/infer_emb_sim.sh -------------------------------------------------------------------------------- /examples/wn18/infer_score.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/examples/wn18/infer_score.sh -------------------------------------------------------------------------------- /examples/wn18/multi_cpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/examples/wn18/multi_cpu.sh -------------------------------------------------------------------------------- /examples/wn18/multi_gpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/examples/wn18/multi_gpu.sh -------------------------------------------------------------------------------- /examples/wn18/raw_head.list: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/examples/wn18/raw_head.list -------------------------------------------------------------------------------- /examples/wn18/raw_rel.list: -------------------------------------------------------------------------------- 1 | _derivationally_related_form 2 | _hyponym 3 | -------------------------------------------------------------------------------- /examples/wn18/raw_tail.list: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/examples/wn18/raw_tail.list -------------------------------------------------------------------------------- /examples/wn18/rel.list: -------------------------------------------------------------------------------- 1 | 0 2 | -------------------------------------------------------------------------------- /examples/wn18/tail.list: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/examples/wn18/tail.list -------------------------------------------------------------------------------- /examples/wn18_weighted/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/examples/wn18_weighted/README.md -------------------------------------------------------------------------------- /img/dgl_ke_arch.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/img/dgl_ke_arch.PNG -------------------------------------------------------------------------------- /img/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/img/logo.png -------------------------------------------------------------------------------- /img/vs-gv-fb15k.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/img/vs-gv-fb15k.png -------------------------------------------------------------------------------- /img/vs-gv-wn18.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/img/vs-gv-wn18.png -------------------------------------------------------------------------------- /img/vs-pbg-fb.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/img/vs-pbg-fb.png -------------------------------------------------------------------------------- /notebook-examples/kge_wikimedia.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/notebook-examples/kge_wikimedia.ipynb -------------------------------------------------------------------------------- /python/dglke/VERSION.txt: -------------------------------------------------------------------------------- 1 | 0.1.0.dev 2 | -------------------------------------------------------------------------------- /python/dglke/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/__init__.py -------------------------------------------------------------------------------- /python/dglke/convert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/convert.py -------------------------------------------------------------------------------- /python/dglke/dataloader/KGDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/dataloader/KGDataset.py -------------------------------------------------------------------------------- /python/dglke/dataloader/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/dataloader/__init__.py -------------------------------------------------------------------------------- /python/dglke/dataloader/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/dataloader/sampler.py -------------------------------------------------------------------------------- /python/dglke/dist_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/dist_train.py -------------------------------------------------------------------------------- /python/dglke/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/eval.py -------------------------------------------------------------------------------- /python/dglke/infer_emb_sim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/infer_emb_sim.py -------------------------------------------------------------------------------- /python/dglke/infer_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/infer_score.py -------------------------------------------------------------------------------- /python/dglke/kvclient.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/kvclient.py -------------------------------------------------------------------------------- /python/dglke/kvserver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/kvserver.py -------------------------------------------------------------------------------- /python/dglke/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/models/__init__.py -------------------------------------------------------------------------------- /python/dglke/models/base_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/models/base_loss.py -------------------------------------------------------------------------------- /python/dglke/models/general_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/models/general_models.py -------------------------------------------------------------------------------- /python/dglke/models/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/models/infer.py -------------------------------------------------------------------------------- /python/dglke/models/ke_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/models/ke_model.py -------------------------------------------------------------------------------- /python/dglke/models/mxnet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/models/mxnet/__init__.py -------------------------------------------------------------------------------- /python/dglke/models/mxnet/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/models/mxnet/loss.py -------------------------------------------------------------------------------- /python/dglke/models/mxnet/score_fun.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/models/mxnet/score_fun.py -------------------------------------------------------------------------------- /python/dglke/models/mxnet/tensor_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/models/mxnet/tensor_models.py -------------------------------------------------------------------------------- /python/dglke/models/pytorch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/models/pytorch/__init__.py -------------------------------------------------------------------------------- /python/dglke/models/pytorch/ke_tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/models/pytorch/ke_tensor.py -------------------------------------------------------------------------------- /python/dglke/models/pytorch/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/models/pytorch/loss.py -------------------------------------------------------------------------------- /python/dglke/models/pytorch/score_fun.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/models/pytorch/score_fun.py -------------------------------------------------------------------------------- /python/dglke/models/pytorch/tensor_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/models/pytorch/tensor_models.py -------------------------------------------------------------------------------- /python/dglke/partition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/partition.py -------------------------------------------------------------------------------- /python/dglke/tests/test_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/tests/test_dataset.py -------------------------------------------------------------------------------- /python/dglke/tests/test_infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/tests/test_infer.py -------------------------------------------------------------------------------- /python/dglke/tests/test_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/tests/test_score.py -------------------------------------------------------------------------------- /python/dglke/tests/test_topk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/tests/test_topk.py -------------------------------------------------------------------------------- /python/dglke/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/train.py -------------------------------------------------------------------------------- /python/dglke/train_mxnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/train_mxnet.py -------------------------------------------------------------------------------- /python/dglke/train_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/train_pytorch.py -------------------------------------------------------------------------------- /python/dglke/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/dglke/utils.py -------------------------------------------------------------------------------- /python/setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/setup.cfg -------------------------------------------------------------------------------- /python/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/python/setup.py -------------------------------------------------------------------------------- /tests/fake_data/raw_udd/test.tsv: -------------------------------------------------------------------------------- 1 | 0|1|9 2 | 3|1|2 3 | 4|0|1 4 | -------------------------------------------------------------------------------- /tests/fake_data/raw_udd/train.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/tests/fake_data/raw_udd/train.tsv -------------------------------------------------------------------------------- /tests/fake_data/raw_udd/valid.tsv: -------------------------------------------------------------------------------- 1 | 3|1|4 2 | 7|0|2 3 | 2|0|1 4 | -------------------------------------------------------------------------------- /tests/fake_data/raw_udd_1/test.tsv: -------------------------------------------------------------------------------- 1 | 0,1,9 2 | 3,1,2 3 | 4,0,1 4 | -------------------------------------------------------------------------------- /tests/fake_data/raw_udd_1/train.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/tests/fake_data/raw_udd_1/train.tsv -------------------------------------------------------------------------------- /tests/fake_data/raw_udd_1/valid.tsv: -------------------------------------------------------------------------------- 1 | 3,1,4 2 | 7,0,2 3 | 2,0,1 4 | -------------------------------------------------------------------------------- /tests/fake_data/raw_udd_2/test.tsv: -------------------------------------------------------------------------------- 1 | 0;1;9 2 | 3;1;2 3 | 4;0;1 4 | -------------------------------------------------------------------------------- /tests/fake_data/raw_udd_2/train.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/tests/fake_data/raw_udd_2/train.tsv -------------------------------------------------------------------------------- /tests/fake_data/raw_udd_2/valid.tsv: -------------------------------------------------------------------------------- 1 | 3;1;4 2 | 7;0;2 3 | 2;0;1 4 | -------------------------------------------------------------------------------- /tests/fake_data/udd/entity.dict: -------------------------------------------------------------------------------- 1 | a 0 2 | b 1 3 | c 2 4 | d 3 5 | e 5 6 | f 7 7 | g 6 8 | h 9 9 | i 4 10 | j 8 11 | -------------------------------------------------------------------------------- /tests/fake_data/udd/relation.dict: -------------------------------------------------------------------------------- 1 | A 0 2 | B 1 3 | -------------------------------------------------------------------------------- /tests/fake_data/udd/test.tsv: -------------------------------------------------------------------------------- 1 | 0 1 9 2 | 3 1 2 3 | 4 0 1 4 | -------------------------------------------------------------------------------- /tests/fake_data/udd/train.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/tests/fake_data/udd/train.tsv -------------------------------------------------------------------------------- /tests/fake_data/udd/valid.tsv: -------------------------------------------------------------------------------- 1 | 3 1 4 2 | 7 0 2 3 | 2 0 1 4 | -------------------------------------------------------------------------------- /tests/fake_data/udd_1/entity.dict: -------------------------------------------------------------------------------- 1 | a|0 2 | b|1 3 | c|2 4 | d|3 5 | e|5 6 | f|7 7 | g|6 8 | h|9 9 | i|4 10 | j|8 11 | -------------------------------------------------------------------------------- /tests/fake_data/udd_1/relation.dict: -------------------------------------------------------------------------------- 1 | A|0 2 | B|1 3 | -------------------------------------------------------------------------------- /tests/fake_data/udd_1/test.tsv: -------------------------------------------------------------------------------- 1 | 0|1|9 2 | 3|1|2 3 | 4|0|1 4 | -------------------------------------------------------------------------------- /tests/fake_data/udd_1/train.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/tests/fake_data/udd_1/train.tsv -------------------------------------------------------------------------------- /tests/fake_data/udd_1/valid.tsv: -------------------------------------------------------------------------------- 1 | 3|1|4 2 | 7|0|2 3 | 2|0|1 4 | -------------------------------------------------------------------------------- /tests/lint/pylintrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/tests/lint/pylintrc -------------------------------------------------------------------------------- /tests/scripts/task_kg_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/tests/scripts/task_kg_test.sh -------------------------------------------------------------------------------- /tests/scripts/task_lint.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/tests/scripts/task_lint.sh -------------------------------------------------------------------------------- /tests/scripts/test_kg_predict.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/dgl-ke/HEAD/tests/scripts/test_kg_predict.sh --------------------------------------------------------------------------------