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

add long final implementation

Browse Source
This commit is contained in:
René Knaebel
2017-11-05 22:52:50 +01:00
parent 14fef66a55
commit 7b8dfcebbe
4 changed files with 53 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
models/__init__.py
View File

@@ -37,13 +37,16 @@ def get_models_by_params(params: dict):
embedding_model = networks.get_embedding(embedding_size, domain_length, filter_embedding, kernel_embedding,
hidden_embedding, 0.5)
old_model = networks.get_model(0.25, flow_features, hidden_embedding, window_size, domain_length,
final = networks.get_model(0.25, flow_features, hidden_embedding, window_size, domain_length,
filter_main, kernel_main, dense_dim, embedding_model, model_output)
inter = networks.get_new_model(0.25, flow_features, hidden_embedding, window_size, domain_length,
filter_main, kernel_main, dense_dim, embedding_model, model_output)
new_model = networks.get_new_model(0.25, flow_features, hidden_embedding, window_size, domain_length,
filter_main, kernel_main, dense_dim, embedding_model, model_output)
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, old_model, new_model
return embedding_model, final, inter, long
def get_server_model_by_params(params: dict):