From 98800920af541b8899761ba4fb8465b9fa5fa97c Mon Sep 17 00:00:00 2001 From: Maddin-619 Date: Wed, 25 Apr 2018 13:14:36 +0200 Subject: [PATCH] import keras_load_model func without previous training --- auto_ml/utils_models.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/auto_ml/utils_models.py b/auto_ml/utils_models.py index 4fb5898..cca68e2 100644 --- a/auto_ml/utils_models.py +++ b/auto_ml/utils_models.py @@ -590,7 +590,10 @@ def insert_deep_learning_model(pipeline_step, file_name): random_name = pipeline_step.model # Load the Keras model here keras_file_name = file_name[:-5] + random_name + '_keras_deep_learning_model.h5' - + try: + keras_load_model + except NameError: + from keras.models import load_model as keras_load_model model = keras_load_model(keras_file_name) # Put the model back in place so that we can still use it to get predictions without having to load it back in from disk