The machine learning client library that is used for interacting with Snowflake to build machine learning solutions.
Project description
Snowpark ML
Snowpark ML is a set of tools including SDKs and underlying infrastructure to build and deploy machine learning models. With Snowpark ML, you can pre-process data, train, manage and deploy ML models all within Snowflake, using a single SDK, and benefit from Snowflake’s proven performance, scalability, stability and governance at every stage of the Machine Learning workflow.
Key Components of Snowpark ML
The Snowpark ML Python SDK provides a number of APIs to support each stage of an end-to-end Machine Learning development and deployment process, and includes two key components.
Snowpark ML Development
Snowpark ML Development provides a collection of python APIs enabling efficient ML model development directly in Snowflake:
-
Modeling API (
snowflake.ml.modeling
) for data preprocessing, feature engineering and model training in Snowflake. This includes thesnowflake.ml.modeling.preprocessing
module for scalable data transformations on large data sets utilizing the compute resources of underlying Snowpark Optimized High Memory Warehouses, and a large collection of ML model development classes based on sklearn, xgboost, and lightgbm. -
Framework Connectors: Optimized, secure and performant data provisioning for Pytorch and Tensorflow frameworks in their native data loader formats.
-
FileSet API: FileSet provides a Python fsspec-compliant API for materializing data into a Snowflake internal stage from a query or Snowpark Dataframe along with a number of convenience APIs.
Snowpark Model Management [Public Preview]
Snowpark Model Management complements the Snowpark ML Development API, and provides model management capabilities along with integrated deployment into Snowflake. Currently, the API consists of:
- Registry: A python API for managing models within Snowflake which also supports deployment of ML models into Snowflake as native MODEL object running with Snowflake Warehouse.
Getting started
Have your Snowflake account ready
If you don't have a Snowflake account yet, you can sign up for a 30-day free trial account.
Installation
Follow the installation instructions in the Snowflake documentation.
Python versions 3.9 to 3.11 are supported. You can use miniconda or anaconda to create a Conda environment (recommended), or virtualenv to create a virtual environment.
Conda channels
The Snowflake Conda Channel contains the official snowpark ML package releases.
The recommended approach is to install snowflake-ml-python
this conda channel:
conda install \
-c https://repo.anaconda.com/pkgs/snowflake \
--override-channels \
snowflake-ml-python
See the developer guide for installation instructions.
The latest version of the snowpark-ml-python
package is also published in a conda channel in this repository. Package versions
in this channel may not yet be present in the official Snowflake conda channel.
Install snowflake-ml-python
from this channel with the following (being sure to replace <version_specifier>
with the
desired version, e.g. 1.0.10
):
conda install \
-c https://raw.githubusercontent.com/snowflakedb/snowflake-ml-python/conda/releases/ \
-c https://repo.anaconda.com/pkgs/snowflake \
--override-channels \
snowflake-ml-python==<version_specifier>
Note that until a snowflake-ml-python
package version is available in the official Snowflake conda channel, there may
be compatibility issues. Server-side functionality that snowflake-ml-python
depends on may not yet be released.
Release History
1.7.1
Bug Fixes
- Registry: Null value is now allowed in the dataframe used in model signature inference. Null values will be ignored and others will be used to infer the signature.
- Registry: Pandas Extension DTypes (
pandas.StringDType()
,pandas.BooleanDType()
, etc.) are now supported in model signature inference. - Registry: Null value is now allowed in the dataframe used to predict.
- Data: Fix missing
snowflake.ml.data.*
module exports in wheel - Dataset: Fix missing
snowflake.ml.dataset.*
module exports in wheel. - Registry: Fix the issue that
tf_keras.Model
is not recognized as keras model when logging.
Behavior Changes
New Features
- Registry: Option to
enable_monitoring
set to False by default. This will gate access to preview features of Model Monitoring. - Model Monitoring:
show_model_monitors
Registry method. This feature is still in Private Preview. - Registry: Support
pd.Series
in input and output data. - Model Monitoring:
add_monitor
Registry method. This feature is still in Private Preview. - Model Monitoring:
resume
andsuspend
ModelMonitor. This feature is still in Private Preview. - Model Monitoring:
get_monitor
Registry method. This feature is still in Private Preview. - Model Monitoring:
delete_monitor
Registry method. This feature is still in Private Preview.
1.7.0 (10-22-2024)
Behavior Change
- Generic: Require python >= 3.9.
- Data Connector: Update
to_torch_dataset
andto_torch_datapipe
to add a dimension for scalar data. This allows for more seamless integration with PyTorchDataLoader
, which creates batches by stacking inputs of each batch.
Examples:
ds = connector.to_torch_dataset(shuffle=False, batch_size=3)
-
Input: "col1": [10, 11, 12]
- Previous batch: array([10., 11., 12.]) with shape (3,)
- New batch: array([[10.], [11.], [12.]]) with shape (3, 1)
-
Input: "col2": [[0, 100], [1, 110], [2, 200]]
- Previous batch: array([[ 0, 100], [ 1, 110], [ 2, 200]]) with shape (3,2)
- New batch: No change
-
Model Registry: External access integrations are optional when creating a model inference service in Snowflake >= 8.40.0.
-
Model Registry: Deprecate
build_external_access_integration
withbuild_external_access_integrations
inModelVersion.create_service()
.
Bug Fixes
- Registry: Updated
log_model
API to accept both signature and sample_input_data parameters. - Feature Store: ExampleHelper uses fully qualified path for table name. change weather features aggregation from 1d to 1h.
- Data Connector: Return numpy array with appropriate object type instead of list for multi-dimensional
data from
to_torch_dataset
andto_torch_datapipe
- Model explainability: Incompatibility between SHAP 0.42.1 and XGB 2.1.1 resolved by using latest SHAP 0.46.0.
New Features
- Registry: Provide pass keyworded variable length of arguments to class ModelContext. Example usage:
mc = custom_model.ModelContext(
config = 'local_model_dir/config.json',
m1 = model1
)
class ExamplePipelineModel(custom_model.CustomModel):
def __init__(self, context: custom_model.ModelContext) -> None:
super().__init__(context)
v = open(self.context['config']).read()
self.bias = json.loads(v)['bias']
@custom_model.inference_api
def predict(self, input: pd.DataFrame) -> pd.DataFrame:
model_output = self.context['m1'].predict(input)
return pd.DataFrame({'output': model_output + self.bias})
- Model Development: Upgrade scikit-learn in UDTF backend for log_loss metric. As a result,
eps
argument is now ignored. - Data Connector: Add the option of passing a
None
sized batch toto_torch_dataset
for better interoperability with PyTorch DataLoader. - Model Registry: Support pandas.CategoricalDtype
- Registry: It is now possible to pass
signatures
andsample_input_data
at the same time to capture background data from explainablity and data lineage.
1.6.4 (2024-10-17)
Bug Fixes
- Registry: Fix an issue that leads to incident when using
ModelVersion.run
with service.
1.6.3 (2024-10-07)
- Model Registry (PrPr) has been removed.
Bug Fixes
- Registry: Fix a bug that when package whose name does not follow PEP-508 is provided when logging the model, an unexpected normalization is happening.
- Registry: Fix
not a valid remote uri
error when logging mlflow models. - Registry: Fix a bug that
ModelVersion.run
is called in a nested way. - Registry: Fix an issue that leads to
log_model
failure when local package version contains parts other than base version. - Fix issue where
sample_weights
were not being applied to search estimators. - Model explainability: Fix bug which creates explain as a function instead of table function when enabling by default.
- Model explainability: Update lightgbm binary classification to return non-json values, from customer feedback.
New Features
- Data: Improve
DataConnector.to_pandas()
performance when loading from Snowpark DataFrames. - Model Registry: Allow users to set a model task while using
log_model
. - Feature Store: FeatureView supports ON_CREATE or ON_SCHEDULE initialize mode.
1.6.2 (2024-09-04)
Bug Fixes
-
Modeling: Support XGBoost version that is larger than 2.
-
Data: Fix multiple epoch iteration over
DataConnector.to_torch_datapipe()
DataPipes. -
Generic: Fix a bug that when an invalid name is provided to argument where fully qualified name is expected, it will be parsed wrongly. Now it raises an exception correctly.
-
Model Explainability: Handle explanations for multiclass XGBoost classification models
-
Model Explainability: Workarounds and better error handling for XGB>2.1.0 not working with SHAP==0.42.1
New Features
- Data: Add top-level exports for
DataConnector
andDataSource
tosnowflake.ml.data
. - Data: Add native batching support via
batch_size
anddrop_last_batch
arguments toDataConnector.to_torch_dataset()
- Feature Store: update_feature_view() supports taking feature view object as argument.
1.6.1 (2024-08-12)
Bug Fixes
- Feature Store: Support large metadata blob when generating dataset
- Feature Store: Added a hidden knob in FeatureView as kargs for setting customized refresh_mode
- Registry: Fix an error message in Model Version
run
whenfunction_name
is not mentioned and model has multiple target methods. - Cortex inference: snowflake.cortex.Complete now only uses the REST API for streaming and the use_rest_api_experimental is no longer needed.
- Feature Store: Add a new API: FeatureView.list_columns() which list all column information.
- Data: Fix
DataFrame
ingestion withArrowIngestor
.
New Features
- Enable
set_params
to set the parameters of the underlying sklearn estimator, if the snowflake-ml model has been fit. - Data: Add
snowflake.ml.data.ingestor_utils
module with utility functions helpful forDataIngestor
implementations. - Data: Add new
to_torch_dataset()
connector toDataConnector
to replace deprecated DataPipe. - Registry: Option to
enable_explainability
set to True by default for XGBoost, LightGBM and CatBoost as PuPr feature. - Registry: Option to
enable_explainability
when registering SHAP supported sklearn models.
1.6.0 (2024-07-29)
Bug Fixes
- Modeling:
SimpleImputer
can impute integer columns with integer values. - Registry: Fix an issue when providing a pandas Dataframe whose index is not starting from 0 as the input to
the
ModelVersion.run
.
New Features
- Feature Store: Add overloads to APIs accept both object and name/version. Impacted APIs include read_feature_view(), refresh_feature_view(), get_refresh_history(), resume_feature_view(), suspend_feature_view(), delete_feature_view().
- Feature Store: Add docstring inline examples for all public APIs.
- Feature Store: Add new utility class
ExampleHelper
to help with load source data to simplify public notebooks. - Registry: Option to
enable_explainability
when registering XGBoost models as a pre-PuPr feature. - Feature Store: add new API
update_entity()
. - Registry: Option to
enable_explainability
when registering Catboost models as a pre-PuPr feature. - Feature Store: Add new argument warehouse to FeatureView constructor to overwrite the default warehouse. Also add a new column 'warehouse' to the output of list_feature_views().
- Registry: Add support for logging model from a model version.
- Modeling: Distributed Hyperparameter Optimization now announce GA refresh version. The latest memory efficient version
will not have the 10GB training limitation for dataset any more. To turn off, please run
from snowflake.ml.modeling._internal.snowpark_implementations import ( distributed_hpo_trainer, ) distributed_hpo_trainer.ENABLE_EFFICIENT_MEMORY_USAGE = False
- Registry: Option to
enable_explainability
when registering LightGBM models as a pre-PuPr feature. - Data: Add new
snowflake.ml.data
preview module which contains data reading utilities likeDataConnector
DataConnector
provides efficient connectors from SnowparkDataFrame
and Snowpark MLDataset
to external frameworks like PyTorch, TensorFlow, and Pandas. CreateDataConnector
instances using the classmethod constructorsDataConnector.from_dataset()
andDataConnector.from_dataframe()
.
- Data: Add new
DataConnector.from_sources()
classmethod constructor for constructing fromDataSource
objects. - Data: Add new
ingestor_class
arg toDataConnector
classmethod constructors for easierDataIngestor
injection. - Dataset:
DatasetReader
now subclasses newDataConnector
class.- Add optional
limit
arg toDatasetReader.to_pandas()
- Add optional
Behavior Changes
- Feature Store: change some positional parameters to keyword arguments in following APIs:
- Entity(): desc.
- FeatureView(): timestamp_col, refresh_freq, desc.
- FeatureStore(): creation_mode.
- update_entity(): desc.
- register_feature_view(): block, overwrite.
- list_feature_views(): entity_name, feature_view_name.
- get_refresh_history(): verbose.
- retrieve_feature_values(): spine_timestamp_col, exclude_columns, include_feature_view_timestamp_col.
- generate_training_set(): save_as, spine_timestamp_col, spine_label_cols, exclude_columns, include_feature_view_timestamp_col.
- generate_dataset(): version, spine_timestamp_col, spine_label_cols, exclude_columns, include_feature_view_timestamp_col, desc, output_type.
1.5.4 (2024-07-11)
Bug Fixes
- Model Registry (PrPr): Fix 401 Unauthorized issue when deploying model to SPCS.
- Feature Store: Downgrades exceptions to warnings for few property setters in feature view. Now you can set desc, refresh_freq and warehouse for draft feature views.
- Modeling: Fix an issue with calling
OrdinalEncoder
withcategories
as a dictionary and a pandas DataFrame - Modeling: Fix an issue with calling
OneHotEncoder
withcategories
as a dictionary and a pandas DataFrame
New Features
- Registry: Allow overriding
device_map
anddevice
when loading huggingface pipeline models. - Registry: Add
set_alias
method toModelVersion
instance to set an alias to model version. - Registry: Add
unset_alias
method toModelVersion
instance to unset an alias to model version. - Registry: Add
partitioned_inference_api
allowing users to create partitioned inference functions in registered models. Enable model inference methods with table functions with vectorized process methods in registered models. - Feature Store: add 3 more columns: refresh_freq, refresh_mode and scheduling_state to the result of
list_feature_views()
. - Feature Store:
update_feature_view()
supports updating description. - Feature Store: add new API
refresh_feature_view()
. - Feature Store: add new API
get_refresh_history()
. - Feature Store: Add
generate_training_set()
API for generating table-backed feature snapshots. - Feature Store: Add
DeprecationWarning
forgenerate_dataset(..., output_type="table")
. - Feature Store:
update_feature_view()
supports updating description. - Feature Store: add new API
refresh_feature_view()
. - Feature Store: add new API
get_refresh_history()
. - Model Development: OrdinalEncoder supports a list of array-likes for
categories
argument. - Model Development: OneHotEncoder supports a list of array-likes for
categories
argument.
1.5.3 (06-17-2024)
Bug Fixes
- Modeling: Fix an issue causing lineage information to be missing for
Pipeline
,GridSearchCV
,SimpleImputer
, andRandomizedSearchCV
- Registry: Fix an issue that leads to incorrect result when using pandas Dataframe with over 100, 000 rows as the input
of
ModelVersion.run
method in Stored Procedure.
New Features
- Registry: Add support for TIMESTAMP_NTZ model signature data type, allowing timestamp input and output.
- Dataset: Add
DatasetVersion.label_cols
andDatasetVersion.exclude_cols
properties.
1.5.2 (06-10-2024)
Bug Fixes
- Registry: Fix an issue that leads to unable to log model in store procedure.
- Modeling: Quick fix
import snowflake.ml.modeling.parameters.enable_anonymous_sproc
cannot be imported due to package dependency error.
Behavior Changes
New Features
1.5.1 (05-22-2024)
Bug Fixes
- Dataset: Fix
snowflake.connector.errors.DataError: Query Result did not match expected number of rows
when accessing DatasetVersion properties when case insensitiveSHOW VERSIONS IN DATASET
check matches multiple version names. - Dataset: Fix bug in SnowFS bulk file read when used with DuckDB
- Registry: Fixed a bug when loading old models.
- Lineage: Fix Dataset source lineage propagation through
snowpark.DataFrame
transformations
Behavior Changes
- Feature Store: convert clear() into a private function. Also make it deletes feature views and entities only.
- Feature Store: Use NULL as default value for timestamp tag value.
New Features
- Feature Store: Added new
snowflake.ml.feature_store.setup_feature_store()
API to assist Feature Store RBAC setup. - Feature Store: Add
output_type
argument toFeatureStore.generate_dataset()
to allow generating data snapshots as Datasets or Tables. - Registry:
log_model
,get_model
,delete_model
now supports fully qualified name. - Modeling: Supports anonymous stored procedure during fit calls so that modeling would not require sufficient
permissions to operate on schema. Please call
import snowflake.ml.modeling.parameters.enable_anonymous_sproc # noqa: F401
1.5.0 (05-01-2024)
Bug Fixes
- Registry: Fix invalid parameter 'SHOW_MODEL_DETAILS_IN_SHOW_VERSIONS_IN_MODEL' error.
Behavior Changes
- Model Development: The behavior of
fit_transform
for all estimators is changed. Firstly, it will cover all the estimator that contains this function, secondly, the output would be the union of pandas DataFrame and snowpark DataFrame.
Model Registry (PrPr)
snowflake.ml.registry.artifact
and related snowflake.ml.model_registry.ModelRegistry
APIs have been removed.
- Removed
snowflake.ml.registry.artifact
module. - Removed
ModelRegistry.log_artifact()
,ModelRegistry.list_artifacts()
,ModelRegistry.get_artifact()
- Removed
artifacts
argument fromModelRegistry.log_model()
Dataset (PrPr)
snowflake.ml.dataset.Dataset
has been redesigned to be backed by Snowflake Dataset entities.
- New
Dataset
s can be created withDataset.create()
and existingDataset
s may be loaded withDataset.load()
. Dataset
s now maintain an immutableselected_version
state. TheDataset.create_version()
andDataset.load_version()
APIs return newDataset
objects with the requestedselected_version
state.- Added
dataset.create_from_dataframe()
anddataset.load_dataset()
convenience APIs as a shortcut to creating and loadingDataset
s with a pre-selected version. Dataset.materialized_table
andDataset.snapshot_table
no longer exist withDataset.fully_qualified_name
as the closest equivalent.Dataset.df
no longer exists. Instead, useDatasetReader.read.to_snowpark_dataframe()
.Dataset.owner
has been moved toDataset.selected_version.owner
Dataset.desc
has been moved toDatasetVersion.selected_version.comment
Dataset.timestamp_col
,Dataset.label_cols
,Dataset.feature_store_metadata
, andDataset.schema_version
have been removed.
Feature Store (PrPr)
-
FeatureStore.generate_dataset
argument list has been changed to match the newsnowflake.ml.dataset.Dataset
definitionmaterialized_table
has been removed and replaced withname
andversion
.name
moved to first positional argumentsave_mode
has been removed asmerge
behavior is no longer supported. The new behavior is alwayserrorifexists
.
-
Change feature view version type from str to
FeatureViewVersion
. It is a restricted string literal. -
Remove as_dataframe arg from FeatureStore.list_feature_views(), now always returns result as DataFrame.
-
Combines few metadata tags into a new tag: SNOWML_FEATURE_VIEW_METADATA. This will make previously created feature views not readable by new SDK.
New Features
- Registry: Add
export
method toModelVersion
instance to export model files. - Registry: Add
load
method toModelVersion
instance to load the underlying object from the model. - Registry: Add
Model.rename
method toModel
instance to rename or move a model.
Dataset (PrPr)
- Added Snowpark DataFrame integration using
Dataset.read.to_snowpark_dataframe()
- Added Pandas DataFrame integration using
Dataset.read.to_pandas()
- Added PyTorch and TensorFlow integrations using
Dataset.read.to_torch_datapipe()
andDataset.read.to_tf_dataset()
respectively. - Added
fsspec
style file integration usingDataset.read.files()
andDataset.read.filesystem()
Feature Store
- use new tag_reference_internal to speed up metadata lookup.
1.4.1 (2024-04-18)
New Features
- Registry: Add support for
catboost
model (catboost.CatBoostClassifier
,catboost.CatBoostRegressor
). - Registry: Add support for
lightgbm
model (lightgbm.Booster
,lightgbm.LightGBMClassifier
,lightgbm.LightGBMRegressor
).
Bug Fixes
- Registry: Fix a bug that leads to relax_version option is not working.
Behavior changes
- Feature Store: update_feature_view takes refresh_freq and warehouse as argument.
1.4.0 (2024-04-08)
Bug Fixes
- Registry: Fix a bug when multiple models are being called from the same query, models other than the first one will have incorrect result. This fix only works for newly logged model.
- Modeling: When registering a model, only method(s) that is mentioned in
save_model
would be added to model signature in SnowML models. - Modeling: Fix a bug that when n_jobs is not 1, model cannot execute methods such as predict, predict_log_proba, and other batch inference methods. The n_jobs would automatically set to 1 because vectorized udf currently doesn't support joblib parallel backend.
- Modeling: Fix a bug that batch inference methods cannot infer the datatype when the first row of data contains NULL.
- Modeling: Matches Distributed HPO output column names with the snowflake identifier.
- Modeling: Relax package versions for all Distributed HPO methods if the installed version is not available in the Snowflake conda channel
- Modeling: Add sklearn as required dependency for LightGBM package.
Behavior Changes
- Registry:
apply
method is no longer by default logged when logging a xgboost model. If that is required, it could be specified manually when logging the model bylog_model(..., options={"target_methods": ["apply", ...]})
. - Feature Store: register_entity returns an entity object.
- Feature Store: register_feature_view
block=true
becomes default.
New Features
- Registry: Add support for
sentence-transformers
model (sentence_transformers.SentenceTransformer
). - Registry: Now version name is no longer required when logging a model. If not provided, a random human readable ID will be generated.
1.3.1 (2024-03-21)
New Features
- FileSet:
snowflake.ml.fileset.sfcfs.SFFileSystem
can now be used in UDFs and stored procedures.
1.3.0 (2024-03-12)
Bug Fixes
- Registry: Fix a bug that leads to module in
code_paths
whenlog_model
cannot be correctly imported. - Registry: Fix incorrect error message when validating input Snowpark DataFrame with array feature.
- Model Registry: Fix an issue when deploying a model to SPCS that some files do not have proper permission.
- Model Development: Relax package versions for all inference methods if the installed version is not available in the Snowflake conda channel
Behavior Changes
- Registry: When running the method of a model, the value range based input validation to avoid input from overflowing
is now optional rather than enforced, this should improve the performance and should not lead to problem for most
kinds of model. If you want to enable this check as previous, specify
strict_input_validation=True
when callingrun
. - Registry: By default
relax_version=True
when logging a model instead of using the specific local dependency versions. This improves dependency versioning by using versions available in Snowflake. To switch back to the previous behavior and use specific local dependency versions, specifyrelax_version=False
when callinglog_model
. - Model Development: The behavior of
fit_predict
for all estimators is changed. Firstly, it will cover all the estimator that contains this function, secondly, the output would be the union of pandas DataFrame and snowpark DataFrame.
New Features
- FileSet:
snowflake.ml.fileset.sfcfs.SFFileSystem
can now be serialized withpickle
.
1.2.3 (2024-02-26)
Bug Fixes
- Registry: Now when providing Decimal Type column to a DOUBLE or FLOAT feature will not error out but auto cast with warnings.
- Registry: Improve the error message when specifying currently unsupported
pip_requirements
argument. - Model Development: Fix precision_recall_fscore_support incorrect results when
average="samples"
. - Model Registry: Fix an issue that leads to description, metrics or tags are not correctly returned in newly created Model Registry (PrPr) due to Snowflake BCR 2024_01
Behavior Changes
- Feature Store:
FeatureStore.suspend_feature_view
andFeatureStore.resume_feature_view
doesn't mutate input feature view argument any more. The updated status only reflected in the returned feature view object.
New Features
- Model Development: support
score_samples
method for all the classes, including Pipeline, GridSearchCV, RandomizedSearchCV, PCA, IsolationForest, ... - Registry: Support deleting a version of a model.
1.2.2 (2024-02-13)
New Features
- Model Registry: Support providing external access integrations when deploying a model to SPCS. This will help and be
required to make sure the deploying process work as long as SPCS will by default deny all network connections. The
following endpoints must be allowed to make deployment work: docker.com:80, docker.com:443, anaconda.com:80,
anaconda.com:443, anaconda.org:80, anaconda.org:443, pypi.org:80, pypi.org:443. If you are using
snowflake.ml.model.models.huggingface_pipeline.HuggingFacePipelineModel
object, the following endpoints are required to be allowed: huggingface.com:80, huggingface.com:443, huggingface.co:80, huggingface.co:443.
1.2.1 (2024-01-25)
New Features
- Model Development: Infers output column data type for transformers when possible.
- Registry:
relax_version
option is available in theoptions
argument when logging the model.
1.2.0 (2024-01-11)
Bug Fixes
- Model Registry: Fix "XGBoost version not compiled with GPU support" error when running CPU inference against open-source XGBoost models deployed to SPCS.
- Model Registry: Fix model deployment to SPCS on Windows machines.
New Features
- Model Development: Introduced XGBoost external memory training feature. This feature enables training XGBoost models on large datasets that don't fit into memory.
- Registry: New Registry class named
snowflake.ml.registry.Registry
providing similar APIs as the old one but works with new MODEL object in Snowflake SQL. Also, we are providingsnowflake.ml.model.Model
andsnowflake.ml.model.ModelVersion
to represent a model and a specific version of a model. - Model Development: Add support for
fit_predict
method inAgglomerativeClustering
,DBSCAN
, andOPTICS
classes; - Model Development: Add support for
fit_transform
method inMDS
,SpectralEmbedding
andTSNE
class.
Additional Notes
- Model Registry: The
snowflake.ml.registry.model_registry.ModelRegistry
has been deprecated starting from version 1.2.0. It will stay in the Private Preview phase. For future implementations, kindly utilizesnowflake.ml.registry.Registry
, except when specifically required. The old model registry will be removed once all its primary functionalities are fully integrated into the new registry.
1.1.2 (2023-12-18)
Bug Fixes
- Generic: Fix the issue that stack trace is hidden by telemetry unexpectedly.
- Model Development: Execute model signature inference without materializing full dataframe in memory.
- Model Registry: Fix occasional 'snowflake-ml-python library does not exist' error when deploying to SPCS.
Behavior Changes
- Model Registry: When calling
predict
with Snowpark DataFrame, both inferred or normalized column names are accepted. - Model Registry: When logging a Snowpark ML Modeling Model, sample input data or manually provided signature will be ignored since they are not necessary.
New Features
- Model Development: SQL implementation of binary
precision_score
metric.
1.1.1 (2023-12-05)
Bug Fixes
- Model Registry: The
predict
target method on registered models is now compatible with unsupervised estimators. - Model Development: Fix confusion_matrix incorrect results when the row number cannot be divided by the batch size.
New Features
- Introduced passthrough_col param in Modeling API. This new param is helpful in scenarios requiring automatic input_cols inference, but need to avoid using specific columns, like index columns, during training or inference.
1.1.0 (2023-12-01)
Bug Fixes
- Model Registry: Fix panda dataframe input not handling first row properly.
- Model Development: OrdinalEncoder and LabelEncoder output_columns do not need to be valid snowflake identifiers. They would previously be excluded if the normalized name did not match the name specified in output_columns.
New Features
- Model Registry: Add support for invoking public endpoint on SPCS service, by providing a "enable_ingress" SPCS deployment option.
- Model Development: Add support for distributed HPO - GridSearchCV and RandomizedSearchCV execution will be distributed on multi-node warehouses.
1.0.12 (2023-11-13)
Bug Fixes
- Model Registry: Fix regression issue that container logging is not shown during model deployment to SPCS.
- Model Development: Enhance the column capacity of OrdinalEncoder.
- Model Registry: Fix unbound
batch_size
error when deploying a model other than Hugging Face Pipeline and LLM with GPU on SPCS.
Behavior Changes
- Model Registry: Raise early error when deploying to SPCS with db/schema that starts with underscore.
- Model Registry:
conda-forge
channel is now automatically added to channel lists when deploying to SPCS. - Model Registry:
relax_version
will not strip all version specifier, instead it will relax==x.y.z
specifier to>=x.y,<(x+1)
. - Model Registry: Python with different patchlevel but the same major and minor will not result a warning when loading the model via Model Registry and would be considered to use when deploying to SPCS.
- Model Registry: When logging a
snowflake.ml.model.models.huggingface_pipeline.HuggingFacePipelineModel
object, versions of local installed libraries won't be picked as dependencies of models, instead it will pick up some pre- defined dependencies to improve user experience.
New Features
- Model Registry: Enable best-effort SPCS job/service log streaming when logging level is set to INFO.
1.0.11 (2023-10-27)
New Features
- Model Registry: Add log_artifact() public method.
- Model Development: Add support for
kneighbors
.
Behavior Changes
- Model Registry: Change log_model() argument from TrainingDataset to List of Artifact.
- Model Registry: Change get_training_dataset() to get_artifact().
Bug Fixes
- Model Development: Fix support for XGBoost and LightGBM models using SKLearn Grid Search and Randomized Search model selectors.
- Model Development: DecimalType is now supported as a DataType.
- Model Development: Fix metrics compatibility with Snowpark Dataframes that use Snowflake identifiers
- Model Registry: Resolve 'delete_deployment' not deleting the SPCS service in certain cases.
1.0.10 (2023-10-13)
Behavior Changes
- Model Development: precision_score, recall_score, f1_score, fbeta_score, precision_recall_fscore_support, mean_absolute_error, mean_squared_error, and mean_absolute_percentage_error metric calculations are now distributed.
- Model Registry:
deploy
will now returnDeployment
for deployment information.
New Features
- Model Registry: When the model signature is auto-inferred, it will be printed to the log for reference.
- Model Registry: For SPCS deployment,
Deployment
details will containsimage_name
,service_spec
andservice_function_sql
.
Bug Fixes
- Model Development: Fix an issue that leading to UTF-8 decoding errors when using modeling modules on Windows.
- Model Development: Fix an issue that alias definitions cause
SnowparkSQLUnexpectedAliasException
in inference. - Model Registry: Fix an issue that signature inference could be incorrect when using Snowpark DataFrame as sample input.
- Model Registry: Fix too strict data type validation when predicting. Now, for example, if you have a INT8 type feature in the signature, if providing a INT64 dataframe but all values are within the range, it would not fail.
1.0.9 (2023-09-28)
Behavior Changes
- Model Development: log_loss metric calculation is now distributed.
Bug Fixes
- Model Registry: Fix an issue that building images fails with specific docker setup.
- Model Registry: Fix an issue that unable to embed local ML library when the library is imported by
zipimport
. - Model Registry: Fix out-of-date doc about
platform
argument in thedeploy
function. - Model Registry: Fix an issue that unable to deploy a GPU-trained PyTorch model to a platform where GPU is not available.
1.0.8 (2023-09-15)
Bug Fixes
- Model Development: Ordinal encoder can be used with mixed input column types.
- Model Development: Fix an issue when the sklearn default value is
np.nan
. - Model Registry: Fix an issue that incorrect docker executable is used when building images.
- Model Registry: Fix an issue that specifying
token
argument when usingsnowflake.ml.model.models.huggingface_pipeline.HuggingFacePipelineModel
withtransformers < 4.32.0
is not effective. - Model Registry: Fix an issue that incorrect system function call is used when deploying to SPCS.
- Model Registry: Fix an issue when using a
transformers.pipeline
that does not have atokenizer
. - Model Registry: Fix incorrectly-inferred image repository name during model deployment to SPCS.
- Model Registry: Fix GPU resource retention issue caused by failed or stuck previous deployments in SPCS.
1.0.7 (2023-09-05)
Bug Fixes
- Model Development & Model Registry: Fix an error related to
pandas.io.json.json_normalize
. - Allow disabling telemetry.
1.0.6 (2023-09-01)
New Features
- Model Registry: add
create_if_not_exists
parameter in constructor. - Model Registry: Added get_or_create_model_registry API.
- Model Registry: Added support for using GPU inference when deploying XGBoost (
xgboost.XGBModel
andxgboost.Booster
), PyTorch (torch.nn.Module
andtorch.jit.ScriptModule
) and TensorFlow (tensorflow.Module
andtensorflow.keras.Model
) models to Snowpark Container Services. - Model Registry: When inferring model signature,
Sequence
of built-in types,Sequence
ofnumpy.ndarray
,Sequence
oftorch.Tensor
,Sequence
oftensorflow.Tensor
andSequence
oftensorflow.Tensor
can be used instead of onlyList
of them. - Model Registry: Added
get_training_dataset
API. - Model Development: Size of metrics result can exceed previous 8MB limit.
- Model Registry: Added support save/load/deploy HuggingFace pipeline object (
transformers.Pipeline
) and our wrapper (snowflake.ml.model.models.huggingface_pipeline.HuggingFacePipelineModel
) to it. Using the wrapper to specify configurations and the model for the pipeline will be loaded dynamically when deploying. Currently, following tasks are supported to log without manually specifying model signatures:- "conversational"
- "fill-mask"
- "question-answering"
- "summarization"
- "table-question-answering"
- "text2text-generation"
- "text-classification" (alias "sentiment-analysis" available)
- "text-generation"
- "token-classification" (alias "ner" available)
- "translation"
- "translation_xx_to_yy"
- "zero-shot-classification"
Bug Fixes
- Model Development: Fixed a bug when using simple imputer with numpy >= 1.25.
- Model Development: Fixed a bug when inferring the type of label columns.
Behavior Changes
- Model Registry:
log_model()
now return aModelReference
object instead of a model ID. - Model Registry: When deploying a model with 1
target method
only, thetarget_method
argument can be omitted. - Model Registry: When using the snowflake-ml-python with version newer than what is available in Snowflake Anaconda
Channel,
embed_local_ml_library
option will be set asTrue
automatically if not. - Model Registry: When deploying a model to Snowpark Container Services and using GPU, the default value of num_workers will be 1.
- Model Registry:
keep_order
andoutput_with_input_features
in the deploy options have been removed. Now the behavior is controlled by the type of the input when callingmodel.predict()
. If the input is apandas.DataFrame
, the behavior will be the same askeep_order=True
andoutput_with_input_features=False
before. If the input is asnowpark.DataFrame
, the behavior will be the same askeep_order=False
andoutput_with_input_features=True
before. - Model Registry: When logging and deploying PyTorch (
torch.nn.Module
andtorch.jit.ScriptModule
) and TensorFlow (tensorflow.Module
andtensorflow.keras.Model
) models, we no longer accept models whose input is a list of tensor and output is a list of tensors. Instead, now we accept models whose input is 1 or more tensors as positional arguments, and output is a tensor or a tuple of tensors. The input and output dataframe when predicting keep the same as before, that is every column is an array feature and contains a tensor.
1.0.5 (2023-08-17)
New Features
- Model Registry: Added support save/load/deploy xgboost Booster model.
- Model Registry: Added support to get the model name and the model version from model references.
Bug Fixes
- Model Registry: Restore the db/schema back to the session after
create_model_registry()
. - Model Registry: Fixed an issue that the UDF name created when deploying a model is not identical to what is provided and cannot be correctly dropped when deployment getting dropped.
- connection_params.SnowflakeLoginOptions(): Added support for
private_key_path
.
1.0.4 (2023-07-28)
New Features
- Model Registry: Added support save/load/deploy Tensorflow models (
tensorflow.Module
). - Model Registry: Added support save/load/deploy MLFlow PyFunc models (
mlflow.pyfunc.PyFuncModel
). - Model Development: Input dataframes can now be joined against data loaded from staged files.
- Model Development: Added support for non-English languages.
Bug Fixes
- Model Registry: Fix an issue that model dependencies are incorrectly reported as unresolvable on certain platforms.
1.0.3 (2023-07-14)
Behavior Changes
- Model Registry: When predicting a model whose output is a list of NumPy ndarray, the output would not be flattened, instead, every ndarray will act as a feature(column) in the output.
New Features
- Model Registry: Added support save/load/deploy PyTorch models (
torch.nn.Module
andtorch.jit.ScriptModule
).
Bug Fixes
- Model Registry: Fix an issue that when database or schema name provided to
create_model_registry
contains special characters, the model registry cannot be created. - Model Registry: Fix an issue that
get_model_description
returns with additional quotes. - Model Registry: Fix incorrect error message when attempting to remove a unset tag of a model.
- Model Registry: Fix a typo in the default deployment table name.
- Model Registry: Snowpark dataframe for sample input or input for
predict
method that contains a column with SnowflakeNUMBER(precision, scale)
data type wherescale = 0
will not lead to error, and will now correctly recognized asINT64
data type in model signature. - Model Registry: Fix an issue that prevent model logged in the system whose default encoding is not UTF-8 compatible from deploying.
- Model Registry: Added earlier and better error message when any file name in the model or the file name of model itself contains characters that are unable to be encoded using ASCII. It is currently not supported to deploy such a model.
1.0.2 (2023-06-22)
Behavior Changes
- Model Registry: Prohibit non-snowflake-native models from being logged.
- Model Registry:
_use_local_snowml
parameter in options ofdeploy()
has been removed. - Model Registry: A default
False
embed_local_ml_library
parameter has been added to the options oflog_model()
. With this set toFalse
(default), the version of the local snowflake-ml-python library will be recorded and used when deploying the model. With this set toTrue
, local snowflake-ml-python library will be embedded into the logged model, and will be used when you load or deploy the model.
New Features
- Model Registry: A new optional argument named
code_paths
has been added to the arguments oflog_model()
for users to specify additional code paths to be imported when loading and deploying the model. - Model Registry: A new optional argument named
options
has been added to the arguments oflog_model()
to specify any additional options when saving the model. - Model Development: Added metrics:
- d2_absolute_error_score
- d2_pinball_score
- explained_variance_score
- mean_absolute_error
- mean_absolute_percentage_error
- mean_squared_error
Bug Fixes
- Model Development:
accuracy_score()
now works when given label column names are lists of a single value.
1.0.1 (2023-06-16)
Behavior Changes
- Model Development: Changed Metrics APIs to imitate sklearn metrics modules:
accuracy_score()
,confusion_matrix()
,precision_recall_fscore_support()
,precision_score()
methods move from respective modules tometrics.classification
.
- Model Registry: The default table/stage created by the Registry now uses "SYSTEM" as a prefix.
- Model Registry:
get_model_history()
method as been enhanced to include the history of model deployment.
New Features
- Model Registry: A default
False
flag namedreplace_udf
has been added to the options ofdeploy()
. Setting this toTrue
will allow overwrite existing UDF with the same name when deploying. - Model Development: Added metrics:
- f1_score
- fbeta_score
- recall_score
- roc_auc_score
- roc_curve
- log_loss
- precision_recall_curve
- Model Registry: A new argument named
permanent
has been added to the argument ofdeploy()
. Setting this toTrue
allows the creation of a permanent deployment without needing to specify the UDF location. - Model Registry: A new method
list_deployments()
has been added to enumerate all permanent deployments originating from a specific model. - Model Registry: A new method
get_deployment()
has been added to fetch a deployment by its deployment name. - Model Registry: A new method
delete_deployment()
has been added to remove an existing permanent deployment.
1.0.0 (2023-06-09)
Behavior Changes
- Model Registry:
predict()
method moves from Registry to ModelReference. - Model Registry:
_snowml_wheel_path
parameter in options ofdeploy()
, is replaced with_use_local_snowml
with default value ofFalse
. Setting this toTrue
will have the same effect of uploading local SnowML code when executing model in the warehouse. - Model Registry: Removed
id
field fromModelReference
constructor. - Model Development: Preprocessing and Metrics move to the modeling package:
snowflake.ml.modeling.preprocessing
andsnowflake.ml.modeling.metrics
. - Model Development:
get_sklearn_object()
method is renamed toto_sklearn()
,to_xgboost()
, andto_lightgbm()
for respective native models.
New Features
- Added PolynomialFeatures transformer to the snowflake.ml.modeling.preprocessing module.
- Added metrics:
- accuracy_score
- confusion_matrix
- precision_recall_fscore_support
- precision_score
Bug Fixes
- Model Registry: Model version can now be any string (not required to be a valid identifier)
- Model Deployment:
deploy()
&predict()
methods now correctly escapes identifiers
0.3.2 (2023-05-23)
Behavior Changes
- Use cloudpickle to serialize and deserialize models throughout the codebase and removed dependency on joblib.
New Features
- Model Deployment: Added support for snowflake.ml models.
0.3.1 (2023-05-18)
Behavior Changes
- Standardized registry API with following
- Create & open registry taking same set of arguments
- Create & Open can choose schema to use
- Set_tag, set_metric, etc now explicitly calls out arg name as metric_name, tag_name, metric_name, etc.
New Features
- Changes to support python 3.9, 3.10
- Added kBinsDiscretizer
- Support for deployment of XGBoost models & int8 types of data
0.3.0 (2023-05-11)
Behavior Changes
- Big Model Registry Refresh
- Fixed API discrepancies between register_model & log_model.
- Model can be referred by Name + Version (no opaque internal id is required)
New Features
- Model Registry: Added support save/load/deploy SKL & XGB Models
0.2.3 (2023-04-27)
Bug Fixes
- Allow using OneHotEncoder along with sklearn style estimators in a pipeline.
New Features
- Model Registry: Added support for delete_model. Use delete_artifact = False to not delete the underlying model data but just unregister.
0.2.2 (2023-04-11)
New Features
- Initial version of snowflake-ml modeling package.
- Provide support for training most of scikit-learn and xgboost estimators and transformers.
Bug Fixes
- Minor fixes in preprocessing package.
0.2.1 (2023-03-23)
New Features
- New in Preprocessing:
- SimpleImputer
- Covariance Matrix
- Optimization of Ordinal Encoder client computations.
Bug Fixes
- Minor fixes in OneHotEncoder.
0.2.0 (2023-02-27)
New Features
- Model Registry
- PyTorch & Tensorflow connector file generic FileSet API
- New to Preprocessing:
- Binarizer
- Normalizer
- Pearson correlation Matrix
- Optimization in Ordinal Encoder to cache vocabulary in temp tables.
0.1.3 (2023-02-02)
New Features
- Initial version of transformers including:
- Label Encoder
- Max Abs Scaler
- Min Max Scaler
- One Hot Encoder
- Ordinal Encoder
- Robust Scaler
- Standard Scaler
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
File details
Details for the file snowflake_ml_python-1.7.1.tar.gz
.
File metadata
- Download URL: snowflake_ml_python-1.7.1.tar.gz
- Upload date:
- Size: 1.7 MB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/3.8.0 pkginfo/1.10.0 readme-renderer/34.0 requests/2.27.1 requests-toolbelt/1.0.0 urllib3/1.26.20 tqdm/4.64.1 importlib-metadata/4.8.3 keyring/23.4.1 rfc3986/1.5.0 colorama/0.4.5 CPython/3.6.8
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 2c6ce4a3f17736c55176367d87f4d69cbcb73873339c45c6d8bb4b066de16cb3 |
|
MD5 | 6e08829bcd4fcd05d190791d3e56adcb |
|
BLAKE2b-256 | 7c28cce9078ff355fea75b660c16c1054297590ababca437d48f409f1151f702 |
File details
Details for the file snowflake_ml_python-1.7.1-py3-none-any.whl
.
File metadata
- Download URL: snowflake_ml_python-1.7.1-py3-none-any.whl
- Upload date:
- Size: 2.0 MB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/3.8.0 pkginfo/1.10.0 readme-renderer/34.0 requests/2.27.1 requests-toolbelt/1.0.0 urllib3/1.26.20 tqdm/4.64.1 importlib-metadata/4.8.3 keyring/23.4.1 rfc3986/1.5.0 colorama/0.4.5 CPython/3.6.8
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | c469523fb801df3b2ea06b0475b5b9032ae17ccb6e9e2c8db2a304ae7abef64f |
|
MD5 | 2effefc6529db134027ef3c698418338 |
|
BLAKE2b-256 | 86481a466b00165316e0d25b66b332ab99cf5d7147a00505723245221aa1ee6d |