├── .gitignore ├── README.md ├── data ├── FB15K237 │ ├── FB15K237.pickle │ ├── README.txt │ ├── entities.dict │ ├── relations.dict │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── WN18RR │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── WN18RR_v1 │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── WN18RR_v1_ind │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── WN18RR_v2 │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── WN18RR_v2_ind │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── WN18RR_v3 │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── WN18RR_v3_ind │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── WN18RR_v4 │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── WN18RR_v4_ind │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── fb237_v1 │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── fb237_v1_ind │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── fb237_v2 │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── fb237_v2_ind │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── fb237_v3 │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── fb237_v3_ind │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── fb237_v4 │ ├── test.txt │ ├── train.txt │ └── valid.txt └── fb237_v4_ind │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── managers ├── __pycache__ │ ├── evaluator.cpython-36.pyc │ └── trainer.cpython-36.pyc ├── evaluator.py └── trainer.py ├── model └── dgl │ ├── __init__.py │ ├── __pycache__ │ ├── __init__.cpython-36.pyc │ ├── aggregators.cpython-36.pyc │ ├── batch_gru.cpython-36.pyc │ ├── discriminator.cpython-36.pyc │ ├── graph_classifier.cpython-36.pyc │ ├── layers.cpython-36.pyc │ └── rgcn_model.cpython-36.pyc │ ├── aggregators.py │ ├── batch_gru.py │ ├── discriminator.py │ ├── graph_classifier.py │ ├── layers.py │ └── rgcn_model.py ├── requirements.txt ├── snri.png ├── subgraph_extraction ├── __pycache__ │ ├── datasets.cpython-36.pyc │ └── graph_sampler.cpython-36.pyc ├── datasets.py └── graph_sampler.py ├── test_auc.py ├── test_ranking.py ├── train.py └── utils ├── __pycache__ ├── data_utils.cpython-36.pyc ├── dgl_utils.cpython-36.pyc ├── graph_utils.cpython-36.pyc └── initialization_utils.cpython-36.pyc ├── clean_data.py ├── data_utils.py ├── dgl_utils.py ├── graph_utils.py ├── initialization_utils.py └── prepare_meta_data.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/README.md -------------------------------------------------------------------------------- /data/FB15K237/FB15K237.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/FB15K237/FB15K237.pickle -------------------------------------------------------------------------------- /data/FB15K237/README.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/FB15K237/README.txt -------------------------------------------------------------------------------- /data/FB15K237/entities.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/FB15K237/entities.dict -------------------------------------------------------------------------------- /data/FB15K237/relations.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/FB15K237/relations.dict -------------------------------------------------------------------------------- /data/FB15K237/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/FB15K237/test.txt -------------------------------------------------------------------------------- /data/FB15K237/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/FB15K237/train.txt -------------------------------------------------------------------------------- /data/FB15K237/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/FB15K237/valid.txt -------------------------------------------------------------------------------- /data/WN18RR/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/WN18RR/test.txt -------------------------------------------------------------------------------- /data/WN18RR/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/WN18RR/train.txt -------------------------------------------------------------------------------- /data/WN18RR/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/WN18RR/valid.txt -------------------------------------------------------------------------------- /data/WN18RR_v1/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/WN18RR_v1/test.txt -------------------------------------------------------------------------------- /data/WN18RR_v1/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/WN18RR_v1/train.txt -------------------------------------------------------------------------------- /data/WN18RR_v1/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/WN18RR_v1/valid.txt -------------------------------------------------------------------------------- /data/WN18RR_v1_ind/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/WN18RR_v1_ind/test.txt -------------------------------------------------------------------------------- /data/WN18RR_v1_ind/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/WN18RR_v1_ind/train.txt -------------------------------------------------------------------------------- /data/WN18RR_v1_ind/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/WN18RR_v1_ind/valid.txt -------------------------------------------------------------------------------- /data/WN18RR_v2/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/WN18RR_v2/test.txt -------------------------------------------------------------------------------- /data/WN18RR_v2/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/WN18RR_v2/train.txt -------------------------------------------------------------------------------- /data/WN18RR_v2/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/WN18RR_v2/valid.txt -------------------------------------------------------------------------------- /data/WN18RR_v2_ind/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/WN18RR_v2_ind/test.txt -------------------------------------------------------------------------------- /data/WN18RR_v2_ind/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/WN18RR_v2_ind/train.txt -------------------------------------------------------------------------------- /data/WN18RR_v2_ind/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/WN18RR_v2_ind/valid.txt -------------------------------------------------------------------------------- /data/WN18RR_v3/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/WN18RR_v3/test.txt -------------------------------------------------------------------------------- /data/WN18RR_v3/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/WN18RR_v3/train.txt -------------------------------------------------------------------------------- /data/WN18RR_v3/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/WN18RR_v3/valid.txt -------------------------------------------------------------------------------- /data/WN18RR_v3_ind/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/WN18RR_v3_ind/test.txt -------------------------------------------------------------------------------- /data/WN18RR_v3_ind/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/WN18RR_v3_ind/train.txt -------------------------------------------------------------------------------- /data/WN18RR_v3_ind/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/WN18RR_v3_ind/valid.txt -------------------------------------------------------------------------------- /data/WN18RR_v4/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/WN18RR_v4/test.txt -------------------------------------------------------------------------------- /data/WN18RR_v4/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/WN18RR_v4/train.txt -------------------------------------------------------------------------------- /data/WN18RR_v4/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/WN18RR_v4/valid.txt -------------------------------------------------------------------------------- /data/WN18RR_v4_ind/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/WN18RR_v4_ind/test.txt -------------------------------------------------------------------------------- /data/WN18RR_v4_ind/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/WN18RR_v4_ind/train.txt -------------------------------------------------------------------------------- /data/WN18RR_v4_ind/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/WN18RR_v4_ind/valid.txt -------------------------------------------------------------------------------- /data/fb237_v1/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/fb237_v1/test.txt -------------------------------------------------------------------------------- /data/fb237_v1/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/fb237_v1/train.txt -------------------------------------------------------------------------------- /data/fb237_v1/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/fb237_v1/valid.txt -------------------------------------------------------------------------------- /data/fb237_v1_ind/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/fb237_v1_ind/test.txt -------------------------------------------------------------------------------- /data/fb237_v1_ind/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/fb237_v1_ind/train.txt -------------------------------------------------------------------------------- /data/fb237_v1_ind/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/fb237_v1_ind/valid.txt -------------------------------------------------------------------------------- /data/fb237_v2/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/fb237_v2/test.txt -------------------------------------------------------------------------------- /data/fb237_v2/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/fb237_v2/train.txt -------------------------------------------------------------------------------- /data/fb237_v2/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/fb237_v2/valid.txt -------------------------------------------------------------------------------- /data/fb237_v2_ind/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/fb237_v2_ind/test.txt -------------------------------------------------------------------------------- /data/fb237_v2_ind/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/fb237_v2_ind/train.txt -------------------------------------------------------------------------------- /data/fb237_v2_ind/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/fb237_v2_ind/valid.txt -------------------------------------------------------------------------------- /data/fb237_v3/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/fb237_v3/test.txt -------------------------------------------------------------------------------- /data/fb237_v3/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/fb237_v3/train.txt -------------------------------------------------------------------------------- /data/fb237_v3/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/fb237_v3/valid.txt -------------------------------------------------------------------------------- /data/fb237_v3_ind/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/fb237_v3_ind/test.txt -------------------------------------------------------------------------------- /data/fb237_v3_ind/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/fb237_v3_ind/train.txt -------------------------------------------------------------------------------- /data/fb237_v3_ind/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/fb237_v3_ind/valid.txt -------------------------------------------------------------------------------- /data/fb237_v4/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/fb237_v4/test.txt -------------------------------------------------------------------------------- /data/fb237_v4/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/fb237_v4/train.txt -------------------------------------------------------------------------------- /data/fb237_v4/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/fb237_v4/valid.txt -------------------------------------------------------------------------------- /data/fb237_v4_ind/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/fb237_v4_ind/test.txt -------------------------------------------------------------------------------- /data/fb237_v4_ind/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/fb237_v4_ind/train.txt -------------------------------------------------------------------------------- /data/fb237_v4_ind/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/data/fb237_v4_ind/valid.txt -------------------------------------------------------------------------------- /managers/__pycache__/evaluator.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/managers/__pycache__/evaluator.cpython-36.pyc -------------------------------------------------------------------------------- /managers/__pycache__/trainer.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/managers/__pycache__/trainer.cpython-36.pyc -------------------------------------------------------------------------------- /managers/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/managers/evaluator.py -------------------------------------------------------------------------------- /managers/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/managers/trainer.py -------------------------------------------------------------------------------- /model/dgl/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/dgl/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/model/dgl/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /model/dgl/__pycache__/aggregators.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/model/dgl/__pycache__/aggregators.cpython-36.pyc -------------------------------------------------------------------------------- /model/dgl/__pycache__/batch_gru.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/model/dgl/__pycache__/batch_gru.cpython-36.pyc -------------------------------------------------------------------------------- /model/dgl/__pycache__/discriminator.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/model/dgl/__pycache__/discriminator.cpython-36.pyc -------------------------------------------------------------------------------- /model/dgl/__pycache__/graph_classifier.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/model/dgl/__pycache__/graph_classifier.cpython-36.pyc -------------------------------------------------------------------------------- /model/dgl/__pycache__/layers.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/model/dgl/__pycache__/layers.cpython-36.pyc -------------------------------------------------------------------------------- /model/dgl/__pycache__/rgcn_model.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/model/dgl/__pycache__/rgcn_model.cpython-36.pyc -------------------------------------------------------------------------------- /model/dgl/aggregators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/model/dgl/aggregators.py -------------------------------------------------------------------------------- /model/dgl/batch_gru.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/model/dgl/batch_gru.py -------------------------------------------------------------------------------- /model/dgl/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/model/dgl/discriminator.py -------------------------------------------------------------------------------- /model/dgl/graph_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/model/dgl/graph_classifier.py -------------------------------------------------------------------------------- /model/dgl/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/model/dgl/layers.py -------------------------------------------------------------------------------- /model/dgl/rgcn_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/model/dgl/rgcn_model.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/requirements.txt -------------------------------------------------------------------------------- /snri.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/snri.png -------------------------------------------------------------------------------- /subgraph_extraction/__pycache__/datasets.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/subgraph_extraction/__pycache__/datasets.cpython-36.pyc -------------------------------------------------------------------------------- /subgraph_extraction/__pycache__/graph_sampler.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/subgraph_extraction/__pycache__/graph_sampler.cpython-36.pyc -------------------------------------------------------------------------------- /subgraph_extraction/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/subgraph_extraction/datasets.py -------------------------------------------------------------------------------- /subgraph_extraction/graph_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/subgraph_extraction/graph_sampler.py -------------------------------------------------------------------------------- /test_auc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/test_auc.py -------------------------------------------------------------------------------- /test_ranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/test_ranking.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/train.py -------------------------------------------------------------------------------- /utils/__pycache__/data_utils.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/utils/__pycache__/data_utils.cpython-36.pyc -------------------------------------------------------------------------------- /utils/__pycache__/dgl_utils.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/utils/__pycache__/dgl_utils.cpython-36.pyc -------------------------------------------------------------------------------- /utils/__pycache__/graph_utils.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/utils/__pycache__/graph_utils.cpython-36.pyc -------------------------------------------------------------------------------- /utils/__pycache__/initialization_utils.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/utils/__pycache__/initialization_utils.cpython-36.pyc -------------------------------------------------------------------------------- /utils/clean_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/utils/clean_data.py -------------------------------------------------------------------------------- /utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/utils/data_utils.py -------------------------------------------------------------------------------- /utils/dgl_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/utils/dgl_utils.py -------------------------------------------------------------------------------- /utils/graph_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/utils/graph_utils.py -------------------------------------------------------------------------------- /utils/initialization_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/utils/initialization_utils.py -------------------------------------------------------------------------------- /utils/prepare_meta_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tebmer/SNRI/HEAD/utils/prepare_meta_data.py --------------------------------------------------------------------------------