Skip to content

Commit 6915606

Browse files
AmirS2laurenyu
authored andcommitted
doc: Update sklearn default predict_fn (#1039)
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
1 parent 9661e29 commit 6915606

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

doc/using_sklearn.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -381,7 +381,7 @@ The ``predict_fn`` function has the following signature:
381381
Where ``input_object`` is the object returned from ``input_fn`` and
382382
``model`` is the model loaded by ``model_fn``.
383383

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

0 commit comments

Comments
 (0)