add server classification model

models/__init__.py

@@ -7,14 +7,14 @@ from . import flat_2, pauls_networks, renes_networks
 def get_models_by_params(params: dict):
     # decomposing param section
     # mainly embedding model
-    network_type = params.get("type")
+    # network_type = params.get("type")
     network_depth = params.get("depth")
     embedding_size = params.get("embedding")
     input_length = params.get("input_length")
     filter_embedding = params.get("filter_embedding")
     kernel_embedding = params.get("kernel_embedding")
     hidden_embedding = params.get("dense_embedding")
-    dropout = params.get("dropout")
+    # dropout = params.get("dropout")
     # mainly prediction model
     flow_features = params.get("flow_features")
     window_size = params.get("window_size")
@@ -44,7 +44,36 @@ def get_models_by_params(params: dict):
     return embedding_model, old_model, new_model
 
 
-def get_metrics():
+def get_server_model_by_params(params: dict):
+    # decomposing param section
+    # mainly embedding model
+    network_depth = params.get("depth")
+    embedding_size = params.get("embedding")
+    input_length = params.get("input_length")
+    filter_embedding = params.get("filter_embedding")
+    kernel_embedding = params.get("kernel_embedding")
+    hidden_embedding = params.get("dense_embedding")
+    # mainly prediction model
+    flow_features = params.get("flow_features")
+    domain_length = params.get("domain_length")
+    dense_dim = params.get("dense_main")
+    # create models
+    if network_depth == "flat1":
+        networks = pauls_networks
+    elif network_depth == "flat2":
+        networks = flat_2
+    elif network_depth == "deep1":
+        networks = renes_networks
+    else:
+        raise Exception("network not found")
+    
+    embedding_model = networks.get_embedding(embedding_size, input_length, filter_embedding, kernel_embedding,
+                                             hidden_embedding, 0.5)
+    
+    return networks.get_server_model(flow_features, domain_length, dense_dim, embedding_model)
+
+
+def get_custom_objects():
     return dict([
         ("precision", precision),
         ("recall", recall),