Intstruction:
Please download the data and put it in the ./data/ folder

We did not upload the preprocessed entity histories for gdelt dataset as it exceeded the limit. It can be obtained by running ./data_process/gdelt/preprocess.py


Commands to run the code and reproduce the results:

Train ICEWS:
python run.py --actor_mode --epochs 12000 --eval 500 --margin 10 --lr 0.001 --seq_encoder att --n_head 4 --hist_len 20 --run_num 5


Train GDELT:
python run.py --actor_mode --dataset gdelt --epochs 12000 --eval 500 --margin 18 --lr 0.001 --seq_encoder att --n_head 4 --run_num 5


Test ICEWS:
python run.py --actor_mode  --margin 10 --lr 0.001 --seq_encoder att --n_head 4 --run_num 4 --hist_len 20 --checkpoint 11000 --test 

Test GDELT: 
python run.py --actor_mode  --margin 10 --lr 0.001 --seq_encoder att --n_head 4 --run_num 4 --checkpoint 11000 --test