ma_cisco_malware/test.sh

13 lines
278 B
Bash

#!/usr/bin/env bash
RESDIR=$1
DATADIR=$2
for output in client both
do
python3 main.py --mode test --batch 1024 \
--models ${RESDIR}/${output}_*/ \
--test ${DATADIR}/futureData.csv \
--model_output ${output}
done