#!/usr/bin/env bash
cd ../src
#
# HGRU INIT (LARGE)
#
THEANO_FLAGS=mode=FAST_RUN,device=gpu python train_hier_gru.py xing-dense lso-test 500 500 \
--loss top1 --hidden_act tanh \
--user_propagation_mode init --user_to_output 0 \
--adapt adagrad --learning_rate 0.1 --momentum 0.1 --batch_size 100 \
--dropout_p_hidden_usr 0.1 \
--dropout_p_hidden_ses 0.2 \
--dropout_p_init 0.2 \
--n_epochs 10 --train_random_order 0 \
--eval_cutoff 20 \
--user_key user_id --item_key item_id --session_key session_id --time_key created_at \
--rnd_seed 1

#
# HGRU ALL (LARGE)
#
THEANO_FLAGS=mode=FAST_RUN,device=gpu python train_hier_gru.py xing-dense lso-test 500 500 \
--loss top1 --hidden_act tanh \
--user_propagation_mode all --user_to_output 0 \
--adapt adagrad --learning_rate 0.05 --momentum 0.3 --batch_size 50 \
--dropout_p_hidden_usr 0.0 \
--dropout_p_hidden_ses 0.2 \
--dropout_p_init 0.2 \
--n_epochs 10 --train_random_order 0 \
--eval_cutoff 20 \
--user_key user_id --item_key item_id --session_key session_id --time_key created_at \
--rnd_seed 1
