#!/bin/bash

if [ "$#" -lt 1 ]; then
  echo "Usage) $0 [seed]"
  exit 0
fi
SEED="$1"

scriptDir=$(dirname -- "$(readlink -f -- "$BASH_SOURCE")")
cd "${scriptDir}"/../../src/ || exit
emb_dim=192


python3 train.py \
--graph WIKI \
--seed "${SEED}" \
--log-dir evokg_linkpred_seed"${SEED}" \
--result-file-prefix opt_edge_ \
--clean-up-run-best-checkpoint \
--static-entity-embed-dim "${emb_dim}" \
--temporal-dynamic-entity-embed-dim "${emb_dim}" \
--structural-dynamic-entity-embed-dim "${emb_dim}" \
--dropout 0.2 \
--eval-from 0 \
--eval-every 3 \
--early-stop \
--patience 4 \
--lr 0.001 \
--gpu 0 \
--optimize edge \
--eval edge \
--full-link-pred-validation \
--full-link-pred-test "${@:2}"


python3 train.py \
--graph WIKI \
--seed "${SEED}" \
--log-dir evokg_linkpred_seed"${SEED}" \
--result-file-prefix opt_both_ \
--clean-up-run-best-checkpoint \
--static-entity-embed-dim "${emb_dim}" \
--temporal-dynamic-entity-embed-dim "${emb_dim}" \
--structural-dynamic-entity-embed-dim "${emb_dim}" \
--dropout 0.2 \
--eval-from 0 \
--eval-every 1 \
--early-stop \
--patience 5 \
--lr 0.0005 \
--gpu 0 \
--load-best edge \
--optimize both \
--eval both \
--full-link-pred-validation \
--full-link-pred-test \
--inter-event-time-mode node2node_inter_event_times "${@:2}"