├── .circleci └── config.yml ├── .coveragerc ├── .gitignore ├── .pylintrc ├── .travis.yml ├── CONTRIBUTING.md ├── ISSUE_TEMPLATE.md ├── LICENSE ├── README.md ├── docs ├── Makefile ├── make.bat └── source │ ├── api.rst │ ├── authors.rst │ ├── conf.py │ ├── contribute.rst │ ├── index.rst │ ├── intro.rst │ ├── kge.rst │ └── start.rst ├── examples ├── README.txt ├── __init__.py ├── custom_hp.yaml ├── custom_ss.yaml ├── experiment.py ├── inference.py ├── kgpipeline.py ├── pretrained │ └── TransE │ │ ├── config.npy │ │ └── model.vec.pt ├── test_file.npy ├── train.py └── tune_model.py ├── figures ├── Freebase15k_testing_hits_plot_1-1.png ├── Freebase15k_testing_latex_table_1.txt ├── Freebase15k_testing_rank_plot_1-1.png ├── Freebase15k_training_loss_plot_0-1.png ├── TransE_ent_n_rel_plot_embedding_plot_0.png ├── TransE_entity_plot_embedding_plot_0.png ├── TransE_rel_plot_embedding_plot_0.png ├── TransE_summary_0.txt └── pykg2vec_structure.png ├── pykg2vec ├── __init__.py ├── _version.py ├── common.py ├── config.py ├── data │ ├── __init__.py │ ├── datasets.py │ ├── generator.py │ └── kgcontroller.py ├── hyperparams │ ├── ANALOGY.yaml │ ├── AcrE.yaml │ ├── CP.yaml │ ├── Complex.yaml │ ├── ComplexN3.yaml │ ├── ConvE.yaml │ ├── ConvKB.yaml │ ├── DistMult.yaml │ ├── HoLE.yaml │ ├── HypER.yaml │ ├── InteractE.yaml │ ├── KG2E.yaml │ ├── MuRP.yaml │ ├── NTN.yaml │ ├── OctonionE.yaml │ ├── ProjE_pointwise.yaml │ ├── QuatE.yaml │ ├── Rescal.yaml │ ├── RotatE.yaml │ ├── SLM.yaml │ ├── SME.yaml │ ├── SME_BL.yaml │ ├── SimplE.yaml │ ├── SimplE_ignr.yaml │ ├── TransD.yaml │ ├── TransE.yaml │ ├── TransH.yaml │ ├── TransM.yaml │ ├── TransR.yaml │ └── TuckER.yaml ├── models │ ├── Domain.py │ ├── KGMeta.py │ ├── __init__.py │ ├── pairwise.py │ ├── pointwise.py │ └── projection.py ├── searchspaces │ ├── ANALOGY.yaml │ ├── AcrE.yaml │ ├── CP.yaml │ ├── Complex.yaml │ ├── ComplexN3.yaml │ ├── ConvE.yaml │ ├── ConvKB.yaml │ ├── DistMult.yaml │ ├── HoLE.yaml │ ├── HypER.yaml │ ├── InteractE.yaml │ ├── KG2E.yaml │ ├── MuRP.yaml │ ├── NTN.yaml │ ├── OctonionE.yaml │ ├── ProjE_pointwise.yaml │ ├── QuatE.yaml │ ├── Rescal.yaml │ ├── RotatE.yaml │ ├── SLM.yaml │ ├── SME.yaml │ ├── SME_BL.yaml │ ├── SimplE.yaml │ ├── SimplE_ignr.yaml │ ├── TransD.yaml │ ├── TransE.yaml │ ├── TransH.yaml │ ├── TransM.yaml │ ├── TransR.yaml │ └── TuckER.yaml ├── test │ ├── __init__.py │ ├── resource │ │ ├── custom_dataset │ │ │ ├── userdefineddataset-test.txt │ │ │ ├── userdefineddataset-train.txt │ │ │ └── userdefineddataset-valid.txt │ │ ├── custom_hyperparams │ │ │ ├── custom.txt │ │ │ ├── custom_hpf.yaml │ │ │ ├── custom_ssf.yaml │ │ │ └── dir │ │ │ │ └── custom.yaml │ │ ├── dataset.tar.gz │ │ ├── dataset.tgz │ │ └── dataset.zip │ ├── test_generator.py │ ├── test_hp_loader.py │ ├── test_inference.py │ ├── test_kg.py │ ├── test_logger.py │ ├── test_model.py │ ├── test_trainer.py │ ├── test_tune_model.py │ └── test_visualization.py └── utils │ ├── __init__.py │ ├── bayesian_optimizer.py │ ├── criterion.py │ ├── evaluator.py │ ├── logger.py │ ├── riemannian_optimizer.py │ ├── trainer.py │ └── visualization.py ├── requirements-dev.txt ├── requirements.txt ├── scripts ├── __init__.py ├── pykg2vec_infer.py ├── pykg2vec_test.py ├── pykg2vec_train.py └── pykg2vec_tune.py └── setup.py /.circleci/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/.circleci/config.yml -------------------------------------------------------------------------------- /.coveragerc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/.coveragerc -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/.gitignore -------------------------------------------------------------------------------- /.pylintrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/.pylintrc -------------------------------------------------------------------------------- /.travis.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/.travis.yml -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /ISSUE_TEMPLATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/ISSUE_TEMPLATE.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/README.md -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/docs/make.bat -------------------------------------------------------------------------------- /docs/source/api.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/docs/source/api.rst -------------------------------------------------------------------------------- /docs/source/authors.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/docs/source/authors.rst -------------------------------------------------------------------------------- /docs/source/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/docs/source/conf.py -------------------------------------------------------------------------------- /docs/source/contribute.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/docs/source/contribute.rst -------------------------------------------------------------------------------- /docs/source/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/docs/source/index.rst -------------------------------------------------------------------------------- /docs/source/intro.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/docs/source/intro.rst -------------------------------------------------------------------------------- /docs/source/kge.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/docs/source/kge.rst -------------------------------------------------------------------------------- /docs/source/start.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/docs/source/start.rst -------------------------------------------------------------------------------- /examples/README.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/examples/README.txt -------------------------------------------------------------------------------- /examples/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /examples/custom_hp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/examples/custom_hp.yaml -------------------------------------------------------------------------------- /examples/custom_ss.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/examples/custom_ss.yaml -------------------------------------------------------------------------------- /examples/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/examples/experiment.py -------------------------------------------------------------------------------- /examples/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/examples/inference.py -------------------------------------------------------------------------------- /examples/kgpipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/examples/kgpipeline.py -------------------------------------------------------------------------------- /examples/pretrained/TransE/config.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/examples/pretrained/TransE/config.npy -------------------------------------------------------------------------------- /examples/pretrained/TransE/model.vec.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/examples/pretrained/TransE/model.vec.pt -------------------------------------------------------------------------------- /examples/test_file.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/examples/test_file.npy -------------------------------------------------------------------------------- /examples/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/examples/train.py -------------------------------------------------------------------------------- /examples/tune_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/examples/tune_model.py -------------------------------------------------------------------------------- /figures/Freebase15k_testing_hits_plot_1-1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/figures/Freebase15k_testing_hits_plot_1-1.png -------------------------------------------------------------------------------- /figures/Freebase15k_testing_latex_table_1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/figures/Freebase15k_testing_latex_table_1.txt -------------------------------------------------------------------------------- /figures/Freebase15k_testing_rank_plot_1-1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/figures/Freebase15k_testing_rank_plot_1-1.png -------------------------------------------------------------------------------- /figures/Freebase15k_training_loss_plot_0-1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/figures/Freebase15k_training_loss_plot_0-1.png -------------------------------------------------------------------------------- /figures/TransE_ent_n_rel_plot_embedding_plot_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/figures/TransE_ent_n_rel_plot_embedding_plot_0.png -------------------------------------------------------------------------------- /figures/TransE_entity_plot_embedding_plot_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/figures/TransE_entity_plot_embedding_plot_0.png -------------------------------------------------------------------------------- /figures/TransE_rel_plot_embedding_plot_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/figures/TransE_rel_plot_embedding_plot_0.png -------------------------------------------------------------------------------- /figures/TransE_summary_0.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/figures/TransE_summary_0.txt -------------------------------------------------------------------------------- /figures/pykg2vec_structure.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/figures/pykg2vec_structure.png -------------------------------------------------------------------------------- /pykg2vec/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/__init__.py -------------------------------------------------------------------------------- /pykg2vec/_version.py: -------------------------------------------------------------------------------- 1 | __version__ = "0.0.52" 2 | -------------------------------------------------------------------------------- /pykg2vec/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/common.py -------------------------------------------------------------------------------- /pykg2vec/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/config.py -------------------------------------------------------------------------------- /pykg2vec/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pykg2vec/data/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/data/datasets.py -------------------------------------------------------------------------------- /pykg2vec/data/generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/data/generator.py -------------------------------------------------------------------------------- /pykg2vec/data/kgcontroller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/data/kgcontroller.py -------------------------------------------------------------------------------- /pykg2vec/hyperparams/ANALOGY.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/ANALOGY.yaml -------------------------------------------------------------------------------- /pykg2vec/hyperparams/AcrE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/AcrE.yaml -------------------------------------------------------------------------------- /pykg2vec/hyperparams/CP.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/CP.yaml -------------------------------------------------------------------------------- /pykg2vec/hyperparams/Complex.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/Complex.yaml -------------------------------------------------------------------------------- /pykg2vec/hyperparams/ComplexN3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/ComplexN3.yaml -------------------------------------------------------------------------------- /pykg2vec/hyperparams/ConvE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/ConvE.yaml -------------------------------------------------------------------------------- /pykg2vec/hyperparams/ConvKB.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/ConvKB.yaml -------------------------------------------------------------------------------- /pykg2vec/hyperparams/DistMult.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/DistMult.yaml -------------------------------------------------------------------------------- /pykg2vec/hyperparams/HoLE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/HoLE.yaml -------------------------------------------------------------------------------- /pykg2vec/hyperparams/HypER.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/HypER.yaml -------------------------------------------------------------------------------- /pykg2vec/hyperparams/InteractE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/InteractE.yaml -------------------------------------------------------------------------------- /pykg2vec/hyperparams/KG2E.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/KG2E.yaml -------------------------------------------------------------------------------- /pykg2vec/hyperparams/MuRP.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/MuRP.yaml -------------------------------------------------------------------------------- /pykg2vec/hyperparams/NTN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/NTN.yaml -------------------------------------------------------------------------------- /pykg2vec/hyperparams/OctonionE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/OctonionE.yaml -------------------------------------------------------------------------------- /pykg2vec/hyperparams/ProjE_pointwise.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/ProjE_pointwise.yaml -------------------------------------------------------------------------------- /pykg2vec/hyperparams/QuatE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/QuatE.yaml -------------------------------------------------------------------------------- /pykg2vec/hyperparams/Rescal.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/Rescal.yaml -------------------------------------------------------------------------------- /pykg2vec/hyperparams/RotatE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/RotatE.yaml -------------------------------------------------------------------------------- /pykg2vec/hyperparams/SLM.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/SLM.yaml -------------------------------------------------------------------------------- /pykg2vec/hyperparams/SME.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/SME.yaml -------------------------------------------------------------------------------- /pykg2vec/hyperparams/SME_BL.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/SME_BL.yaml -------------------------------------------------------------------------------- /pykg2vec/hyperparams/SimplE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/SimplE.yaml -------------------------------------------------------------------------------- /pykg2vec/hyperparams/SimplE_ignr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/SimplE_ignr.yaml -------------------------------------------------------------------------------- /pykg2vec/hyperparams/TransD.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/TransD.yaml -------------------------------------------------------------------------------- /pykg2vec/hyperparams/TransE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/TransE.yaml -------------------------------------------------------------------------------- /pykg2vec/hyperparams/TransH.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/TransH.yaml -------------------------------------------------------------------------------- /pykg2vec/hyperparams/TransM.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/TransM.yaml -------------------------------------------------------------------------------- /pykg2vec/hyperparams/TransR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/TransR.yaml -------------------------------------------------------------------------------- /pykg2vec/hyperparams/TuckER.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/hyperparams/TuckER.yaml -------------------------------------------------------------------------------- /pykg2vec/models/Domain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/models/Domain.py -------------------------------------------------------------------------------- /pykg2vec/models/KGMeta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/models/KGMeta.py -------------------------------------------------------------------------------- /pykg2vec/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pykg2vec/models/pairwise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/models/pairwise.py -------------------------------------------------------------------------------- /pykg2vec/models/pointwise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/models/pointwise.py -------------------------------------------------------------------------------- /pykg2vec/models/projection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/models/projection.py -------------------------------------------------------------------------------- /pykg2vec/searchspaces/ANALOGY.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/ANALOGY.yaml -------------------------------------------------------------------------------- /pykg2vec/searchspaces/AcrE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/AcrE.yaml -------------------------------------------------------------------------------- /pykg2vec/searchspaces/CP.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/CP.yaml -------------------------------------------------------------------------------- /pykg2vec/searchspaces/Complex.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/Complex.yaml -------------------------------------------------------------------------------- /pykg2vec/searchspaces/ComplexN3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/ComplexN3.yaml -------------------------------------------------------------------------------- /pykg2vec/searchspaces/ConvE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/ConvE.yaml -------------------------------------------------------------------------------- /pykg2vec/searchspaces/ConvKB.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/ConvKB.yaml -------------------------------------------------------------------------------- /pykg2vec/searchspaces/DistMult.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/DistMult.yaml -------------------------------------------------------------------------------- /pykg2vec/searchspaces/HoLE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/HoLE.yaml -------------------------------------------------------------------------------- /pykg2vec/searchspaces/HypER.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/HypER.yaml -------------------------------------------------------------------------------- /pykg2vec/searchspaces/InteractE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/InteractE.yaml -------------------------------------------------------------------------------- /pykg2vec/searchspaces/KG2E.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/KG2E.yaml -------------------------------------------------------------------------------- /pykg2vec/searchspaces/MuRP.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/MuRP.yaml -------------------------------------------------------------------------------- /pykg2vec/searchspaces/NTN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/NTN.yaml -------------------------------------------------------------------------------- /pykg2vec/searchspaces/OctonionE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/OctonionE.yaml -------------------------------------------------------------------------------- /pykg2vec/searchspaces/ProjE_pointwise.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/ProjE_pointwise.yaml -------------------------------------------------------------------------------- /pykg2vec/searchspaces/QuatE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/QuatE.yaml -------------------------------------------------------------------------------- /pykg2vec/searchspaces/Rescal.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/Rescal.yaml -------------------------------------------------------------------------------- /pykg2vec/searchspaces/RotatE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/RotatE.yaml -------------------------------------------------------------------------------- /pykg2vec/searchspaces/SLM.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/SLM.yaml -------------------------------------------------------------------------------- /pykg2vec/searchspaces/SME.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/SME.yaml -------------------------------------------------------------------------------- /pykg2vec/searchspaces/SME_BL.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/SME_BL.yaml -------------------------------------------------------------------------------- /pykg2vec/searchspaces/SimplE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/SimplE.yaml -------------------------------------------------------------------------------- /pykg2vec/searchspaces/SimplE_ignr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/SimplE_ignr.yaml -------------------------------------------------------------------------------- /pykg2vec/searchspaces/TransD.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/TransD.yaml -------------------------------------------------------------------------------- /pykg2vec/searchspaces/TransE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/TransE.yaml -------------------------------------------------------------------------------- /pykg2vec/searchspaces/TransH.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/TransH.yaml -------------------------------------------------------------------------------- /pykg2vec/searchspaces/TransM.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/TransM.yaml -------------------------------------------------------------------------------- /pykg2vec/searchspaces/TransR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/TransR.yaml -------------------------------------------------------------------------------- /pykg2vec/searchspaces/TuckER.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/searchspaces/TuckER.yaml -------------------------------------------------------------------------------- /pykg2vec/test/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pykg2vec/test/resource/custom_dataset/userdefineddataset-test.txt: -------------------------------------------------------------------------------- 1 | head_1 relation_1 tail_1 -------------------------------------------------------------------------------- /pykg2vec/test/resource/custom_dataset/userdefineddataset-train.txt: -------------------------------------------------------------------------------- 1 | head_2 relation_2 tail_2 -------------------------------------------------------------------------------- /pykg2vec/test/resource/custom_dataset/userdefineddataset-valid.txt: -------------------------------------------------------------------------------- 1 | head_3 relation_3 tail_3 -------------------------------------------------------------------------------- /pykg2vec/test/resource/custom_hyperparams/custom.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/test/resource/custom_hyperparams/custom.txt -------------------------------------------------------------------------------- /pykg2vec/test/resource/custom_hyperparams/custom_hpf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/test/resource/custom_hyperparams/custom_hpf.yaml -------------------------------------------------------------------------------- /pykg2vec/test/resource/custom_hyperparams/custom_ssf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/test/resource/custom_hyperparams/custom_ssf.yaml -------------------------------------------------------------------------------- /pykg2vec/test/resource/custom_hyperparams/dir/custom.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/test/resource/custom_hyperparams/dir/custom.yaml -------------------------------------------------------------------------------- /pykg2vec/test/resource/dataset.tar.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/test/resource/dataset.tar.gz -------------------------------------------------------------------------------- /pykg2vec/test/resource/dataset.tgz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/test/resource/dataset.tgz -------------------------------------------------------------------------------- /pykg2vec/test/resource/dataset.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/test/resource/dataset.zip -------------------------------------------------------------------------------- /pykg2vec/test/test_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/test/test_generator.py -------------------------------------------------------------------------------- /pykg2vec/test/test_hp_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/test/test_hp_loader.py -------------------------------------------------------------------------------- /pykg2vec/test/test_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/test/test_inference.py -------------------------------------------------------------------------------- /pykg2vec/test/test_kg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/test/test_kg.py -------------------------------------------------------------------------------- /pykg2vec/test/test_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/test/test_logger.py -------------------------------------------------------------------------------- /pykg2vec/test/test_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/test/test_model.py -------------------------------------------------------------------------------- /pykg2vec/test/test_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/test/test_trainer.py -------------------------------------------------------------------------------- /pykg2vec/test/test_tune_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/test/test_tune_model.py -------------------------------------------------------------------------------- /pykg2vec/test/test_visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/test/test_visualization.py -------------------------------------------------------------------------------- /pykg2vec/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pykg2vec/utils/bayesian_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/utils/bayesian_optimizer.py -------------------------------------------------------------------------------- /pykg2vec/utils/criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/utils/criterion.py -------------------------------------------------------------------------------- /pykg2vec/utils/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/utils/evaluator.py -------------------------------------------------------------------------------- /pykg2vec/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/utils/logger.py -------------------------------------------------------------------------------- /pykg2vec/utils/riemannian_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/utils/riemannian_optimizer.py -------------------------------------------------------------------------------- /pykg2vec/utils/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/utils/trainer.py -------------------------------------------------------------------------------- /pykg2vec/utils/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/pykg2vec/utils/visualization.py -------------------------------------------------------------------------------- /requirements-dev.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/requirements-dev.txt -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /scripts/pykg2vec_infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/scripts/pykg2vec_infer.py -------------------------------------------------------------------------------- /scripts/pykg2vec_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/scripts/pykg2vec_test.py -------------------------------------------------------------------------------- /scripts/pykg2vec_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/scripts/pykg2vec_train.py -------------------------------------------------------------------------------- /scripts/pykg2vec_tune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/scripts/pykg2vec_tune.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sujit-O/pykg2vec/HEAD/setup.py --------------------------------------------------------------------------------