add hyperband savefile config, minor change of parameter name

This commit is contained in:
2017-10-03 18:58:54 +02:00
parent 68254d6629
commit 371a1dad05
4 changed files with 13 additions and 4 deletions

View File

@@ -9,7 +9,7 @@ def get_models_by_params(params: dict):
# mainly embedding model
network_type = params.get("type")
network_depth = params.get("depth")
embedding_size = params.get("embedding_size")
embedding_size = params.get("embedding")
input_length = params.get("input_length")
filter_embedding = params.get("filter_embedding")
kernel_embedding = params.get("kernel_embedding")