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

add argument for model outputs

Browse Source 
BUG: need to check --new_model --model_output server
This commit is contained in:
René Knaebel
2017-08-02 12:58:09 +02:00
parent f4da147688
commit 1e781d5491
5 changed files with 45 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
models/__init__.py
View File

@@ -22,16 +22,17 @@ def get_models_by_params(params: dict):
filter_main = params.get("filter_main")
kernel_main = params.get("kernels_main")
dense_dim = params.get("dense_main")
model_output = params.get("model_output", "both")
# create models
networks = renes_networks if network_type == "rene" else pauls_networks
embedding_model = networks.get_embedding(embedding_size, input_length, filter_embedding, kernel_embedding,
hidden_embedding, drop_out=dropout)
hidden_embedding, dropout)
predict_model = networks.get_model(dropout, flow_features, domain_features, window_size, domain_length,
filter_main, kernel_main, dense_dim, embedding_model)
filter_main, kernel_main, dense_dim, embedding_model, model_output)
new_model = networks.get_new_model(dropout, flow_features, domain_features, window_size, domain_length,
filter_main, kernel_main, dense_dim, embedding_model)
filter_main, kernel_main, dense_dim, embedding_model, model_output)
return embedding_model, predict_model, new_model