Skip to content

Latest commit

 

History

History
43 lines (38 loc) · 1019 Bytes

README.md

File metadata and controls

43 lines (38 loc) · 1019 Bytes
Raw

KGReasoning

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:

Model with multiple individual layers is on branch AttentionLayer Model with shared layer on branch SharedAttention

checkout to branch you want:

git checkout -b BranchName origin/BranchName

CUDA_VISIBLE_DEVICES=0 python train.py --warm_start_time 48 --emb_dim 256 --emb_dim_sm 64 --batch_size 128 --lr 0.0002 --dataset ICEWS14_forecasting --epoch 20 --sampling 3 --device 0 --DP_steps 3 --DP_num_neighbors 15 --max_attended_nodes 40 --emb_static_temporal_ratio 2