Skip to main content


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

Supported Amazon SageMaker models

Updated on May 17, 2024

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
TensorFlowCSVCSV
XGBoostCSVJSON
K-meansCSVJSON
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.

Pega.com is not optimized for Internet Explorer. For the optimal experience, please use:

Close Deprecation Notice
Contact us