ma_cisco_malware/run.sh

49 lines
1.8 KiB
Bash

python3 main.py --mode train \
--train /tmp/rk/currentData.csv \
--model /tmp/rk/results/simple_both \
--epochs 25 \
--hidden_char_dims 64 \
--domain_embd 32 \
--batch 256 \
--balanced_weights \
--model_output both
python3 main.py --mode test --batch 512 --model /tmp/rk/results/simple_both --test /tmp/rk/futureData.csv
python3 main.py --mode train \
--train /tmp/rk/currentData.csv \
--model /tmp/rk/results/simple_client \
--epochs 25 \
--hidden_char_dims 64 \
--domain_embd 32 \
--batch 256 \
--balanced_weights \
--model_output client
python3 main.py --mode test --batch 512 --model /tmp/rk/results/simple_client --test /tmp/rk/futureData.csv
python3 main.py --mode train \
--train /tmp/rk/currentData.csv \
--model /tmp/rk/results/simple_new_both \
--epochs 25 \
--hidden_char_dims 64 \
--domain_embd 32 \
--batch 256 \
--balanced_weights \
--model_output both \
--new_model
python3 main.py --mode test --batch 512 --model /tmp/rk/results/simple_new_both --test /tmp/rk/futureData.csv
python3 main.py --mode train \
--train /tmp/rk/currentData.csv \
--model /tmp/rk/results/simple_new_client \
--epochs 25 \
--hidden_char_dims 64 \
--domain_embd 32 \
--batch 256 \
--balanced_weights \
--model_output client \
--new_model
python3 main.py --mode test --batch 512 --model /tmp/rk/results/simple_new_client --test /tmp/rk/futureData.csv