├── .gitignore ├── README.md ├── data ├── FB15K237 │ ├── FB15K237.pickle │ ├── README.txt │ ├── entities.dict │ ├── relations.dict │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── NELL-995 │ ├── 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 ├── nell_v1 │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── nell_v1_ind │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── nell_v2 │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── nell_v2_ind │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── nell_v3 │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── nell_v3_ind │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── nell_v4 │ ├── test.txt │ ├── train.txt │ └── valid.txt └── nell_v4_ind │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── ensembling ├── blend.py ├── compute_auc.py ├── compute_rank_metrics.py ├── get_ensemble_predictions.sh ├── get_kge_ensemble.sh └── score_triplets_kge.py ├── kge ├── README.md ├── dataloader.py ├── model.py ├── run.py └── run.sh ├── managers ├── evaluator.py └── trainer.py ├── model └── dgl │ ├── __init__.py │ ├── aggregators.py │ ├── graph_classifier.py │ ├── layers.py │ └── rgcn_model.py ├── requirements.txt ├── ruleN ├── RuleN.jar ├── auc_apply_rules.sh ├── get_auc_results.py ├── get_ranking_results.py ├── learn_rules.sh ├── process_predictions.py └── ranking_apply_rules.sh ├── subgraph_extraction ├── datasets.py └── graph_sampler.py ├── test_auc.py ├── test_ranking.py ├── train.py └── utils ├── clean_data.py ├── data_utils.py ├── dgl_utils.py ├── graph_utils.py ├── initialization_utils.py └── prepare_meta_data.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/README.md -------------------------------------------------------------------------------- /data/FB15K237/FB15K237.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/FB15K237/FB15K237.pickle -------------------------------------------------------------------------------- /data/FB15K237/README.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/FB15K237/README.txt -------------------------------------------------------------------------------- /data/FB15K237/entities.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/FB15K237/entities.dict -------------------------------------------------------------------------------- /data/FB15K237/relations.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/FB15K237/relations.dict -------------------------------------------------------------------------------- /data/FB15K237/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/FB15K237/test.txt -------------------------------------------------------------------------------- /data/FB15K237/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/FB15K237/train.txt -------------------------------------------------------------------------------- /data/FB15K237/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/FB15K237/valid.txt -------------------------------------------------------------------------------- /data/NELL-995/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/NELL-995/test.txt -------------------------------------------------------------------------------- /data/NELL-995/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/NELL-995/train.txt -------------------------------------------------------------------------------- /data/NELL-995/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/NELL-995/valid.txt -------------------------------------------------------------------------------- /data/WN18RR/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/WN18RR/test.txt -------------------------------------------------------------------------------- /data/WN18RR/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/WN18RR/train.txt -------------------------------------------------------------------------------- /data/WN18RR/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/WN18RR/valid.txt -------------------------------------------------------------------------------- /data/WN18RR_v1/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/WN18RR_v1/test.txt -------------------------------------------------------------------------------- /data/WN18RR_v1/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/WN18RR_v1/train.txt -------------------------------------------------------------------------------- /data/WN18RR_v1/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/WN18RR_v1/valid.txt -------------------------------------------------------------------------------- /data/WN18RR_v1_ind/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/WN18RR_v1_ind/test.txt -------------------------------------------------------------------------------- /data/WN18RR_v1_ind/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/WN18RR_v1_ind/train.txt -------------------------------------------------------------------------------- /data/WN18RR_v1_ind/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/WN18RR_v1_ind/valid.txt -------------------------------------------------------------------------------- /data/WN18RR_v2/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/WN18RR_v2/test.txt -------------------------------------------------------------------------------- /data/WN18RR_v2/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/WN18RR_v2/train.txt -------------------------------------------------------------------------------- /data/WN18RR_v2/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/WN18RR_v2/valid.txt -------------------------------------------------------------------------------- /data/WN18RR_v2_ind/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/WN18RR_v2_ind/test.txt -------------------------------------------------------------------------------- /data/WN18RR_v2_ind/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/WN18RR_v2_ind/train.txt -------------------------------------------------------------------------------- /data/WN18RR_v2_ind/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/WN18RR_v2_ind/valid.txt -------------------------------------------------------------------------------- /data/WN18RR_v3/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/WN18RR_v3/test.txt -------------------------------------------------------------------------------- /data/WN18RR_v3/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/WN18RR_v3/train.txt -------------------------------------------------------------------------------- /data/WN18RR_v3/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/WN18RR_v3/valid.txt -------------------------------------------------------------------------------- /data/WN18RR_v3_ind/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/WN18RR_v3_ind/test.txt -------------------------------------------------------------------------------- /data/WN18RR_v3_ind/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/WN18RR_v3_ind/train.txt -------------------------------------------------------------------------------- /data/WN18RR_v3_ind/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/WN18RR_v3_ind/valid.txt -------------------------------------------------------------------------------- /data/WN18RR_v4/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/WN18RR_v4/test.txt -------------------------------------------------------------------------------- /data/WN18RR_v4/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/WN18RR_v4/train.txt -------------------------------------------------------------------------------- /data/WN18RR_v4/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/WN18RR_v4/valid.txt -------------------------------------------------------------------------------- /data/WN18RR_v4_ind/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/WN18RR_v4_ind/test.txt -------------------------------------------------------------------------------- /data/WN18RR_v4_ind/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/WN18RR_v4_ind/train.txt -------------------------------------------------------------------------------- /data/WN18RR_v4_ind/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/WN18RR_v4_ind/valid.txt -------------------------------------------------------------------------------- /data/fb237_v1/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/fb237_v1/test.txt -------------------------------------------------------------------------------- /data/fb237_v1/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/fb237_v1/train.txt -------------------------------------------------------------------------------- /data/fb237_v1/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/fb237_v1/valid.txt -------------------------------------------------------------------------------- /data/fb237_v1_ind/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/fb237_v1_ind/test.txt -------------------------------------------------------------------------------- /data/fb237_v1_ind/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/fb237_v1_ind/train.txt -------------------------------------------------------------------------------- /data/fb237_v1_ind/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/fb237_v1_ind/valid.txt -------------------------------------------------------------------------------- /data/fb237_v2/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/fb237_v2/test.txt -------------------------------------------------------------------------------- /data/fb237_v2/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/fb237_v2/train.txt -------------------------------------------------------------------------------- /data/fb237_v2/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/fb237_v2/valid.txt -------------------------------------------------------------------------------- /data/fb237_v2_ind/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/fb237_v2_ind/test.txt -------------------------------------------------------------------------------- /data/fb237_v2_ind/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/fb237_v2_ind/train.txt -------------------------------------------------------------------------------- /data/fb237_v2_ind/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/fb237_v2_ind/valid.txt -------------------------------------------------------------------------------- /data/fb237_v3/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/fb237_v3/test.txt -------------------------------------------------------------------------------- /data/fb237_v3/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/fb237_v3/train.txt -------------------------------------------------------------------------------- /data/fb237_v3/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/fb237_v3/valid.txt -------------------------------------------------------------------------------- /data/fb237_v3_ind/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/fb237_v3_ind/test.txt -------------------------------------------------------------------------------- /data/fb237_v3_ind/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/fb237_v3_ind/train.txt -------------------------------------------------------------------------------- /data/fb237_v3_ind/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/fb237_v3_ind/valid.txt -------------------------------------------------------------------------------- /data/fb237_v4/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/fb237_v4/test.txt -------------------------------------------------------------------------------- /data/fb237_v4/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/fb237_v4/train.txt -------------------------------------------------------------------------------- /data/fb237_v4/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/fb237_v4/valid.txt -------------------------------------------------------------------------------- /data/fb237_v4_ind/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/fb237_v4_ind/test.txt -------------------------------------------------------------------------------- /data/fb237_v4_ind/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/fb237_v4_ind/train.txt -------------------------------------------------------------------------------- /data/fb237_v4_ind/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/fb237_v4_ind/valid.txt -------------------------------------------------------------------------------- /data/nell_v1/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/nell_v1/test.txt -------------------------------------------------------------------------------- /data/nell_v1/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/nell_v1/train.txt -------------------------------------------------------------------------------- /data/nell_v1/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/nell_v1/valid.txt -------------------------------------------------------------------------------- /data/nell_v1_ind/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/nell_v1_ind/test.txt -------------------------------------------------------------------------------- /data/nell_v1_ind/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/nell_v1_ind/train.txt -------------------------------------------------------------------------------- /data/nell_v1_ind/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/nell_v1_ind/valid.txt -------------------------------------------------------------------------------- /data/nell_v2/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/nell_v2/test.txt -------------------------------------------------------------------------------- /data/nell_v2/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/nell_v2/train.txt -------------------------------------------------------------------------------- /data/nell_v2/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/nell_v2/valid.txt -------------------------------------------------------------------------------- /data/nell_v2_ind/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/nell_v2_ind/test.txt -------------------------------------------------------------------------------- /data/nell_v2_ind/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/nell_v2_ind/train.txt -------------------------------------------------------------------------------- /data/nell_v2_ind/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/nell_v2_ind/valid.txt -------------------------------------------------------------------------------- /data/nell_v3/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/nell_v3/test.txt -------------------------------------------------------------------------------- /data/nell_v3/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/nell_v3/train.txt -------------------------------------------------------------------------------- /data/nell_v3/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/nell_v3/valid.txt -------------------------------------------------------------------------------- /data/nell_v3_ind/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/nell_v3_ind/test.txt -------------------------------------------------------------------------------- /data/nell_v3_ind/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/nell_v3_ind/train.txt -------------------------------------------------------------------------------- /data/nell_v3_ind/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/nell_v3_ind/valid.txt -------------------------------------------------------------------------------- /data/nell_v4/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/nell_v4/test.txt -------------------------------------------------------------------------------- /data/nell_v4/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/nell_v4/train.txt -------------------------------------------------------------------------------- /data/nell_v4/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/nell_v4/valid.txt -------------------------------------------------------------------------------- /data/nell_v4_ind/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/nell_v4_ind/test.txt -------------------------------------------------------------------------------- /data/nell_v4_ind/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/nell_v4_ind/train.txt -------------------------------------------------------------------------------- /data/nell_v4_ind/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/data/nell_v4_ind/valid.txt -------------------------------------------------------------------------------- /ensembling/blend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/ensembling/blend.py -------------------------------------------------------------------------------- /ensembling/compute_auc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/ensembling/compute_auc.py -------------------------------------------------------------------------------- /ensembling/compute_rank_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/ensembling/compute_rank_metrics.py -------------------------------------------------------------------------------- /ensembling/get_ensemble_predictions.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/ensembling/get_ensemble_predictions.sh -------------------------------------------------------------------------------- /ensembling/get_kge_ensemble.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/ensembling/get_kge_ensemble.sh -------------------------------------------------------------------------------- /ensembling/score_triplets_kge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/ensembling/score_triplets_kge.py -------------------------------------------------------------------------------- /kge/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/kge/README.md -------------------------------------------------------------------------------- /kge/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/kge/dataloader.py -------------------------------------------------------------------------------- /kge/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/kge/model.py -------------------------------------------------------------------------------- /kge/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/kge/run.py -------------------------------------------------------------------------------- /kge/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/kge/run.sh -------------------------------------------------------------------------------- /managers/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/managers/evaluator.py -------------------------------------------------------------------------------- /managers/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/managers/trainer.py -------------------------------------------------------------------------------- /model/dgl/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/dgl/aggregators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/model/dgl/aggregators.py -------------------------------------------------------------------------------- /model/dgl/graph_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/model/dgl/graph_classifier.py -------------------------------------------------------------------------------- /model/dgl/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/model/dgl/layers.py -------------------------------------------------------------------------------- /model/dgl/rgcn_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/model/dgl/rgcn_model.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/requirements.txt -------------------------------------------------------------------------------- /ruleN/RuleN.jar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/ruleN/RuleN.jar -------------------------------------------------------------------------------- /ruleN/auc_apply_rules.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/ruleN/auc_apply_rules.sh -------------------------------------------------------------------------------- /ruleN/get_auc_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/ruleN/get_auc_results.py -------------------------------------------------------------------------------- /ruleN/get_ranking_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/ruleN/get_ranking_results.py -------------------------------------------------------------------------------- /ruleN/learn_rules.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/ruleN/learn_rules.sh -------------------------------------------------------------------------------- /ruleN/process_predictions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/ruleN/process_predictions.py -------------------------------------------------------------------------------- /ruleN/ranking_apply_rules.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/ruleN/ranking_apply_rules.sh -------------------------------------------------------------------------------- /subgraph_extraction/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/subgraph_extraction/datasets.py -------------------------------------------------------------------------------- /subgraph_extraction/graph_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/subgraph_extraction/graph_sampler.py -------------------------------------------------------------------------------- /test_auc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/test_auc.py -------------------------------------------------------------------------------- /test_ranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/test_ranking.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/train.py -------------------------------------------------------------------------------- /utils/clean_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/utils/clean_data.py -------------------------------------------------------------------------------- /utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/utils/data_utils.py -------------------------------------------------------------------------------- /utils/dgl_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/utils/dgl_utils.py -------------------------------------------------------------------------------- /utils/graph_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/utils/graph_utils.py -------------------------------------------------------------------------------- /utils/initialization_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/utils/initialization_utils.py -------------------------------------------------------------------------------- /utils/prepare_meta_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/grail/HEAD/utils/prepare_meta_data.py --------------------------------------------------------------------------------