run_ICEWS14.sh

#!/bin/bash
###
 # @Descripttion: 
 # @version: 
 # @Author: Assassin 567
 # @Date: 2021-11-11 10:39:39
 # @LastEditors: Hello KG
 # @LastEditTime: 2021-11-13 13:10:02
### 

for l_w in 0 
do

for loss_error_weight in 0.1 
do

for w_2 in 1e-4 
do

for DP_steps_set in 2 
do

for DP_num_edges_set in 15 
do

echo "GUP used : 3~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~"
echo "DP_steps_set: $DP_steps_set .   DP_num_edges_set: [200,$DP_num_edges_set,$DP_num_edges_set]~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~"
echo "Search regularization-weight:  $w_2.  loss_margin  $l_w. loss_error_weight: $loss_error_weight"

CUDA_VISIBLE_DEVICES=3 python ./train.py --warm_start_time 48 --emb_dim 768 10 10 10 --batch_size 256 --lr 0.002 \
--dataset ICEWS14 --epoch 6 --sampling 3 --device 0 --DP_steps $DP_steps_set --DP_num_edges 200 $DP_num_edges_set $DP_num_edges_set \
--max_attended_edges 30 --node_score_aggregation sum --ent_score_aggregation sum \
--ratio_update 0.75 --reg_weight $w_2 --loss_error_weight $loss_error_weight --loss_margin $l_w

done
done
done
done
done