From f0429d4fe877752415af34661c07433f3be53dd1 Mon Sep 17 00:00:00 2001 From: philipp Date: Thu, 7 Jun 2018 10:17:45 +0200 Subject: [PATCH] Update layer_result_generators.py Changed "output" to "outputs" and "input" to "inputs". Prevent "UserWarning: Update your `Model` call to the Keras 2 API: `Model(inputs=Tensor("co..., outputs=Tensor("ac...)`" --- quiver_engine/layer_result_generators.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/quiver_engine/layer_result_generators.py b/quiver_engine/layer_result_generators.py index ceffabe..2e8446b 100644 --- a/quiver_engine/layer_result_generators.py +++ b/quiver_engine/layer_result_generators.py @@ -3,6 +3,6 @@ def get_outputs_generator(model, layer_name): return Model( - input=model.input, - output=model.get_layer(layer_name).output + inputs=model.input, + outputs=model.get_layer(layer_name).output ).predict