ma_cisco_malware/run_model.sh

29 lines
697 B
Bash
Raw Normal View History

#!/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