snowflake-ml-python API Reference

The snowflake-ml-python package is a set of tools for creating and working with machine learning models in Python on Snowflake. See the Snowflake ML Developer Guide for more information.

Additional ML APIs are available in snowflake-ml-python when running on Container Runtime for ML. See Snowflake ML Container Runtime Reference (Python) for details on these APIs.

snowflake-ml-python also includes Python APIs for Snowflake Cortex.

Acknowledgements

Portions of the Snowpark ML API Reference are derived from scikit-learn, xgboost, and lightgbm documentation.

scikit-learn Copyright © 2007-2025 The scikit-learn developers. All rights reserved.
xgboost Copyright © 2019 by xgboost contributors.
lightgbm Copyright © Microsoft Corporation.

Container Runtime for ML APIs

The following APIs are available only in the version of snowpark-ml-python available in the Container Runtime for ML, accessible in Snowflake Notebooks running on Snowpark Container Services (SPCS).

Standard APIs

The following APIs are available in both the Container Runtime for ML and in the standalone client version of snowpark-ml-python accessible through conda and pip, in Snowsight Python worksheets, and in Snowflake notebooks running on a warehouse.

Language: English