Skip to content

TemporalKGTeam/xERTE

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 

Repository files navigation

xERTE: Explainable Subgraph Reasoning for Forecasting on Temporal Knowledge Graphs

This repository contains code for the reprsentation proposed in the ICLR2021 paper titled "Explainable Subgraph Reasoning for Forecasting on Temporal Knowledge Graphs".

Installation

  • clone repository
  • install virtualenv
pip install virtualenv
  • create virtual environment
virtualenv -p python3 venv
  • activate virtualenv
source venv/bin/activate
  • install packages
pip install -r requirements.txt
  • specify directory to save Checkpoint
cd tKGR
vim local_config.py

For example if you want to save checkpoints where local_config.py is

local_config.py

from pathlib import Path
save_dir = Path(__file__).parent.absolute()

Run:

Training:

python train.py --warm_start_time 48 --emb_dim 256 128 64 32 --batch_size 128 --lr 0.0002 --dataset ICEWS14_forecasting --epoch 10 --sampling 3 --device 0 --DP_steps 3 --DP_num_edges 15 --max_attended_edges 40 --node_score_aggregation sum --ent_score_aggregation sum
python train.py --warm_start_time 48 --emb_dim 256 128 64 32 --batch_size 128 --lr 0.0002 --dataset ICEWS0515_forecasting --epoch 10 --sampling 3 --device 0 --DP_steps 3 --DP_num_edges 15 --max_attended_edges 40 --node_score_aggregation sum --ent_score_aggregation sum
python train.py --warm_start_time 48 --emb_dim 256 128 64 32 --batch_size 128 --lr 0.0002 --dataset ICEWS18_forecasting --epoch 10 --sampling 3 --device 0 --DP_steps 3 --DP_num_edges 15 --max_attended_edges 60 --node_score_aggregation sum --ent_score_aggregation sum --ratio_update 0.75
python train.py --warm_start_time 48 --emb_dim 256 128 64 32 --batch_size 128 --lr 0.0002 --dataset YAGO1830 --epoch 10 --sampling 3 --device 0 --DP_steps 3 --DP_num_edges 15 --max_attended_edges 60 --node_score_aggregation sum --ent_score_aggregation sum --ratio_update 0.75

Evaluation:

python eval.py --load_checkpoint checkpoints_dir_name/checkpoint_name.pt --device 0

different num_neighbors can be specified by adding

--DP_num_neighbors 30 --max_attended_edges 40

--mongo can store configurations and results into mongodb, it's not necessary.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published