├── .gitignore ├── .idea └── workspace.xml ├── LICENSE ├── README.md ├── code ├── __init__.py ├── common │ ├── __init__.py │ ├── auxilliaries.py │ ├── evaluation.py │ ├── io.py │ ├── model_builder.py │ ├── optimizer_parameter_parser.py │ ├── settings_reader.py │ └── shared_functions.py ├── decoders │ ├── __init__.py │ ├── bilinear_diag.py │ ├── complex.py │ └── nonlinear_transform.py ├── encoders │ ├── __init__.py │ ├── affine_transform.py │ ├── bipartite_gcn.py │ ├── direct_message_gcn.py │ ├── message_gcns │ │ ├── __init__.py │ │ ├── gcn_basis.py │ │ ├── gcn_basis_concat.py │ │ ├── gcn_basis_plus_diag.py │ │ ├── gcn_basis_stored.py │ │ ├── gcn_basis_times_diag.py │ │ ├── gcn_diag.py │ │ ├── gcn_only_bias.py │ │ └── message_gcn.py │ ├── random_vertex_embedding.py │ └── relation_embedding.py ├── extras │ ├── __init__.py │ ├── dropover.py │ ├── graph_representations.py │ ├── highway_layer.py │ ├── residual_layer.py │ └── variational_encoding.py ├── model.py ├── optimization │ ├── __init__.py │ ├── abstract.py │ ├── optimize.py │ ├── setup.py │ ├── shared │ │ └── algorithms.py │ ├── tensorflow_backend │ │ └── algorithms.py │ └── theano_backend │ │ └── algorithms.py ├── split_model.py ├── tools │ ├── __init__.py │ ├── cluster.py │ ├── dictionaries.py │ ├── ensemble.py │ ├── make_degree_dataset.py │ ├── make_single_label_dataset.py │ └── make_split_dataset.py └── train.py ├── data ├── FB-Toutanova │ ├── README.txt │ ├── entities.dict │ ├── relations.dict │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── FB15k │ ├── entities.dict │ ├── relations.dict │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── Toy │ ├── entities.dict │ ├── relations.dict │ ├── test.txt │ ├── train.txt │ └── valid.txt └── wn18 │ ├── README │ ├── Wordnet3.0-LICENSE │ ├── entities.dict │ ├── relations.dict │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── run-train.sh ├── run-train_slurm.sh └── settings ├── complex.exp ├── distmult.exp ├── gcn_basis.exp └── gcn_block.exp /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/.gitignore -------------------------------------------------------------------------------- /.idea/workspace.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/.idea/workspace.xml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/README.md -------------------------------------------------------------------------------- /code/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/common/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/common/auxilliaries.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/common/auxilliaries.py -------------------------------------------------------------------------------- /code/common/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/common/evaluation.py -------------------------------------------------------------------------------- /code/common/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/common/io.py -------------------------------------------------------------------------------- /code/common/model_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/common/model_builder.py -------------------------------------------------------------------------------- /code/common/optimizer_parameter_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/common/optimizer_parameter_parser.py -------------------------------------------------------------------------------- /code/common/settings_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/common/settings_reader.py -------------------------------------------------------------------------------- /code/common/shared_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/common/shared_functions.py -------------------------------------------------------------------------------- /code/decoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/decoders/bilinear_diag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/decoders/bilinear_diag.py -------------------------------------------------------------------------------- /code/decoders/complex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/decoders/complex.py -------------------------------------------------------------------------------- /code/decoders/nonlinear_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/decoders/nonlinear_transform.py -------------------------------------------------------------------------------- /code/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/encoders/affine_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/encoders/affine_transform.py -------------------------------------------------------------------------------- /code/encoders/bipartite_gcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/encoders/bipartite_gcn.py -------------------------------------------------------------------------------- /code/encoders/direct_message_gcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/encoders/direct_message_gcn.py -------------------------------------------------------------------------------- /code/encoders/message_gcns/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/encoders/message_gcns/gcn_basis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/encoders/message_gcns/gcn_basis.py -------------------------------------------------------------------------------- /code/encoders/message_gcns/gcn_basis_concat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/encoders/message_gcns/gcn_basis_concat.py -------------------------------------------------------------------------------- /code/encoders/message_gcns/gcn_basis_plus_diag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/encoders/message_gcns/gcn_basis_plus_diag.py -------------------------------------------------------------------------------- /code/encoders/message_gcns/gcn_basis_stored.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/encoders/message_gcns/gcn_basis_stored.py -------------------------------------------------------------------------------- /code/encoders/message_gcns/gcn_basis_times_diag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/encoders/message_gcns/gcn_basis_times_diag.py -------------------------------------------------------------------------------- /code/encoders/message_gcns/gcn_diag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/encoders/message_gcns/gcn_diag.py -------------------------------------------------------------------------------- /code/encoders/message_gcns/gcn_only_bias.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/encoders/message_gcns/gcn_only_bias.py -------------------------------------------------------------------------------- /code/encoders/message_gcns/message_gcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/encoders/message_gcns/message_gcn.py -------------------------------------------------------------------------------- /code/encoders/random_vertex_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/encoders/random_vertex_embedding.py -------------------------------------------------------------------------------- /code/encoders/relation_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/encoders/relation_embedding.py -------------------------------------------------------------------------------- /code/extras/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/extras/dropover.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/extras/dropover.py -------------------------------------------------------------------------------- /code/extras/graph_representations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/extras/graph_representations.py -------------------------------------------------------------------------------- /code/extras/highway_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/extras/highway_layer.py -------------------------------------------------------------------------------- /code/extras/residual_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/extras/residual_layer.py -------------------------------------------------------------------------------- /code/extras/variational_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/extras/variational_encoding.py -------------------------------------------------------------------------------- /code/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/model.py -------------------------------------------------------------------------------- /code/optimization/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/optimization/abstract.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/optimization/abstract.py -------------------------------------------------------------------------------- /code/optimization/optimize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/optimization/optimize.py -------------------------------------------------------------------------------- /code/optimization/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/optimization/setup.py -------------------------------------------------------------------------------- /code/optimization/shared/algorithms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/optimization/shared/algorithms.py -------------------------------------------------------------------------------- /code/optimization/tensorflow_backend/algorithms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/optimization/tensorflow_backend/algorithms.py -------------------------------------------------------------------------------- /code/optimization/theano_backend/algorithms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/optimization/theano_backend/algorithms.py -------------------------------------------------------------------------------- /code/split_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/split_model.py -------------------------------------------------------------------------------- /code/tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/tools/cluster.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/tools/cluster.py -------------------------------------------------------------------------------- /code/tools/dictionaries.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/tools/dictionaries.py -------------------------------------------------------------------------------- /code/tools/ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/tools/ensemble.py -------------------------------------------------------------------------------- /code/tools/make_degree_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/tools/make_degree_dataset.py -------------------------------------------------------------------------------- /code/tools/make_single_label_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/tools/make_single_label_dataset.py -------------------------------------------------------------------------------- /code/tools/make_split_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/tools/make_split_dataset.py -------------------------------------------------------------------------------- /code/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/code/train.py -------------------------------------------------------------------------------- /data/FB-Toutanova/README.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/data/FB-Toutanova/README.txt -------------------------------------------------------------------------------- /data/FB-Toutanova/entities.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/data/FB-Toutanova/entities.dict -------------------------------------------------------------------------------- /data/FB-Toutanova/relations.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/data/FB-Toutanova/relations.dict -------------------------------------------------------------------------------- /data/FB-Toutanova/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/data/FB-Toutanova/test.txt -------------------------------------------------------------------------------- /data/FB-Toutanova/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/data/FB-Toutanova/train.txt -------------------------------------------------------------------------------- /data/FB-Toutanova/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/data/FB-Toutanova/valid.txt -------------------------------------------------------------------------------- /data/FB15k/entities.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/data/FB15k/entities.dict -------------------------------------------------------------------------------- /data/FB15k/relations.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/data/FB15k/relations.dict -------------------------------------------------------------------------------- /data/FB15k/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/data/FB15k/test.txt -------------------------------------------------------------------------------- /data/FB15k/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/data/FB15k/train.txt -------------------------------------------------------------------------------- /data/FB15k/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/data/FB15k/valid.txt -------------------------------------------------------------------------------- /data/Toy/entities.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/data/Toy/entities.dict -------------------------------------------------------------------------------- /data/Toy/relations.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/data/Toy/relations.dict -------------------------------------------------------------------------------- /data/Toy/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/data/Toy/test.txt -------------------------------------------------------------------------------- /data/Toy/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/data/Toy/train.txt -------------------------------------------------------------------------------- /data/Toy/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/data/Toy/valid.txt -------------------------------------------------------------------------------- /data/wn18/README: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/data/wn18/README -------------------------------------------------------------------------------- /data/wn18/Wordnet3.0-LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/data/wn18/Wordnet3.0-LICENSE -------------------------------------------------------------------------------- /data/wn18/entities.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/data/wn18/entities.dict -------------------------------------------------------------------------------- /data/wn18/relations.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/data/wn18/relations.dict -------------------------------------------------------------------------------- /data/wn18/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/data/wn18/test.txt -------------------------------------------------------------------------------- /data/wn18/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/data/wn18/train.txt -------------------------------------------------------------------------------- /data/wn18/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/data/wn18/valid.txt -------------------------------------------------------------------------------- /run-train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/run-train.sh -------------------------------------------------------------------------------- /run-train_slurm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/run-train_slurm.sh -------------------------------------------------------------------------------- /settings/complex.exp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/settings/complex.exp -------------------------------------------------------------------------------- /settings/distmult.exp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/settings/distmult.exp -------------------------------------------------------------------------------- /settings/gcn_basis.exp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/settings/gcn_basis.exp -------------------------------------------------------------------------------- /settings/gcn_block.exp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichSchli/RelationPrediction/HEAD/settings/gcn_block.exp --------------------------------------------------------------------------------