add tsne (does not work with big data)

fix model loading with custom selu function
This commit is contained in:
2017-09-22 10:01:12 +02:00
parent e2bf2dc90f
commit 607d74998c
4 changed files with 33 additions and 11 deletions

View File

@@ -1,5 +1,6 @@
import keras.backend as K
from models.renes_networks import selu
from . import flat_2, pauls_networks, renes_networks
@@ -24,11 +25,11 @@ def get_models_by_params(params: dict):
dense_dim = params.get("dense_main")
model_output = params.get("model_output", "both")
# create models
if network_depth == "small":
if network_depth == "flat1":
networks = pauls_networks
elif network_depth == "flat":
elif network_depth == "flat2":
networks = flat_2
elif network_depth == "medium":
elif network_depth == "deep1":
networks = renes_networks
else:
raise Exception("network not found")
@@ -49,6 +50,7 @@ def get_metrics():
("precision", precision),
("recall", recall),
("f1_score", f1_score),
("selu", selu)
])