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

minor changes for pauls test

Browse Source
This commit is contained in:
René Knaebel 2017-09-11 12:59:13 +02:00
parent 6d8d7b19f3
commit 3a44efa775
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
main.py
View File

@ -201,7 +201,6 @@ def main_train(param=None):
epochs=args.epochs, epochs=args.epochs,
callbacks=callbacks, callbacks=callbacks,
shuffle=True, shuffle=True,
validation_split=0.3,
class_weight=custom_class_weights) class_weight=custom_class_weights)
else: else:

4
run_model.sh
View File

@ -21,8 +21,8 @@ do
--filter_embd 256 --kernel_embd 8 --dense_embd 128 \ --filter_embd 256 --kernel_embd 8 --dense_embd 128 \
--domain_embd 32 \ --domain_embd 32 \
--filter_main 32 --kernel_main 8 --dense_main 1024 \ --filter_main 32 --kernel_main 8 --dense_main 1024 \
--batch 256 \ --batch 128 \
--balanced_weights \ # --balanced_weights \
--model_output ${OUTPUT} \ --model_output ${OUTPUT} \
--type ${TYPE} \ --type ${TYPE} \
--depth ${DEPTH} --depth ${DEPTH}