ma_cisco_malware/run.sh

#!/usr/bin/env bash


for output in client both
do
    for depth in small medium
    do
        for mtype in inter final staggered
        do

            python main.py --mode train \
                        --train /tmp/rk/currentData.csv \
                        --model /tmp/rk/results/${output}_${depth}_${mtype} \
                        --epochs 50 \
                        --embd 64 \
                        --hidden_chaar_dims 128 \
                        --domain_embd 32 \
                        --batch 256 \
                        --balanced_weights \
                        --model_output ${output} \
                        --type ${mtype} \
                        --depth ${depth}

        done
    done
done