TwiRGCN - To Reproduce Results + Hyperparameters

################################################

# The following commands need to be run to set up environments and dependencies:

conda create --name twirgcn_env
conda activate trgcn_env
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge
conda install pyg -c pyg -c conda-forge
conda install -c conda-forge transformers

################################################

# How to run?

## TwiRGCN (average):

Train:

CUDA_VISIBLE_DEVICES=6 python -W ignore ./twirgcn_train.py --frozen 0 --eval_k 1 --max_epochs 100 \
 --lr 0.00004 --batch_size 32 --mode train --tkbc_model_file tcomplex_tq_tnt_200.pt  \
 --dataset timequestions --valid_freq 1 --valid_batch_size 5 \
 --save_to twirgcn_avg --lm_frozen 0 --eval_split valid --model twirgcn \
--attn_mode average


Eval:

 CUDA_VISIBLE_DEVICES=6 python -W ignore ./twirgcn_train.py \
 --mode eval --tkbc_model_file tcomplex_tq_tnt_200.pt \
 --dataset timequestions --model twirgcn --eval_split test  \
 --load_from twirgcn_avg --valid_batch_size 5 \
--attn_mode average

#-----------------------------------------

## TwiRGCN (average):

Train:

CUDA_VISIBLE_DEVICES=6 python -W ignore ./twirgcn_train.py --frozen 0 --eval_k 1 --max_epochs 100 \
 --lr 0.00004 --batch_size 32 --mode train --tkbc_model_file tcomplex_tq_tnt_200.pt  \
 --dataset timequestions --valid_freq 1 --valid_batch_size 5 \
 --save_to twirgcn_interval --lm_frozen 0 --eval_split valid --model twirgcn \
--attn_mode interval 


Eval:

 CUDA_VISIBLE_DEVICES=6 python -W ignore ./twirgcn_train.py \
 --mode eval --tkbc_model_file tcomplex_tq_tnt_200.pt \
 --dataset timequestions --model twirgcn --eval_split test  \
 --load_from twirgcn_interval --valid_batch_size 5 \
--attn_mode interval


################################################

Output files with top 10 predictions of model:

twirgcn_avg_emnlp_22_outputs.json
twirgcn_interval_emnlp_22_outputs.json


################################################



