ma_cisco_malware/run.sh

42 lines
1.2 KiB
Bash

#!/usr/bin/env bash
for output in client both
do
for depth in small medium
do
for mtype in inter final
do
python main.py --mode train \
--train /tmp/rk/currentData.csv \
--model /tmp/rk/results/${output}_${depth}_${mtype} \
--epochs 50 \
--embd 64 \
--hidden_char_dims 128 \
--domain_embd 32 \
--batch 256 \
--balanced_weights \
--model_output ${output} \
--type ${mtype} \
--depth ${depth}
done
done
done
for depth in small medium
do
python main.py --mode train \
--train /tmp/rk/currentData.csv \
--model /tmp/rk/results/both_${depth}_inter \
--epochs 50 \
--embd 64 \
--hidden_char_dims 128 \
--domain_embd 32 \
--batch 256 \
--balanced_weights \
--model_output both \
--type inter \
--depth ${depth}
done