rene/ma_cisco_malware
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix run scripts

Browse Source
This commit is contained in:
René Knaebel 2017-08-05 09:40:40 +02:00
parent 5a02f582cd
commit 954dfcf9f9
3 changed files with 18 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
main.py
View File

@ -238,7 +238,8 @@ def main_visualization():
# "{}/server_cov.png".format(args.model_path), # "{}/server_cov.png".format(args.model_path),
# normalize=False, title="Server Confusion Matrix") # normalize=False, title="Server Confusion Matrix")
logger.info("visualize embedding") logger.info("visualize embedding")
domain_encs, labels = dataset.load_or_generate_domains(args.test_data, args.domain_length)
domain_embedding = np.load(args.model_path + "/domain_embds.npy")
visualize.plot_embedding(domain_embedding, labels, path="{}/embd.png".format(args.model_path)) visualize.plot_embedding(domain_embedding, labels, path="{}/embd.png".format(args.model_path))

16
run.sh
View File

@ -2,7 +2,7 @@
python main.py --mode train \ python main.py --mode train \
--train /tmp/rk/currentData.csv \ --train /tmp/rk/currentData.csv \
--model /tmp/rk/results/simple_both \ --model /tmp/rk/results/small_both \
--epochs 25 \ --epochs 25 \
--embd 64 \ --embd 64 \
--hidden_chaar_dims 128 \ --hidden_chaar_dims 128 \
@ -13,7 +13,7 @@ python main.py --mode train \
python main.py --mode train \ python main.py --mode train \
--train /tmp/rk/currentData.csv \ --train /tmp/rk/currentData.csv \
--model /tmp/rk/results/simple_client \ --model /tmp/rk/results/small_client \
--epochs 25 \ --epochs 25 \
--embd 64 \ --embd 64 \
--hidden_char_dims 128 \ --hidden_char_dims 128 \
@ -24,7 +24,7 @@ python main.py --mode train \
python main.py --mode train \ python main.py --mode train \
--train /tmp/rk/currentData.csv \ --train /tmp/rk/currentData.csv \
--model /tmp/rk/results/simple_new_both \ --model /tmp/rk/results/small_new_both \
--epochs 25 \ --epochs 25 \
--embd 64 \ --embd 64 \
--hidden_char_dims 128 \ --hidden_char_dims 128 \
@ -36,7 +36,7 @@ python main.py --mode train \
python main.py --mode train \ python main.py --mode train \
--train /tmp/rk/currentData.csv \ --train /tmp/rk/currentData.csv \
--model /tmp/rk/results/simple_new_client \ --model /tmp/rk/results/small_new_client \
--epochs 25 \ --epochs 25 \
--embd 64 \ --embd 64 \
--hidden_char_dims 128 \ --hidden_char_dims 128 \
@ -49,7 +49,7 @@ python main.py --mode train \
## ##
python main.py --mode train \ python main.py --mode train \
--train /tmp/rk/currentData.csv \ --train /tmp/rk/currentData.csv \
--model /tmp/rk/results/simple_both \ --model /tmp/rk/results/medium_both \
--epochs 25 \ --epochs 25 \
--embd 64 \ --embd 64 \
--hidden_chaar_dims 128 \ --hidden_chaar_dims 128 \
@ -61,7 +61,7 @@ python main.py --mode train \
python main.py --mode train \ python main.py --mode train \
--train /tmp/rk/currentData.csv \ --train /tmp/rk/currentData.csv \
--model /tmp/rk/results/simple_client \ --model /tmp/rk/results/medium_client \
--epochs 25 \ --epochs 25 \
--embd 64 \ --embd 64 \
--hidden_char_dims 128 \ --hidden_char_dims 128 \
@ -73,7 +73,7 @@ python main.py --mode train \
python main.py --mode train \ python main.py --mode train \
--train /tmp/rk/currentData.csv \ --train /tmp/rk/currentData.csv \
--model /tmp/rk/results/simple_new_both \ --model /tmp/rk/results/medium_new_both \
--epochs 25 \ --epochs 25 \
--embd 64 \ --embd 64 \
--hidden_char_dims 128 \ --hidden_char_dims 128 \
@ -86,7 +86,7 @@ python main.py --mode train \
python main.py --mode train \ python main.py --mode train \
--train /tmp/rk/currentData.csv \ --train /tmp/rk/currentData.csv \
--model /tmp/rk/results/simple_new_client \ --model /tmp/rk/results/medium_new_client \
--epochs 25 \ --epochs 25 \
--embd 64 \ --embd 64 \
--hidden_char_dims 128 \ --hidden_char_dims 128 \

16
test.sh
View File

@ -1,11 +1,11 @@
#!/usr/bin/env bash #!/usr/bin/env bash
python main.py --mode test --batch 1024 --model /tmp/rk/results/simple_both --test /tmp/rk/futureData.csv --model_output both --type paul python main.py --mode test --batch 1024 --model /tmp/rk/results/small_both --test /tmp/rk/futureData.csv --model_output both --type paul
python main.py --mode test --batch 1024 --model /tmp/rk/results/simple_client --test /tmp/rk/futureData.csv --model_output client --type paul python main.py --mode test --batch 1024 --model /tmp/rk/results/small_client --test /tmp/rk/futureData.csv --model_output client --type paul
python main.py --mode test --batch 1024 --model /tmp/rk/results/simple_new_both --test /tmp/rk/futureData.csv --model_output both --type paul python main.py --mode test --batch 1024 --model /tmp/rk/results/small_new_both --test /tmp/rk/futureData.csv --model_output both --type paul
python main.py --mode test --batch 1024 --model /tmp/rk/results/simple_new_client --test /tmp/rk/futureData.csv --model_output client --type paul python main.py --mode test --batch 1024 --model /tmp/rk/results/small_new_client --test /tmp/rk/futureData.csv --model_output client --type paul
python main.py --mode test --batch 1024 --model /tmp/rk/results/simple_both --test /tmp/rk/futureData.csv --model_output both --type rene python main.py --mode test --batch 1024 --model /tmp/rk/results/medium_both --test /tmp/rk/futureData.csv --model_output both --type rene
python main.py --mode test --batch 1024 --model /tmp/rk/results/simple_client --test /tmp/rk/futureData.csv --model_output client --type rene python main.py --mode test --batch 1024 --model /tmp/rk/results/medium_client --test /tmp/rk/futureData.csv --model_output client --type rene
python main.py --mode test --batch 1024 --model /tmp/rk/results/simple_new_both --test /tmp/rk/futureData.csv --model_output both --type rene python main.py --mode test --batch 1024 --model /tmp/rk/results/medium_new_both --test /tmp/rk/futureData.csv --model_output both --type rene
python main.py --mode test --batch 1024 --model /tmp/rk/results/simple_new_client --test /tmp/rk/futureData.csv --model_output client --type rene python main.py --mode test --batch 1024 --model /tmp/rk/results/medium_new_client --test /tmp/rk/futureData.csv --model_output client --type rene