Skip to main content

         This documentation site is for previous versions. Visit our new documentation site for current releases.      

This content has been archived and is no longer being updated.

Links may not function; however, this content may be relevant to outdated versions of the product.

Supported Amazon SageMaker models

Updated on March 11, 2021

Learn more about the Amazon SageMaker models to which you can connect in Prediction Studio.

Supported Amazon SageMaker models

You can connect to Amazon SageMaker models that use the following algorithms:

  • TensorFlow
  • XGBoost
  • K-means
  • K-nearest neighbors
  • Linear learner
  • Random cut forest

You can also connect to an Amazon SageMaker model that uses a custom algorithm. To connect to a custom model, configure the Amazon SageMaker docker container. For more information, see the Amazon Web Services documentation.

Supported input and output formats for Amazon SageMaker models

The supported input and output formats depend on the model algorithm. Consult the following table to learn more about the supported input and output formats for supported models:

Supported input and output formats for Amazon SageMaker models

Model algorithmSupported input formatSupported output format
K-nearest neighborsCSVJSON
Linear learnerCSVJSON
Random cut forestCSVJSON

Have a question? Get answers now.

Visit the Support Center to ask questions, engage in discussions, share ideas, and help others.

Did you find this content helpful?

Want to help us improve this content?

We'd prefer it if you saw us at our best. is not optimized for Internet Explorer. For the optimal experience, please use:

Close Deprecation Notice
Contact us