Skip to content

Commit

Permalink
doc: Update sklearn default predict_fn (#1039)
Browse files Browse the repository at this point in the history
The current sklearn sagemaker image actually calls `model.predict()`,
not `model()`. Update documentation to reflect this.

Current implemenation: https://github.com/aws/sagemaker-scikit-learn-container/blob/85a994234f0ed46fbf569dd927d3c4825f3445d3/src/sagemaker_sklearn_container/serving.py#L66
  • Loading branch information
AmirS2 authored and laurenyu committed Sep 18, 2019
1 parent 9661e29 commit 6915606
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion doc/using_sklearn.rst
Original file line number Diff line number Diff line change
Expand Up @@ -381,7 +381,7 @@ The ``predict_fn`` function has the following signature:
Where ``input_object`` is the object returned from ``input_fn`` and
``model`` is the model loaded by ``model_fn``.

The default implementation of ``predict_fn`` invokes the loaded model's ``__call__`` function on ``input_object``,
The default implementation of ``predict_fn`` invokes the loaded model's ``predict`` function on ``input_object``,
and returns the resulting value. The return-type should be a NumPy array to be compatible with the default
``output_fn``.

Expand Down

0 comments on commit 6915606

Please sign in to comment.