Merge pull request #73 from jacquelinegarrahan/fix-keras-return

Check in corrected Keras model
slaclab · Oct 12, 2022 · 4744583 · 4744583
2 parents ac235a5 + c978dae
commit 4744583
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lume_model/keras/__init__.py b/lume_model/keras/__init__.py
@@ -80,8 +80,12 @@ def evaluate(self, input_variables: Dict[str, InputVariable]) -> Dict[str, Outpu
         """
         self.input_variables = input_variables
 
+        input_dict = {var.name: var.value for var in input_variables.values()}
+
         # converts from input_dict -> formatted input
-        formatted_input = self.format_input(input_dictionary)
+        formatted_input = self.format_input(input_dict)
+
+        print(formatted_input)
 
         # call prediction in threadsafe manner
         model_output = self._model.predict(formatted_input)