- Download DRKG or use the following command:
wget https://dgl-data.s3-us-west-2.amazonaws.com/dataset/DRKG/drkg.tar.gz
-
When you untar
drkg.tar.gz
, you will finddrkg.tsv
. -
Add
drkg.tsv
toInput/DRKG
.
- TrainGNN.ipynb : Takes as input
drkg.tsv
and outputs a trained Graph Neural Network (can be either GCN, GraphSAGE or GAT) toOutput/GNNModels
. - ExplainabilityUseCase.ipynb : Takes as input a model and its precitions, and returns a dictionary (for a specific use case) containing the most important nodes that were most influential for the prediction(s) using Saliency Maps.
- PlotSubgraph.ipynb : Plots the explanatory subgraph.