aws / aws/sagemaker-scikit-learn-container

Add support for transform_fn

Open
#40 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
188
Forks
121
PR merge metrics
No merged PRs in 30d

Description

At the moment, you can only provide 'input_fn', 'predict_fn' and 'output_fn' which limits flexibility. I have a use case that requires accept_type to be used by the predict_fn, which is not possible in this structure. MXNet containers have the option to use 'transform_fn'.

I think this line limits the container to 'input_fn', 'predict_fn' and 'output_fn' :
https://github.com/aws/sagemaker-scikit-learn-container/blob/master/src/sagemaker_sklearn_container/serving.py#L90

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.