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

add soft parameter sharing network

Browse Source
This commit is contained in:
René Knaebel
2017-11-06 21:51:49 +01:00
parent 7b8dfcebbe
commit b1f48c1895
4 changed files with 82 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
models/__init__.py
View File

@@ -46,7 +46,10 @@ def get_models_by_params(params: dict):
long = networks.get_new_model2(0.25, flow_features, hidden_embedding, window_size, domain_length,
filter_main, kernel_main, dense_dim, embedding_model, model_output)
return embedding_model, final, inter, long
soft = networks.get_new_soft(0.25, flow_features, hidden_embedding, window_size, domain_length,
filter_main, kernel_main, dense_dim, embedding_model, model_output)
return embedding_model, final, inter, long, soft
def get_server_model_by_params(params: dict):