def default_predict_fn()

in src/sagemaker_sklearn_container/handler_service.py [0:0]

• 3 lines of code
• 1 McCabe index (conditional complexity)

        def default_predict_fn(input_data, model):
            """A default predict_fn for Scikit-learn. Calls a model on data deserialized in input_fn.
            Args:
                input_data: input data (Numpy array) for prediction deserialized by input_fn
                model: Scikit-learn model loaded in memory by model_fn
            Returns: a prediction
            """
            output = model.predict(input_data)
            return output