#!/usr/bin/env bash


N=$1
OUTPUT=$2
DEPTH=$3
TYPE=$4
RESDIR=$5
mkdir -p /tmp/rk/${RESDIR}
DATADIR=$6

EPOCHS=100

for i in {1..$N}
do
    python main.py --mode train \
                --train ${DATADIR}/currentData.csv \
                --model ${RESDIR}/${OUTPUT}_${TYPE}_$i \
                --epochs $EPOCHS \
                --embd 128 \
                --filter_embd 256 --kernel_embd 8 --dense_embd 128 \
                --domain_embd 32 \
                --filter_main 32  --kernel_main 8 --dense_main 1024 \
                --batch 256 \
                --balanced_weights \
                --model_output ${OUTPUT} \
                --type ${TYPE} \
                --depth ${DEPTH}
done