add tsne (does not work with big data)
fix model loading with custom selu function
This commit is contained in:
@@ -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)
|
||||
])
|
||||
|
||||
|
||||
|
Reference in New Issue
Block a user