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

add visualization for training curves, pr, roc

Browse Source
This commit is contained in:
René Knaebel
2017-07-14 14:58:17 +02:00
parent d0418b9efa
commit b35f23e518
7 changed files with 299 additions and 99 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
Makefile
View File

@@ -1,5 +1,11 @@
run:
python3 main.py --modes train --batch 128 --model results/test --train data/rk_mini.csv.gz --epochs 10
test:
python3 main.py --modes train --epochs 1 --batch 128 --train data/rk_mini.csv.gz
python3 main.py --modes test --batch 128 --model results/test --test data/rk_mini.csv.gz
fancy:
python3 main.py --modes fancy --batch 128 --model results/test --test data/rk_mini.csv.gz
hyper:
python3 main.py --modes hyperband --epochs 1 --batch 64 --train data/rk_data.csv.gz
python3 main.py --modes hyperband --batch 64 --train data/rk_data.csv.gz