add new network architecture - server label moves to the middle

2017-07-29 19:42:36 +02:00
parent 8cd1023165
commit 820a5d1a4d
5 changed files with 151 additions and 40 deletions
--- a/models/init.py
+++ b/models/init.py
@@ -32,7 +32,10 @@ def get_models_by_params(params: dict):
    predict_model = networks.get_model(dropout, flow_features, domain_features, window_size, domain_length,
                                       filter_main, kernel_main, dense_dim, embedding_model)

-    return embedding_model, predict_model
+    new_model = networks.get_new_model(dropout, flow_features, domain_features, window_size, domain_length,
+                                       filter_main, kernel_main, dense_dim, embedding_model)
+
+    return embedding_model, predict_model, new_model


 def get_metrics():