Merge pull request #5 from badbye/patch-1

Update interface.py
This commit is contained in:
Dawood Khan 2019-07-21 18:24:28 -07:00 committed by GitHub
commit 38f814560a

View File

@ -135,8 +135,7 @@ class Interface:
Method that calls the relevant method of the model object to make a prediction. Method that calls the relevant method of the model object to make a prediction.
:param preprocessed_input: the preprocessed input returned by the input interface :param preprocessed_input: the preprocessed input returned by the input interface
""" """
# print(preprocessed_input.shape)
# print(preprocessed_input.shape)
if self.model_type == "sklearn": if self.model_type == "sklearn":
return self.model_obj.predict(preprocessed_input) return self.model_obj.predict(preprocessed_input)
elif self.model_type == "keras": elif self.model_type == "keras":