├── .gitignore ├── CHANGELOG.md ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── data ├── download_all.sh └── preprocess │ ├── preprocess_default.py │ ├── preprocess_wn11.py │ └── util.py ├── docs ├── Makefile ├── conf.py ├── examples │ ├── dump-example-model.csv │ ├── dump-example-search.csv │ ├── expanded-config-example.yaml │ └── train_and_valid_trace_after_one_epoch.yaml ├── index.rst ├── make.bat └── source │ ├── images │ └── logo │ │ ├── libkge-1280.png │ │ ├── libkge-160.png │ │ ├── libkge-320.png │ │ ├── libkge-640.png │ │ ├── libkge-80.png │ │ ├── libkge-header-1280.png │ │ ├── libkge-header-1920.png │ │ ├── libkge-header-2880.png │ │ ├── libkge-header-800.png │ │ ├── libkge-header.pdf │ │ ├── libkge-header.svg │ │ ├── libkge.pdf │ │ └── libkge.svg │ ├── kge.job.rst │ ├── kge.model.embedder.rst │ ├── kge.model.rst │ ├── kge.rst │ └── kge.util.rst ├── examples ├── toy-complex-search-ax.yaml ├── toy-complex-search-grash.yaml ├── toy-complex-search-grid.yaml ├── toy-complex-search-manual.yaml ├── toy-complex-train-sampling.yaml ├── toy-complex-train.yaml ├── toy-conve-train.yaml ├── toy-rt3-train.yaml └── toy-transe-train.yaml ├── kge ├── __init__.py ├── __main__.py ├── cli.py ├── config-default.yaml ├── config.py ├── dataset.py ├── indexing.py ├── job │ ├── __init__.py │ ├── eval.py │ ├── eval_entity_pair_ranking.py │ ├── eval_entity_ranking.py │ ├── eval_training_loss.py │ ├── job.py │ ├── search.py │ ├── search_auto.py │ ├── search_ax.py │ ├── search_grash.py │ ├── search_grid.py │ ├── search_manual.py │ ├── trace.py │ ├── train.py │ ├── train_1vsAll.py │ ├── train_KvsAll.py │ ├── train_negative_sampling.py │ └── util.py ├── misc.py ├── model │ ├── __init__.py │ ├── complex.py │ ├── complex.yaml │ ├── conve.py │ ├── conve.yaml │ ├── cp.py │ ├── cp.yaml │ ├── distmult.py │ ├── distmult.yaml │ ├── embedder │ │ ├── __init__.py │ │ ├── lookup_embedder.py │ │ ├── lookup_embedder.yaml │ │ ├── projection_embedder.py │ │ ├── projection_embedder.yaml │ │ ├── tucker3_relation_embedder.py │ │ └── tucker3_relation_embedder.yaml │ ├── kge_model.py │ ├── reciprocal_relations_model.py │ ├── reciprocal_relations_model.yaml │ ├── relational_tucker3.py │ ├── relational_tucker3.yaml │ ├── rescal.py │ ├── rescal.yaml │ ├── rotate.py │ ├── rotate.yaml │ ├── simple.py │ ├── simple.yaml │ ├── transe.py │ ├── transe.yaml │ ├── transformer.py │ ├── transformer.yaml │ ├── transh.py │ └── transh.yaml └── util │ ├── __init__.py │ ├── configspace_converter.py │ ├── dump.py │ ├── io.py │ ├── loss.py │ ├── metric.py │ ├── optimizer.py │ ├── package.py │ ├── sampler.py │ ├── seed.py │ └── subgraph.py ├── local └── README ├── setup.cfg ├── setup.py └── tests ├── __init__.py ├── data ├── dataset_preprocess │ ├── test.txt │ ├── train.txt │ └── valid.txt └── dataset_test │ ├── dataset.yaml │ ├── entity_ids.del │ ├── relation_ids.del │ ├── test.del │ ├── train.del │ └── valid.del ├── test_dataset.py ├── test_model.py ├── test_preprocess.py ├── test_train.py └── util.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/.gitignore -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/README.md -------------------------------------------------------------------------------- /data/download_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/data/download_all.sh -------------------------------------------------------------------------------- /data/preprocess/preprocess_default.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/data/preprocess/preprocess_default.py -------------------------------------------------------------------------------- /data/preprocess/preprocess_wn11.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/data/preprocess/preprocess_wn11.py -------------------------------------------------------------------------------- /data/preprocess/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/data/preprocess/util.py -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/examples/dump-example-model.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/docs/examples/dump-example-model.csv -------------------------------------------------------------------------------- /docs/examples/dump-example-search.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/docs/examples/dump-example-search.csv -------------------------------------------------------------------------------- /docs/examples/expanded-config-example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/docs/examples/expanded-config-example.yaml -------------------------------------------------------------------------------- /docs/examples/train_and_valid_trace_after_one_epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/docs/examples/train_and_valid_trace_after_one_epoch.yaml -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/docs/make.bat -------------------------------------------------------------------------------- /docs/source/images/logo/libkge-1280.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/docs/source/images/logo/libkge-1280.png -------------------------------------------------------------------------------- /docs/source/images/logo/libkge-160.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/docs/source/images/logo/libkge-160.png -------------------------------------------------------------------------------- /docs/source/images/logo/libkge-320.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/docs/source/images/logo/libkge-320.png -------------------------------------------------------------------------------- /docs/source/images/logo/libkge-640.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/docs/source/images/logo/libkge-640.png -------------------------------------------------------------------------------- /docs/source/images/logo/libkge-80.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/docs/source/images/logo/libkge-80.png -------------------------------------------------------------------------------- /docs/source/images/logo/libkge-header-1280.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/docs/source/images/logo/libkge-header-1280.png -------------------------------------------------------------------------------- /docs/source/images/logo/libkge-header-1920.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/docs/source/images/logo/libkge-header-1920.png -------------------------------------------------------------------------------- /docs/source/images/logo/libkge-header-2880.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/docs/source/images/logo/libkge-header-2880.png -------------------------------------------------------------------------------- /docs/source/images/logo/libkge-header-800.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/docs/source/images/logo/libkge-header-800.png -------------------------------------------------------------------------------- /docs/source/images/logo/libkge-header.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/docs/source/images/logo/libkge-header.pdf -------------------------------------------------------------------------------- /docs/source/images/logo/libkge-header.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/docs/source/images/logo/libkge-header.svg -------------------------------------------------------------------------------- /docs/source/images/logo/libkge.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/docs/source/images/logo/libkge.pdf -------------------------------------------------------------------------------- /docs/source/images/logo/libkge.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/docs/source/images/logo/libkge.svg -------------------------------------------------------------------------------- /docs/source/kge.job.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/docs/source/kge.job.rst -------------------------------------------------------------------------------- /docs/source/kge.model.embedder.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/docs/source/kge.model.embedder.rst -------------------------------------------------------------------------------- /docs/source/kge.model.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/docs/source/kge.model.rst -------------------------------------------------------------------------------- /docs/source/kge.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/docs/source/kge.rst -------------------------------------------------------------------------------- /docs/source/kge.util.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/docs/source/kge.util.rst -------------------------------------------------------------------------------- /examples/toy-complex-search-ax.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/examples/toy-complex-search-ax.yaml -------------------------------------------------------------------------------- /examples/toy-complex-search-grash.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/examples/toy-complex-search-grash.yaml -------------------------------------------------------------------------------- /examples/toy-complex-search-grid.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/examples/toy-complex-search-grid.yaml -------------------------------------------------------------------------------- /examples/toy-complex-search-manual.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/examples/toy-complex-search-manual.yaml -------------------------------------------------------------------------------- /examples/toy-complex-train-sampling.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/examples/toy-complex-train-sampling.yaml -------------------------------------------------------------------------------- /examples/toy-complex-train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/examples/toy-complex-train.yaml -------------------------------------------------------------------------------- /examples/toy-conve-train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/examples/toy-conve-train.yaml -------------------------------------------------------------------------------- /examples/toy-rt3-train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/examples/toy-rt3-train.yaml -------------------------------------------------------------------------------- /examples/toy-transe-train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/examples/toy-transe-train.yaml -------------------------------------------------------------------------------- /kge/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/__init__.py -------------------------------------------------------------------------------- /kge/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/__main__.py -------------------------------------------------------------------------------- /kge/cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/cli.py -------------------------------------------------------------------------------- /kge/config-default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/config-default.yaml -------------------------------------------------------------------------------- /kge/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/config.py -------------------------------------------------------------------------------- /kge/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/dataset.py -------------------------------------------------------------------------------- /kge/indexing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/indexing.py -------------------------------------------------------------------------------- /kge/job/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/job/__init__.py -------------------------------------------------------------------------------- /kge/job/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/job/eval.py -------------------------------------------------------------------------------- /kge/job/eval_entity_pair_ranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/job/eval_entity_pair_ranking.py -------------------------------------------------------------------------------- /kge/job/eval_entity_ranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/job/eval_entity_ranking.py -------------------------------------------------------------------------------- /kge/job/eval_training_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/job/eval_training_loss.py -------------------------------------------------------------------------------- /kge/job/job.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/job/job.py -------------------------------------------------------------------------------- /kge/job/search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/job/search.py -------------------------------------------------------------------------------- /kge/job/search_auto.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/job/search_auto.py -------------------------------------------------------------------------------- /kge/job/search_ax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/job/search_ax.py -------------------------------------------------------------------------------- /kge/job/search_grash.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/job/search_grash.py -------------------------------------------------------------------------------- /kge/job/search_grid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/job/search_grid.py -------------------------------------------------------------------------------- /kge/job/search_manual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/job/search_manual.py -------------------------------------------------------------------------------- /kge/job/trace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/job/trace.py -------------------------------------------------------------------------------- /kge/job/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/job/train.py -------------------------------------------------------------------------------- /kge/job/train_1vsAll.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/job/train_1vsAll.py -------------------------------------------------------------------------------- /kge/job/train_KvsAll.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/job/train_KvsAll.py -------------------------------------------------------------------------------- /kge/job/train_negative_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/job/train_negative_sampling.py -------------------------------------------------------------------------------- /kge/job/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/job/util.py -------------------------------------------------------------------------------- /kge/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/misc.py -------------------------------------------------------------------------------- /kge/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/__init__.py -------------------------------------------------------------------------------- /kge/model/complex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/complex.py -------------------------------------------------------------------------------- /kge/model/complex.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/complex.yaml -------------------------------------------------------------------------------- /kge/model/conve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/conve.py -------------------------------------------------------------------------------- /kge/model/conve.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/conve.yaml -------------------------------------------------------------------------------- /kge/model/cp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/cp.py -------------------------------------------------------------------------------- /kge/model/cp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/cp.yaml -------------------------------------------------------------------------------- /kge/model/distmult.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/distmult.py -------------------------------------------------------------------------------- /kge/model/distmult.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/distmult.yaml -------------------------------------------------------------------------------- /kge/model/embedder/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /kge/model/embedder/lookup_embedder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/embedder/lookup_embedder.py -------------------------------------------------------------------------------- /kge/model/embedder/lookup_embedder.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/embedder/lookup_embedder.yaml -------------------------------------------------------------------------------- /kge/model/embedder/projection_embedder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/embedder/projection_embedder.py -------------------------------------------------------------------------------- /kge/model/embedder/projection_embedder.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/embedder/projection_embedder.yaml -------------------------------------------------------------------------------- /kge/model/embedder/tucker3_relation_embedder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/embedder/tucker3_relation_embedder.py -------------------------------------------------------------------------------- /kge/model/embedder/tucker3_relation_embedder.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/embedder/tucker3_relation_embedder.yaml -------------------------------------------------------------------------------- /kge/model/kge_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/kge_model.py -------------------------------------------------------------------------------- /kge/model/reciprocal_relations_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/reciprocal_relations_model.py -------------------------------------------------------------------------------- /kge/model/reciprocal_relations_model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/reciprocal_relations_model.yaml -------------------------------------------------------------------------------- /kge/model/relational_tucker3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/relational_tucker3.py -------------------------------------------------------------------------------- /kge/model/relational_tucker3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/relational_tucker3.yaml -------------------------------------------------------------------------------- /kge/model/rescal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/rescal.py -------------------------------------------------------------------------------- /kge/model/rescal.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/rescal.yaml -------------------------------------------------------------------------------- /kge/model/rotate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/rotate.py -------------------------------------------------------------------------------- /kge/model/rotate.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/rotate.yaml -------------------------------------------------------------------------------- /kge/model/simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/simple.py -------------------------------------------------------------------------------- /kge/model/simple.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/simple.yaml -------------------------------------------------------------------------------- /kge/model/transe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/transe.py -------------------------------------------------------------------------------- /kge/model/transe.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/transe.yaml -------------------------------------------------------------------------------- /kge/model/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/transformer.py -------------------------------------------------------------------------------- /kge/model/transformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/transformer.yaml -------------------------------------------------------------------------------- /kge/model/transh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/transh.py -------------------------------------------------------------------------------- /kge/model/transh.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/model/transh.yaml -------------------------------------------------------------------------------- /kge/util/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/util/__init__.py -------------------------------------------------------------------------------- /kge/util/configspace_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/util/configspace_converter.py -------------------------------------------------------------------------------- /kge/util/dump.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/util/dump.py -------------------------------------------------------------------------------- /kge/util/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/util/io.py -------------------------------------------------------------------------------- /kge/util/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/util/loss.py -------------------------------------------------------------------------------- /kge/util/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/util/metric.py -------------------------------------------------------------------------------- /kge/util/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/util/optimizer.py -------------------------------------------------------------------------------- /kge/util/package.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/util/package.py -------------------------------------------------------------------------------- /kge/util/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/util/sampler.py -------------------------------------------------------------------------------- /kge/util/seed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/util/seed.py -------------------------------------------------------------------------------- /kge/util/subgraph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/kge/util/subgraph.py -------------------------------------------------------------------------------- /local/README: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/local/README -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/setup.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/data/dataset_preprocess/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/tests/data/dataset_preprocess/test.txt -------------------------------------------------------------------------------- /tests/data/dataset_preprocess/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/tests/data/dataset_preprocess/train.txt -------------------------------------------------------------------------------- /tests/data/dataset_preprocess/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/tests/data/dataset_preprocess/valid.txt -------------------------------------------------------------------------------- /tests/data/dataset_test/dataset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/tests/data/dataset_test/dataset.yaml -------------------------------------------------------------------------------- /tests/data/dataset_test/entity_ids.del: -------------------------------------------------------------------------------- 1 | 0 Anna 2 | 1 Bob 3 | 2 Charlie 4 | 3 Debbie 5 | -------------------------------------------------------------------------------- /tests/data/dataset_test/relation_ids.del: -------------------------------------------------------------------------------- 1 | 0 friends_with 2 | 1 loves 3 | 2 hates 4 | -------------------------------------------------------------------------------- /tests/data/dataset_test/test.del: -------------------------------------------------------------------------------- 1 | 2 2 1 2 | 2 0 3 3 | -------------------------------------------------------------------------------- /tests/data/dataset_test/train.del: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/tests/data/dataset_test/train.del -------------------------------------------------------------------------------- /tests/data/dataset_test/valid.del: -------------------------------------------------------------------------------- 1 | 1 0 0 2 | 1 1 0 3 | -------------------------------------------------------------------------------- /tests/test_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/tests/test_dataset.py -------------------------------------------------------------------------------- /tests/test_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/tests/test_model.py -------------------------------------------------------------------------------- /tests/test_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/tests/test_preprocess.py -------------------------------------------------------------------------------- /tests/test_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/tests/test_train.py -------------------------------------------------------------------------------- /tests/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uma-pi1/kge/HEAD/tests/util.py --------------------------------------------------------------------------------