ma_cisco_malware/run.sh

48 lines
1.4 KiB
Bash
Raw Normal View History

#!/usr/bin/env bash
2017-08-02 15:13:09 +02:00
RESDIR=$1
mkdir -p /tmp/rk/RESDIR
DATADIR=$2
for output in client both
do
for depth in small
do
for mtype in inter final
do
2017-08-02 15:13:09 +02:00
python main.py --mode train \
--train ${DATADIR}/currentData.csv \
--model ${RESDIR}/${output}_${depth}_${mtype} \
--epochs 50 \
--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 ${mtype} \
--depth ${depth}
2017-08-02 15:13:09 +02:00
done
done
done
for depth in small
do
python main.py --mode train \
--train ${DATADIR}/currentData.csv \
--model ${RESDIR}/both_${depth}_inter \
--epochs 50 \
--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 both \
--type staggered \
--depth ${depth}
done