├── .gitattributes ├── .gitignore ├── KGQA ├── LSTM │ ├── dataloader.py │ ├── main.py │ └── model.py └── RoBERTa │ ├── README.md │ ├── dataloader.py │ ├── main.py │ ├── model.py │ ├── pruning_dataloader.py │ ├── pruning_main.py │ ├── pruning_model.py │ ├── pruning_test.txt │ ├── pruning_train.txt │ ├── relation_matching_eval.ipynb │ ├── test_kge_embedding.ipynb │ ├── webqsp_scores_full_kg.pkl │ └── webqsp_scores_full_kg_fixed.pkl ├── LICENSE ├── README.md ├── kge ├── __init__.py ├── __main__.py ├── cli.py ├── config-default.yaml ├── config.py ├── dataset.py ├── indexing.py ├── job │ ├── __init__.py │ ├── auto_search.py │ ├── ax_search.py │ ├── entity_pair_ranking.py │ ├── entity_ranking.py │ ├── eval.py │ ├── grid_search.py │ ├── job.py │ ├── manual_search.py │ ├── search.py │ ├── trace.py │ ├── train.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 └── util │ ├── __init__.py │ ├── dump.py │ ├── io.py │ ├── loss.py │ ├── optimizer.py │ ├── package.py │ └── sampler.py ├── model.png ├── requirements.txt └── train_embeddings ├── load_data.py ├── main.py └── model.py /.gitattributes: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/.gitignore -------------------------------------------------------------------------------- /KGQA/LSTM/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/KGQA/LSTM/dataloader.py -------------------------------------------------------------------------------- /KGQA/LSTM/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/KGQA/LSTM/main.py -------------------------------------------------------------------------------- /KGQA/LSTM/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/KGQA/LSTM/model.py -------------------------------------------------------------------------------- /KGQA/RoBERTa/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/KGQA/RoBERTa/README.md -------------------------------------------------------------------------------- /KGQA/RoBERTa/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/KGQA/RoBERTa/dataloader.py -------------------------------------------------------------------------------- /KGQA/RoBERTa/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/KGQA/RoBERTa/main.py -------------------------------------------------------------------------------- /KGQA/RoBERTa/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/KGQA/RoBERTa/model.py -------------------------------------------------------------------------------- /KGQA/RoBERTa/pruning_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/KGQA/RoBERTa/pruning_dataloader.py -------------------------------------------------------------------------------- /KGQA/RoBERTa/pruning_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/KGQA/RoBERTa/pruning_main.py -------------------------------------------------------------------------------- /KGQA/RoBERTa/pruning_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/KGQA/RoBERTa/pruning_model.py -------------------------------------------------------------------------------- /KGQA/RoBERTa/pruning_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/KGQA/RoBERTa/pruning_test.txt -------------------------------------------------------------------------------- /KGQA/RoBERTa/pruning_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/KGQA/RoBERTa/pruning_train.txt -------------------------------------------------------------------------------- /KGQA/RoBERTa/relation_matching_eval.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/KGQA/RoBERTa/relation_matching_eval.ipynb -------------------------------------------------------------------------------- /KGQA/RoBERTa/test_kge_embedding.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/KGQA/RoBERTa/test_kge_embedding.ipynb -------------------------------------------------------------------------------- /KGQA/RoBERTa/webqsp_scores_full_kg.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/KGQA/RoBERTa/webqsp_scores_full_kg.pkl -------------------------------------------------------------------------------- /KGQA/RoBERTa/webqsp_scores_full_kg_fixed.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/KGQA/RoBERTa/webqsp_scores_full_kg_fixed.pkl -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/README.md -------------------------------------------------------------------------------- /kge/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/__init__.py -------------------------------------------------------------------------------- /kge/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/__main__.py -------------------------------------------------------------------------------- /kge/cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/cli.py -------------------------------------------------------------------------------- /kge/config-default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/config-default.yaml -------------------------------------------------------------------------------- /kge/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/config.py -------------------------------------------------------------------------------- /kge/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/dataset.py -------------------------------------------------------------------------------- /kge/indexing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/indexing.py -------------------------------------------------------------------------------- /kge/job/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/job/__init__.py -------------------------------------------------------------------------------- /kge/job/auto_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/job/auto_search.py -------------------------------------------------------------------------------- /kge/job/ax_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/job/ax_search.py -------------------------------------------------------------------------------- /kge/job/entity_pair_ranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/job/entity_pair_ranking.py -------------------------------------------------------------------------------- /kge/job/entity_ranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/job/entity_ranking.py -------------------------------------------------------------------------------- /kge/job/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/job/eval.py -------------------------------------------------------------------------------- /kge/job/grid_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/job/grid_search.py -------------------------------------------------------------------------------- /kge/job/job.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/job/job.py -------------------------------------------------------------------------------- /kge/job/manual_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/job/manual_search.py -------------------------------------------------------------------------------- /kge/job/search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/job/search.py -------------------------------------------------------------------------------- /kge/job/trace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/job/trace.py -------------------------------------------------------------------------------- /kge/job/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/job/train.py -------------------------------------------------------------------------------- /kge/job/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/job/util.py -------------------------------------------------------------------------------- /kge/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/misc.py -------------------------------------------------------------------------------- /kge/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/model/__init__.py -------------------------------------------------------------------------------- /kge/model/complex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/model/complex.py -------------------------------------------------------------------------------- /kge/model/complex.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/model/complex.yaml -------------------------------------------------------------------------------- /kge/model/conve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/model/conve.py -------------------------------------------------------------------------------- /kge/model/conve.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/model/conve.yaml -------------------------------------------------------------------------------- /kge/model/cp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/model/cp.py -------------------------------------------------------------------------------- /kge/model/cp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/model/cp.yaml -------------------------------------------------------------------------------- /kge/model/distmult.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/model/distmult.py -------------------------------------------------------------------------------- /kge/model/distmult.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/model/distmult.yaml -------------------------------------------------------------------------------- /kge/model/embedder/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /kge/model/embedder/lookup_embedder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/model/embedder/lookup_embedder.py -------------------------------------------------------------------------------- /kge/model/embedder/lookup_embedder.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/model/embedder/lookup_embedder.yaml -------------------------------------------------------------------------------- /kge/model/embedder/projection_embedder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/model/embedder/projection_embedder.py -------------------------------------------------------------------------------- /kge/model/embedder/projection_embedder.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/model/embedder/projection_embedder.yaml -------------------------------------------------------------------------------- /kge/model/embedder/tucker3_relation_embedder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/model/embedder/tucker3_relation_embedder.py -------------------------------------------------------------------------------- /kge/model/embedder/tucker3_relation_embedder.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/model/embedder/tucker3_relation_embedder.yaml -------------------------------------------------------------------------------- /kge/model/kge_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/model/kge_model.py -------------------------------------------------------------------------------- /kge/model/reciprocal_relations_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/model/reciprocal_relations_model.py -------------------------------------------------------------------------------- /kge/model/reciprocal_relations_model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/model/reciprocal_relations_model.yaml -------------------------------------------------------------------------------- /kge/model/relational_tucker3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/model/relational_tucker3.py -------------------------------------------------------------------------------- /kge/model/relational_tucker3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/model/relational_tucker3.yaml -------------------------------------------------------------------------------- /kge/model/rescal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/model/rescal.py -------------------------------------------------------------------------------- /kge/model/rescal.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/model/rescal.yaml -------------------------------------------------------------------------------- /kge/model/rotate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/model/rotate.py -------------------------------------------------------------------------------- /kge/model/rotate.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/model/rotate.yaml -------------------------------------------------------------------------------- /kge/model/simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/model/simple.py -------------------------------------------------------------------------------- /kge/model/simple.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/model/simple.yaml -------------------------------------------------------------------------------- /kge/model/transe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/model/transe.py -------------------------------------------------------------------------------- /kge/model/transe.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/model/transe.yaml -------------------------------------------------------------------------------- /kge/util/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/util/__init__.py -------------------------------------------------------------------------------- /kge/util/dump.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/util/dump.py -------------------------------------------------------------------------------- /kge/util/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/util/io.py -------------------------------------------------------------------------------- /kge/util/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/util/loss.py -------------------------------------------------------------------------------- /kge/util/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/util/optimizer.py -------------------------------------------------------------------------------- /kge/util/package.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/util/package.py -------------------------------------------------------------------------------- /kge/util/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/kge/util/sampler.py -------------------------------------------------------------------------------- /model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/model.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/requirements.txt -------------------------------------------------------------------------------- /train_embeddings/load_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/train_embeddings/load_data.py -------------------------------------------------------------------------------- /train_embeddings/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/train_embeddings/main.py -------------------------------------------------------------------------------- /train_embeddings/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/malllabiisc/EmbedKGQA/HEAD/train_embeddings/model.py --------------------------------------------------------------------------------