snowflake-ml-python 1.6.0__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (417) hide show
  1. snowflake_ml_python-1.6.0/CHANGELOG.md +779 -0
  2. snowflake_ml_python-1.6.0/LICENSE.txt +202 -0
  3. snowflake_ml_python-1.6.0/PKG-INFO +1152 -0
  4. snowflake_ml_python-1.6.0/README.md +82 -0
  5. snowflake_ml_python-1.6.0/pyproject.toml +49 -0
  6. snowflake_ml_python-1.6.0/setup.cfg +4 -0
  7. snowflake_ml_python-1.6.0/snowflake/cortex/__init__.py +16 -0
  8. snowflake_ml_python-1.6.0/snowflake/cortex/_classify_text.py +36 -0
  9. snowflake_ml_python-1.6.0/snowflake/cortex/_complete.py +316 -0
  10. snowflake_ml_python-1.6.0/snowflake/cortex/_extract_answer.py +36 -0
  11. snowflake_ml_python-1.6.0/snowflake/cortex/_sentiment.py +33 -0
  12. snowflake_ml_python-1.6.0/snowflake/cortex/_sse_client.py +81 -0
  13. snowflake_ml_python-1.6.0/snowflake/cortex/_summarize.py +33 -0
  14. snowflake_ml_python-1.6.0/snowflake/cortex/_translate.py +39 -0
  15. snowflake_ml_python-1.6.0/snowflake/cortex/_util.py +66 -0
  16. snowflake_ml_python-1.6.0/snowflake/ml/_internal/container_services/image_registry/credential.py +84 -0
  17. snowflake_ml_python-1.6.0/snowflake/ml/_internal/container_services/image_registry/http_client.py +127 -0
  18. snowflake_ml_python-1.6.0/snowflake/ml/_internal/container_services/image_registry/imagelib.py +400 -0
  19. snowflake_ml_python-1.6.0/snowflake/ml/_internal/container_services/image_registry/registry_client.py +212 -0
  20. snowflake_ml_python-1.6.0/snowflake/ml/_internal/env.py +8 -0
  21. snowflake_ml_python-1.6.0/snowflake/ml/_internal/env_utils.py +675 -0
  22. snowflake_ml_python-1.6.0/snowflake/ml/_internal/exceptions/dataset_error_messages.py +5 -0
  23. snowflake_ml_python-1.6.0/snowflake/ml/_internal/exceptions/dataset_errors.py +24 -0
  24. snowflake_ml_python-1.6.0/snowflake/ml/_internal/exceptions/error_codes.py +110 -0
  25. snowflake_ml_python-1.6.0/snowflake/ml/_internal/exceptions/error_messages.py +1 -0
  26. snowflake_ml_python-1.6.0/snowflake/ml/_internal/exceptions/exceptions.py +45 -0
  27. snowflake_ml_python-1.6.0/snowflake/ml/_internal/exceptions/fileset_error_messages.py +8 -0
  28. snowflake_ml_python-1.6.0/snowflake/ml/_internal/exceptions/fileset_errors.py +36 -0
  29. snowflake_ml_python-1.6.0/snowflake/ml/_internal/exceptions/modeling_error_messages.py +10 -0
  30. snowflake_ml_python-1.6.0/snowflake/ml/_internal/exceptions/sql_error_codes.py +6 -0
  31. snowflake_ml_python-1.6.0/snowflake/ml/_internal/file_utils.py +381 -0
  32. snowflake_ml_python-1.6.0/snowflake/ml/_internal/human_readable_id/adjectives.txt +128 -0
  33. snowflake_ml_python-1.6.0/snowflake/ml/_internal/human_readable_id/animals.txt +128 -0
  34. snowflake_ml_python-1.6.0/snowflake/ml/_internal/human_readable_id/hrid_generator.py +40 -0
  35. snowflake_ml_python-1.6.0/snowflake/ml/_internal/human_readable_id/hrid_generator_base.py +135 -0
  36. snowflake_ml_python-1.6.0/snowflake/ml/_internal/init_utils.py +67 -0
  37. snowflake_ml_python-1.6.0/snowflake/ml/_internal/lineage/lineage_utils.py +104 -0
  38. snowflake_ml_python-1.6.0/snowflake/ml/_internal/migrator_utils.py +3 -0
  39. snowflake_ml_python-1.6.0/snowflake/ml/_internal/telemetry.py +651 -0
  40. snowflake_ml_python-1.6.0/snowflake/ml/_internal/type_utils.py +69 -0
  41. snowflake_ml_python-1.6.0/snowflake/ml/_internal/utils/formatting.py +98 -0
  42. snowflake_ml_python-1.6.0/snowflake/ml/_internal/utils/identifier.py +357 -0
  43. snowflake_ml_python-1.6.0/snowflake/ml/_internal/utils/import_utils.py +60 -0
  44. snowflake_ml_python-1.6.0/snowflake/ml/_internal/utils/log_stream_processor.py +30 -0
  45. snowflake_ml_python-1.6.0/snowflake/ml/_internal/utils/parallelize.py +112 -0
  46. snowflake_ml_python-1.6.0/snowflake/ml/_internal/utils/pkg_version_utils.py +149 -0
  47. snowflake_ml_python-1.6.0/snowflake/ml/_internal/utils/query_result_checker.py +264 -0
  48. snowflake_ml_python-1.6.0/snowflake/ml/_internal/utils/result.py +66 -0
  49. snowflake_ml_python-1.6.0/snowflake/ml/_internal/utils/retryable_http.py +39 -0
  50. snowflake_ml_python-1.6.0/snowflake/ml/_internal/utils/session_token_manager.py +46 -0
  51. snowflake_ml_python-1.6.0/snowflake/ml/_internal/utils/snowflake_env.py +95 -0
  52. snowflake_ml_python-1.6.0/snowflake/ml/_internal/utils/snowpark_dataframe_utils.py +123 -0
  53. snowflake_ml_python-1.6.0/snowflake/ml/_internal/utils/spcs_attribution_utils.py +122 -0
  54. snowflake_ml_python-1.6.0/snowflake/ml/_internal/utils/sql_identifier.py +94 -0
  55. snowflake_ml_python-1.6.0/snowflake/ml/_internal/utils/table_manager.py +106 -0
  56. snowflake_ml_python-1.6.0/snowflake/ml/_internal/utils/temp_file_utils.py +53 -0
  57. snowflake_ml_python-1.6.0/snowflake/ml/_internal/utils/uri.py +77 -0
  58. snowflake_ml_python-1.6.0/snowflake/ml/data/_internal/arrow_ingestor.py +228 -0
  59. snowflake_ml_python-1.6.0/snowflake/ml/data/_internal/ingestor_utils.py +58 -0
  60. snowflake_ml_python-1.6.0/snowflake/ml/data/data_connector.py +133 -0
  61. snowflake_ml_python-1.6.0/snowflake/ml/data/data_ingestor.py +28 -0
  62. snowflake_ml_python-1.6.0/snowflake/ml/data/data_source.py +23 -0
  63. snowflake_ml_python-1.6.0/snowflake/ml/dataset/__init__.py +11 -0
  64. snowflake_ml_python-1.6.0/snowflake/ml/dataset/dataset.py +497 -0
  65. snowflake_ml_python-1.6.0/snowflake/ml/dataset/dataset_factory.py +52 -0
  66. snowflake_ml_python-1.6.0/snowflake/ml/dataset/dataset_metadata.py +103 -0
  67. snowflake_ml_python-1.6.0/snowflake/ml/dataset/dataset_reader.py +99 -0
  68. snowflake_ml_python-1.6.0/snowflake/ml/feature_store/__init__.py +15 -0
  69. snowflake_ml_python-1.6.0/snowflake/ml/feature_store/access_manager.py +289 -0
  70. snowflake_ml_python-1.6.0/snowflake/ml/feature_store/entity.py +102 -0
  71. snowflake_ml_python-1.6.0/snowflake/ml/feature_store/examples/citibike_trip_features/entities.py +20 -0
  72. snowflake_ml_python-1.6.0/snowflake/ml/feature_store/examples/citibike_trip_features/features/station_feature.py +31 -0
  73. snowflake_ml_python-1.6.0/snowflake/ml/feature_store/examples/citibike_trip_features/features/trip_feature.py +24 -0
  74. snowflake_ml_python-1.6.0/snowflake/ml/feature_store/examples/citibike_trip_features/source.yaml +4 -0
  75. snowflake_ml_python-1.6.0/snowflake/ml/feature_store/examples/example_helper.py +240 -0
  76. snowflake_ml_python-1.6.0/snowflake/ml/feature_store/examples/new_york_taxi_features/entities.py +12 -0
  77. snowflake_ml_python-1.6.0/snowflake/ml/feature_store/examples/new_york_taxi_features/features/dropoff_features.py +39 -0
  78. snowflake_ml_python-1.6.0/snowflake/ml/feature_store/examples/new_york_taxi_features/features/pickup_features.py +58 -0
  79. snowflake_ml_python-1.6.0/snowflake/ml/feature_store/examples/new_york_taxi_features/source.yaml +5 -0
  80. snowflake_ml_python-1.6.0/snowflake/ml/feature_store/examples/source_data/citibike_trips.yaml +36 -0
  81. snowflake_ml_python-1.6.0/snowflake/ml/feature_store/examples/source_data/fraud_transactions.yaml +29 -0
  82. snowflake_ml_python-1.6.0/snowflake/ml/feature_store/examples/source_data/nyc_yellow_trips.yaml +4 -0
  83. snowflake_ml_python-1.6.0/snowflake/ml/feature_store/examples/source_data/winequality_red.yaml +32 -0
  84. snowflake_ml_python-1.6.0/snowflake/ml/feature_store/examples/wine_quality_features/entities.py +14 -0
  85. snowflake_ml_python-1.6.0/snowflake/ml/feature_store/examples/wine_quality_features/features/managed_wine_features.py +29 -0
  86. snowflake_ml_python-1.6.0/snowflake/ml/feature_store/examples/wine_quality_features/features/static_wine_features.py +21 -0
  87. snowflake_ml_python-1.6.0/snowflake/ml/feature_store/examples/wine_quality_features/source.yaml +5 -0
  88. snowflake_ml_python-1.6.0/snowflake/ml/feature_store/feature_store.py +2552 -0
  89. snowflake_ml_python-1.6.0/snowflake/ml/feature_store/feature_view.py +718 -0
  90. snowflake_ml_python-1.6.0/snowflake/ml/fileset/embedded_stage_fs.py +153 -0
  91. snowflake_ml_python-1.6.0/snowflake/ml/fileset/fileset.py +582 -0
  92. snowflake_ml_python-1.6.0/snowflake/ml/fileset/parquet_parser.py +170 -0
  93. snowflake_ml_python-1.6.0/snowflake/ml/fileset/sfcfs.py +345 -0
  94. snowflake_ml_python-1.6.0/snowflake/ml/fileset/snowfs.py +120 -0
  95. snowflake_ml_python-1.6.0/snowflake/ml/fileset/stage_fs.py +434 -0
  96. snowflake_ml_python-1.6.0/snowflake/ml/fileset/tf_dataset.py +88 -0
  97. snowflake_ml_python-1.6.0/snowflake/ml/fileset/torch_datapipe.py +57 -0
  98. snowflake_ml_python-1.6.0/snowflake/ml/lineage/__init__.py +3 -0
  99. snowflake_ml_python-1.6.0/snowflake/ml/lineage/lineage_node.py +139 -0
  100. snowflake_ml_python-1.6.0/snowflake/ml/model/__init__.py +6 -0
  101. snowflake_ml_python-1.6.0/snowflake/ml/model/_api.py +568 -0
  102. snowflake_ml_python-1.6.0/snowflake/ml/model/_client/model/model_impl.py +419 -0
  103. snowflake_ml_python-1.6.0/snowflake/ml/model/_client/model/model_version_impl.py +533 -0
  104. snowflake_ml_python-1.6.0/snowflake/ml/model/_client/ops/metadata_ops.py +128 -0
  105. snowflake_ml_python-1.6.0/snowflake/ml/model/_client/ops/model_ops.py +814 -0
  106. snowflake_ml_python-1.6.0/snowflake/ml/model/_client/sql/_base.py +34 -0
  107. snowflake_ml_python-1.6.0/snowflake/ml/model/_client/sql/model.py +137 -0
  108. snowflake_ml_python-1.6.0/snowflake/ml/model/_client/sql/model_version.py +484 -0
  109. snowflake_ml_python-1.6.0/snowflake/ml/model/_client/sql/stage.py +20 -0
  110. snowflake_ml_python-1.6.0/snowflake/ml/model/_client/sql/tag.py +94 -0
  111. snowflake_ml_python-1.6.0/snowflake/ml/model/_deploy_client/image_builds/base_image_builder.py +12 -0
  112. snowflake_ml_python-1.6.0/snowflake/ml/model/_deploy_client/image_builds/client_image_builder.py +249 -0
  113. snowflake_ml_python-1.6.0/snowflake/ml/model/_deploy_client/image_builds/docker_context.py +130 -0
  114. snowflake_ml_python-1.6.0/snowflake/ml/model/_deploy_client/image_builds/gunicorn_run.sh +36 -0
  115. snowflake_ml_python-1.6.0/snowflake/ml/model/_deploy_client/image_builds/inference_server/main.py +268 -0
  116. snowflake_ml_python-1.6.0/snowflake/ml/model/_deploy_client/image_builds/server_image_builder.py +215 -0
  117. snowflake_ml_python-1.6.0/snowflake/ml/model/_deploy_client/image_builds/templates/dockerfile_template +53 -0
  118. snowflake_ml_python-1.6.0/snowflake/ml/model/_deploy_client/image_builds/templates/image_build_job_spec_template +38 -0
  119. snowflake_ml_python-1.6.0/snowflake/ml/model/_deploy_client/image_builds/templates/kaniko_shell_script_template +105 -0
  120. snowflake_ml_python-1.6.0/snowflake/ml/model/_deploy_client/snowservice/deploy.py +611 -0
  121. snowflake_ml_python-1.6.0/snowflake/ml/model/_deploy_client/snowservice/deploy_options.py +116 -0
  122. snowflake_ml_python-1.6.0/snowflake/ml/model/_deploy_client/snowservice/instance_types.py +10 -0
  123. snowflake_ml_python-1.6.0/snowflake/ml/model/_deploy_client/snowservice/templates/service_spec_template +28 -0
  124. snowflake_ml_python-1.6.0/snowflake/ml/model/_deploy_client/snowservice/templates/service_spec_template_with_model +21 -0
  125. snowflake_ml_python-1.6.0/snowflake/ml/model/_deploy_client/utils/constants.py +48 -0
  126. snowflake_ml_python-1.6.0/snowflake/ml/model/_deploy_client/utils/snowservice_client.py +280 -0
  127. snowflake_ml_python-1.6.0/snowflake/ml/model/_deploy_client/warehouse/deploy.py +202 -0
  128. snowflake_ml_python-1.6.0/snowflake/ml/model/_deploy_client/warehouse/infer_template.py +99 -0
  129. snowflake_ml_python-1.6.0/snowflake/ml/model/_model_composer/model_composer.py +188 -0
  130. snowflake_ml_python-1.6.0/snowflake/ml/model/_model_composer/model_manifest/model_manifest.py +146 -0
  131. snowflake_ml_python-1.6.0/snowflake/ml/model/_model_composer/model_manifest/model_manifest_schema.py +94 -0
  132. snowflake_ml_python-1.6.0/snowflake/ml/model/_model_composer/model_method/function_generator.py +77 -0
  133. snowflake_ml_python-1.6.0/snowflake/ml/model/_model_composer/model_method/infer_function.py_template +50 -0
  134. snowflake_ml_python-1.6.0/snowflake/ml/model/_model_composer/model_method/infer_partitioned.py_template +55 -0
  135. snowflake_ml_python-1.6.0/snowflake/ml/model/_model_composer/model_method/infer_table_function.py_template +50 -0
  136. snowflake_ml_python-1.6.0/snowflake/ml/model/_model_composer/model_method/model_method.py +153 -0
  137. snowflake_ml_python-1.6.0/snowflake/ml/model/_packager/model_env/model_env.py +415 -0
  138. snowflake_ml_python-1.6.0/snowflake/ml/model/_packager/model_handler.py +79 -0
  139. snowflake_ml_python-1.6.0/snowflake/ml/model/_packager/model_handlers/_base.py +171 -0
  140. snowflake_ml_python-1.6.0/snowflake/ml/model/_packager/model_handlers/_utils.py +115 -0
  141. snowflake_ml_python-1.6.0/snowflake/ml/model/_packager/model_handlers/catboost.py +248 -0
  142. snowflake_ml_python-1.6.0/snowflake/ml/model/_packager/model_handlers/custom.py +188 -0
  143. snowflake_ml_python-1.6.0/snowflake/ml/model/_packager/model_handlers/huggingface_pipeline.py +458 -0
  144. snowflake_ml_python-1.6.0/snowflake/ml/model/_packager/model_handlers/lightgbm.py +286 -0
  145. snowflake_ml_python-1.6.0/snowflake/ml/model/_packager/model_handlers/llm.py +263 -0
  146. snowflake_ml_python-1.6.0/snowflake/ml/model/_packager/model_handlers/mlflow.py +234 -0
  147. snowflake_ml_python-1.6.0/snowflake/ml/model/_packager/model_handlers/pytorch.py +212 -0
  148. snowflake_ml_python-1.6.0/snowflake/ml/model/_packager/model_handlers/sentence_transformers.py +214 -0
  149. snowflake_ml_python-1.6.0/snowflake/ml/model/_packager/model_handlers/sklearn.py +199 -0
  150. snowflake_ml_python-1.6.0/snowflake/ml/model/_packager/model_handlers/snowmlmodel.py +192 -0
  151. snowflake_ml_python-1.6.0/snowflake/ml/model/_packager/model_handlers/tensorflow.py +212 -0
  152. snowflake_ml_python-1.6.0/snowflake/ml/model/_packager/model_handlers/torchscript.py +209 -0
  153. snowflake_ml_python-1.6.0/snowflake/ml/model/_packager/model_handlers/xgboost.py +285 -0
  154. snowflake_ml_python-1.6.0/snowflake/ml/model/_packager/model_handlers_migrator/base_migrator.py +36 -0
  155. snowflake_ml_python-1.6.0/snowflake/ml/model/_packager/model_meta/_core_requirements.py +11 -0
  156. snowflake_ml_python-1.6.0/snowflake/ml/model/_packager/model_meta/_packaging_requirements.py +3 -0
  157. snowflake_ml_python-1.6.0/snowflake/ml/model/_packager/model_meta/model_blob_meta.py +50 -0
  158. snowflake_ml_python-1.6.0/snowflake/ml/model/_packager/model_meta/model_meta.py +420 -0
  159. snowflake_ml_python-1.6.0/snowflake/ml/model/_packager/model_meta/model_meta_schema.py +94 -0
  160. snowflake_ml_python-1.6.0/snowflake/ml/model/_packager/model_meta_migrator/base_migrator.py +33 -0
  161. snowflake_ml_python-1.6.0/snowflake/ml/model/_packager/model_meta_migrator/migrator_plans.py +23 -0
  162. snowflake_ml_python-1.6.0/snowflake/ml/model/_packager/model_meta_migrator/migrator_v1.py +48 -0
  163. snowflake_ml_python-1.6.0/snowflake/ml/model/_packager/model_packager.py +153 -0
  164. snowflake_ml_python-1.6.0/snowflake/ml/model/_packager/model_runtime/_snowml_inference_alternative_requirements.py +10 -0
  165. snowflake_ml_python-1.6.0/snowflake/ml/model/_packager/model_runtime/model_runtime.py +111 -0
  166. snowflake_ml_python-1.6.0/snowflake/ml/model/_signatures/base_handler.py +47 -0
  167. snowflake_ml_python-1.6.0/snowflake/ml/model/_signatures/builtins_handler.py +68 -0
  168. snowflake_ml_python-1.6.0/snowflake/ml/model/_signatures/core.py +512 -0
  169. snowflake_ml_python-1.6.0/snowflake/ml/model/_signatures/numpy_handler.py +134 -0
  170. snowflake_ml_python-1.6.0/snowflake/ml/model/_signatures/pandas_handler.py +169 -0
  171. snowflake_ml_python-1.6.0/snowflake/ml/model/_signatures/pytorch_handler.py +107 -0
  172. snowflake_ml_python-1.6.0/snowflake/ml/model/_signatures/snowpark_handler.py +128 -0
  173. snowflake_ml_python-1.6.0/snowflake/ml/model/_signatures/tensorflow_handler.py +145 -0
  174. snowflake_ml_python-1.6.0/snowflake/ml/model/_signatures/utils.py +290 -0
  175. snowflake_ml_python-1.6.0/snowflake/ml/model/custom_model.py +241 -0
  176. snowflake_ml_python-1.6.0/snowflake/ml/model/deploy_platforms.py +6 -0
  177. snowflake_ml_python-1.6.0/snowflake/ml/model/model_signature.py +690 -0
  178. snowflake_ml_python-1.6.0/snowflake/ml/model/models/huggingface_pipeline.py +214 -0
  179. snowflake_ml_python-1.6.0/snowflake/ml/model/models/llm.py +104 -0
  180. snowflake_ml_python-1.6.0/snowflake/ml/model/type_hints.py +435 -0
  181. snowflake_ml_python-1.6.0/snowflake/ml/modeling/_internal/constants.py +1 -0
  182. snowflake_ml_python-1.6.0/snowflake/ml/modeling/_internal/estimator_utils.py +277 -0
  183. snowflake_ml_python-1.6.0/snowflake/ml/modeling/_internal/local_implementations/pandas_handlers.py +190 -0
  184. snowflake_ml_python-1.6.0/snowflake/ml/modeling/_internal/local_implementations/pandas_trainer.py +134 -0
  185. snowflake_ml_python-1.6.0/snowflake/ml/modeling/_internal/ml_runtime_implementations/ml_runtime_handlers.py +151 -0
  186. snowflake_ml_python-1.6.0/snowflake/ml/modeling/_internal/ml_runtime_implementations/ml_runtime_trainer.py +66 -0
  187. snowflake_ml_python-1.6.0/snowflake/ml/modeling/_internal/model_specifications.py +141 -0
  188. snowflake_ml_python-1.6.0/snowflake/ml/modeling/_internal/model_trainer.py +31 -0
  189. snowflake_ml_python-1.6.0/snowflake/ml/modeling/_internal/model_trainer_builder.py +214 -0
  190. snowflake_ml_python-1.6.0/snowflake/ml/modeling/_internal/model_transformer_builder.py +85 -0
  191. snowflake_ml_python-1.6.0/snowflake/ml/modeling/_internal/snowpark_implementations/distributed_hpo_trainer.py +1173 -0
  192. snowflake_ml_python-1.6.0/snowflake/ml/modeling/_internal/snowpark_implementations/distributed_search_udf_file.py +173 -0
  193. snowflake_ml_python-1.6.0/snowflake/ml/modeling/_internal/snowpark_implementations/snowpark_handlers.py +380 -0
  194. snowflake_ml_python-1.6.0/snowflake/ml/modeling/_internal/snowpark_implementations/snowpark_trainer.py +800 -0
  195. snowflake_ml_python-1.6.0/snowflake/ml/modeling/_internal/snowpark_implementations/xgboost_external_memory_trainer.py +451 -0
  196. snowflake_ml_python-1.6.0/snowflake/ml/modeling/_internal/transformer_protocols.py +190 -0
  197. snowflake_ml_python-1.6.0/snowflake/ml/modeling/calibration/__init__.py +8 -0
  198. snowflake_ml_python-1.6.0/snowflake/ml/modeling/calibration/calibrated_classifier_cv.py +1156 -0
  199. snowflake_ml_python-1.6.0/snowflake/ml/modeling/cluster/__init__.py +8 -0
  200. snowflake_ml_python-1.6.0/snowflake/ml/modeling/cluster/affinity_propagation.py +1129 -0
  201. snowflake_ml_python-1.6.0/snowflake/ml/modeling/cluster/agglomerative_clustering.py +1160 -0
  202. snowflake_ml_python-1.6.0/snowflake/ml/modeling/cluster/birch.py +1124 -0
  203. snowflake_ml_python-1.6.0/snowflake/ml/modeling/cluster/bisecting_k_means.py +1175 -0
  204. snowflake_ml_python-1.6.0/snowflake/ml/modeling/cluster/dbscan.py +1135 -0
  205. snowflake_ml_python-1.6.0/snowflake/ml/modeling/cluster/feature_agglomeration.py +1171 -0
  206. snowflake_ml_python-1.6.0/snowflake/ml/modeling/cluster/k_means.py +1170 -0
  207. snowflake_ml_python-1.6.0/snowflake/ml/modeling/cluster/mean_shift.py +1140 -0
  208. snowflake_ml_python-1.6.0/snowflake/ml/modeling/cluster/mini_batch_k_means.py +1196 -0
  209. snowflake_ml_python-1.6.0/snowflake/ml/modeling/cluster/optics.py +1208 -0
  210. snowflake_ml_python-1.6.0/snowflake/ml/modeling/cluster/spectral_biclustering.py +1144 -0
  211. snowflake_ml_python-1.6.0/snowflake/ml/modeling/cluster/spectral_clustering.py +1204 -0
  212. snowflake_ml_python-1.6.0/snowflake/ml/modeling/cluster/spectral_coclustering.py +1123 -0
  213. snowflake_ml_python-1.6.0/snowflake/ml/modeling/compose/__init__.py +8 -0
  214. snowflake_ml_python-1.6.0/snowflake/ml/modeling/compose/column_transformer.py +1157 -0
  215. snowflake_ml_python-1.6.0/snowflake/ml/modeling/compose/transformed_target_regressor.py +1116 -0
  216. snowflake_ml_python-1.6.0/snowflake/ml/modeling/covariance/__init__.py +8 -0
  217. snowflake_ml_python-1.6.0/snowflake/ml/modeling/covariance/elliptic_envelope.py +1119 -0
  218. snowflake_ml_python-1.6.0/snowflake/ml/modeling/covariance/empirical_covariance.py +1087 -0
  219. snowflake_ml_python-1.6.0/snowflake/ml/modeling/covariance/graphical_lasso.py +1135 -0
  220. snowflake_ml_python-1.6.0/snowflake/ml/modeling/covariance/graphical_lasso_cv.py +1161 -0
  221. snowflake_ml_python-1.6.0/snowflake/ml/modeling/covariance/ledoit_wolf.py +1094 -0
  222. snowflake_ml_python-1.6.0/snowflake/ml/modeling/covariance/min_cov_det.py +1106 -0
  223. snowflake_ml_python-1.6.0/snowflake/ml/modeling/covariance/oas.py +1087 -0
  224. snowflake_ml_python-1.6.0/snowflake/ml/modeling/covariance/shrunk_covariance.py +1093 -0
  225. snowflake_ml_python-1.6.0/snowflake/ml/modeling/decomposition/__init__.py +8 -0
  226. snowflake_ml_python-1.6.0/snowflake/ml/modeling/decomposition/dictionary_learning.py +1201 -0
  227. snowflake_ml_python-1.6.0/snowflake/ml/modeling/decomposition/factor_analysis.py +1142 -0
  228. snowflake_ml_python-1.6.0/snowflake/ml/modeling/decomposition/fast_ica.py +1156 -0
  229. snowflake_ml_python-1.6.0/snowflake/ml/modeling/decomposition/incremental_pca.py +1108 -0
  230. snowflake_ml_python-1.6.0/snowflake/ml/modeling/decomposition/kernel_pca.py +1204 -0
  231. snowflake_ml_python-1.6.0/snowflake/ml/modeling/decomposition/mini_batch_dictionary_learning.py +1226 -0
  232. snowflake_ml_python-1.6.0/snowflake/ml/modeling/decomposition/mini_batch_sparse_pca.py +1171 -0
  233. snowflake_ml_python-1.6.0/snowflake/ml/modeling/decomposition/pca.py +1177 -0
  234. snowflake_ml_python-1.6.0/snowflake/ml/modeling/decomposition/sparse_pca.py +1146 -0
  235. snowflake_ml_python-1.6.0/snowflake/ml/modeling/decomposition/truncated_svd.py +1127 -0
  236. snowflake_ml_python-1.6.0/snowflake/ml/modeling/discriminant_analysis/__init__.py +8 -0
  237. snowflake_ml_python-1.6.0/snowflake/ml/modeling/discriminant_analysis/linear_discriminant_analysis.py +1152 -0
  238. snowflake_ml_python-1.6.0/snowflake/ml/modeling/discriminant_analysis/quadratic_discriminant_analysis.py +1110 -0
  239. snowflake_ml_python-1.6.0/snowflake/ml/modeling/ensemble/__init__.py +8 -0
  240. snowflake_ml_python-1.6.0/snowflake/ml/modeling/ensemble/ada_boost_classifier.py +1135 -0
  241. snowflake_ml_python-1.6.0/snowflake/ml/modeling/ensemble/ada_boost_regressor.py +1126 -0
  242. snowflake_ml_python-1.6.0/snowflake/ml/modeling/ensemble/bagging_classifier.py +1167 -0
  243. snowflake_ml_python-1.6.0/snowflake/ml/modeling/ensemble/bagging_regressor.py +1161 -0
  244. snowflake_ml_python-1.6.0/snowflake/ml/modeling/ensemble/extra_trees_classifier.py +1268 -0
  245. snowflake_ml_python-1.6.0/snowflake/ml/modeling/ensemble/extra_trees_regressor.py +1243 -0
  246. snowflake_ml_python-1.6.0/snowflake/ml/modeling/ensemble/gradient_boosting_classifier.py +1282 -0
  247. snowflake_ml_python-1.6.0/snowflake/ml/modeling/ensemble/gradient_boosting_regressor.py +1285 -0
  248. snowflake_ml_python-1.6.0/snowflake/ml/modeling/ensemble/hist_gradient_boosting_classifier.py +1263 -0
  249. snowflake_ml_python-1.6.0/snowflake/ml/modeling/ensemble/hist_gradient_boosting_regressor.py +1248 -0
  250. snowflake_ml_python-1.6.0/snowflake/ml/modeling/ensemble/isolation_forest.py +1154 -0
  251. snowflake_ml_python-1.6.0/snowflake/ml/modeling/ensemble/random_forest_classifier.py +1264 -0
  252. snowflake_ml_python-1.6.0/snowflake/ml/modeling/ensemble/random_forest_regressor.py +1239 -0
  253. snowflake_ml_python-1.6.0/snowflake/ml/modeling/ensemble/stacking_regressor.py +1144 -0
  254. snowflake_ml_python-1.6.0/snowflake/ml/modeling/ensemble/voting_classifier.py +1130 -0
  255. snowflake_ml_python-1.6.0/snowflake/ml/modeling/ensemble/voting_regressor.py +1108 -0
  256. snowflake_ml_python-1.6.0/snowflake/ml/modeling/feature_selection/__init__.py +8 -0
  257. snowflake_ml_python-1.6.0/snowflake/ml/modeling/feature_selection/generic_univariate_select.py +1094 -0
  258. snowflake_ml_python-1.6.0/snowflake/ml/modeling/feature_selection/select_fdr.py +1090 -0
  259. snowflake_ml_python-1.6.0/snowflake/ml/modeling/feature_selection/select_fpr.py +1090 -0
  260. snowflake_ml_python-1.6.0/snowflake/ml/modeling/feature_selection/select_fwe.py +1090 -0
  261. snowflake_ml_python-1.6.0/snowflake/ml/modeling/feature_selection/select_k_best.py +1091 -0
  262. snowflake_ml_python-1.6.0/snowflake/ml/modeling/feature_selection/select_percentile.py +1090 -0
  263. snowflake_ml_python-1.6.0/snowflake/ml/modeling/feature_selection/sequential_feature_selector.py +1150 -0
  264. snowflake_ml_python-1.6.0/snowflake/ml/modeling/feature_selection/variance_threshold.py +1083 -0
  265. snowflake_ml_python-1.6.0/snowflake/ml/modeling/framework/_utils.py +256 -0
  266. snowflake_ml_python-1.6.0/snowflake/ml/modeling/framework/base.py +820 -0
  267. snowflake_ml_python-1.6.0/snowflake/ml/modeling/gaussian_process/__init__.py +8 -0
  268. snowflake_ml_python-1.6.0/snowflake/ml/modeling/gaussian_process/gaussian_process_classifier.py +1180 -0
  269. snowflake_ml_python-1.6.0/snowflake/ml/modeling/gaussian_process/gaussian_process_regressor.py +1167 -0
  270. snowflake_ml_python-1.6.0/snowflake/ml/modeling/impute/__init__.py +9 -0
  271. snowflake_ml_python-1.6.0/snowflake/ml/modeling/impute/iterative_imputer.py +1211 -0
  272. snowflake_ml_python-1.6.0/snowflake/ml/modeling/impute/knn_imputer.py +1137 -0
  273. snowflake_ml_python-1.6.0/snowflake/ml/modeling/impute/missing_indicator.py +1111 -0
  274. snowflake_ml_python-1.6.0/snowflake/ml/modeling/impute/simple_imputer.py +458 -0
  275. snowflake_ml_python-1.6.0/snowflake/ml/modeling/kernel_approximation/__init__.py +8 -0
  276. snowflake_ml_python-1.6.0/snowflake/ml/modeling/kernel_approximation/additive_chi2_sampler.py +1086 -0
  277. snowflake_ml_python-1.6.0/snowflake/ml/modeling/kernel_approximation/nystroem.py +1134 -0
  278. snowflake_ml_python-1.6.0/snowflake/ml/modeling/kernel_approximation/polynomial_count_sketch.py +1110 -0
  279. snowflake_ml_python-1.6.0/snowflake/ml/modeling/kernel_approximation/rbf_sampler.py +1097 -0
  280. snowflake_ml_python-1.6.0/snowflake/ml/modeling/kernel_approximation/skewed_chi2_sampler.py +1095 -0
  281. snowflake_ml_python-1.6.0/snowflake/ml/modeling/kernel_ridge/__init__.py +8 -0
  282. snowflake_ml_python-1.6.0/snowflake/ml/modeling/kernel_ridge/kernel_ridge.py +1129 -0
  283. snowflake_ml_python-1.6.0/snowflake/ml/modeling/lightgbm/__init__.py +8 -0
  284. snowflake_ml_python-1.6.0/snowflake/ml/modeling/lightgbm/lgbm_classifier.py +1122 -0
  285. snowflake_ml_python-1.6.0/snowflake/ml/modeling/lightgbm/lgbm_regressor.py +1118 -0
  286. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/__init__.py +8 -0
  287. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/ard_regression.py +1143 -0
  288. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/bayesian_ridge.py +1154 -0
  289. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/elastic_net.py +1153 -0
  290. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/elastic_net_cv.py +1189 -0
  291. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/gamma_regressor.py +1134 -0
  292. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/huber_regressor.py +1117 -0
  293. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/lars.py +1146 -0
  294. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/lars_cv.py +1154 -0
  295. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/lasso.py +1147 -0
  296. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/lasso_cv.py +1175 -0
  297. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/lasso_lars.py +1167 -0
  298. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/lasso_lars_cv.py +1168 -0
  299. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/lasso_lars_ic.py +1151 -0
  300. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/linear_regression.py +1104 -0
  301. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/logistic_regression.py +1224 -0
  302. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/logistic_regression_cv.py +1245 -0
  303. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/multi_task_elastic_net.py +1137 -0
  304. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/multi_task_elastic_net_cv.py +1178 -0
  305. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/multi_task_lasso.py +1129 -0
  306. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/multi_task_lasso_cv.py +1164 -0
  307. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/orthogonal_matching_pursuit.py +1112 -0
  308. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/passive_aggressive_classifier.py +1188 -0
  309. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/passive_aggressive_regressor.py +1172 -0
  310. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/perceptron.py +1187 -0
  311. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/poisson_regressor.py +1134 -0
  312. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/ransac_regressor.py +1190 -0
  313. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/ridge.py +1182 -0
  314. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/ridge_classifier.py +1184 -0
  315. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/ridge_classifier_cv.py +1135 -0
  316. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/ridge_cv.py +1154 -0
  317. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/sgd_classifier.py +1279 -0
  318. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/sgd_one_class_svm.py +1177 -0
  319. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/sgd_regressor.py +1239 -0
  320. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/theil_sen_regressor.py +1141 -0
  321. snowflake_ml_python-1.6.0/snowflake/ml/modeling/linear_model/tweedie_regressor.py +1167 -0
  322. snowflake_ml_python-1.6.0/snowflake/ml/modeling/manifold/__init__.py +8 -0
  323. snowflake_ml_python-1.6.0/snowflake/ml/modeling/manifold/isomap.py +1165 -0
  324. snowflake_ml_python-1.6.0/snowflake/ml/modeling/manifold/mds.py +1146 -0
  325. snowflake_ml_python-1.6.0/snowflake/ml/modeling/manifold/spectral_embedding.py +1148 -0
  326. snowflake_ml_python-1.6.0/snowflake/ml/modeling/manifold/tsne.py +1207 -0
  327. snowflake_ml_python-1.6.0/snowflake/ml/modeling/metrics/__init__.py +16 -0
  328. snowflake_ml_python-1.6.0/snowflake/ml/modeling/metrics/classification.py +1567 -0
  329. snowflake_ml_python-1.6.0/snowflake/ml/modeling/metrics/correlation.py +100 -0
  330. snowflake_ml_python-1.6.0/snowflake/ml/modeling/metrics/covariance.py +102 -0
  331. snowflake_ml_python-1.6.0/snowflake/ml/modeling/metrics/metrics_utils.py +340 -0
  332. snowflake_ml_python-1.6.0/snowflake/ml/modeling/metrics/ranking.py +385 -0
  333. snowflake_ml_python-1.6.0/snowflake/ml/modeling/metrics/regression.py +625 -0
  334. snowflake_ml_python-1.6.0/snowflake/ml/modeling/mixture/__init__.py +8 -0
  335. snowflake_ml_python-1.6.0/snowflake/ml/modeling/mixture/bayesian_gaussian_mixture.py +1220 -0
  336. snowflake_ml_python-1.6.0/snowflake/ml/modeling/mixture/gaussian_mixture.py +1193 -0
  337. snowflake_ml_python-1.6.0/snowflake/ml/modeling/model_selection/__init__.py +9 -0
  338. snowflake_ml_python-1.6.0/snowflake/ml/modeling/model_selection/grid_search_cv.py +865 -0
  339. snowflake_ml_python-1.6.0/snowflake/ml/modeling/model_selection/randomized_search_cv.py +880 -0
  340. snowflake_ml_python-1.6.0/snowflake/ml/modeling/multiclass/__init__.py +8 -0
  341. snowflake_ml_python-1.6.0/snowflake/ml/modeling/multiclass/one_vs_one_classifier.py +1097 -0
  342. snowflake_ml_python-1.6.0/snowflake/ml/modeling/multiclass/one_vs_rest_classifier.py +1110 -0
  343. snowflake_ml_python-1.6.0/snowflake/ml/modeling/multiclass/output_code_classifier.py +1107 -0
  344. snowflake_ml_python-1.6.0/snowflake/ml/modeling/naive_bayes/__init__.py +8 -0
  345. snowflake_ml_python-1.6.0/snowflake/ml/modeling/naive_bayes/bernoulli_nb.py +1111 -0
  346. snowflake_ml_python-1.6.0/snowflake/ml/modeling/naive_bayes/categorical_nb.py +1117 -0
  347. snowflake_ml_python-1.6.0/snowflake/ml/modeling/naive_bayes/complement_nb.py +1111 -0
  348. snowflake_ml_python-1.6.0/snowflake/ml/modeling/naive_bayes/gaussian_nb.py +1092 -0
  349. snowflake_ml_python-1.6.0/snowflake/ml/modeling/naive_bayes/multinomial_nb.py +1105 -0
  350. snowflake_ml_python-1.6.0/snowflake/ml/modeling/neighbors/__init__.py +8 -0
  351. snowflake_ml_python-1.6.0/snowflake/ml/modeling/neighbors/k_neighbors_classifier.py +1164 -0
  352. snowflake_ml_python-1.6.0/snowflake/ml/modeling/neighbors/k_neighbors_regressor.py +1162 -0
  353. snowflake_ml_python-1.6.0/snowflake/ml/modeling/neighbors/kernel_density.py +1139 -0
  354. snowflake_ml_python-1.6.0/snowflake/ml/modeling/neighbors/local_outlier_factor.py +1173 -0
  355. snowflake_ml_python-1.6.0/snowflake/ml/modeling/neighbors/nearest_centroid.py +1098 -0
  356. snowflake_ml_python-1.6.0/snowflake/ml/modeling/neighbors/nearest_neighbors.py +1148 -0
  357. snowflake_ml_python-1.6.0/snowflake/ml/modeling/neighbors/neighborhood_components_analysis.py +1169 -0
  358. snowflake_ml_python-1.6.0/snowflake/ml/modeling/neighbors/radius_neighbors_classifier.py +1174 -0
  359. snowflake_ml_python-1.6.0/snowflake/ml/modeling/neighbors/radius_neighbors_regressor.py +1160 -0
  360. snowflake_ml_python-1.6.0/snowflake/ml/modeling/neural_network/__init__.py +8 -0
  361. snowflake_ml_python-1.6.0/snowflake/ml/modeling/neural_network/bernoulli_rbm.py +1121 -0
  362. snowflake_ml_python-1.6.0/snowflake/ml/modeling/neural_network/mlp_classifier.py +1276 -0
  363. snowflake_ml_python-1.6.0/snowflake/ml/modeling/neural_network/mlp_regressor.py +1268 -0
  364. snowflake_ml_python-1.6.0/snowflake/ml/modeling/parameters/disable_distributed_hpo.py +4 -0
  365. snowflake_ml_python-1.6.0/snowflake/ml/modeling/parameters/enable_anonymous_sproc.py +5 -0
  366. snowflake_ml_python-1.6.0/snowflake/ml/modeling/pipeline/__init__.py +9 -0
  367. snowflake_ml_python-1.6.0/snowflake/ml/modeling/pipeline/pipeline.py +1141 -0
  368. snowflake_ml_python-1.6.0/snowflake/ml/modeling/preprocessing/__init__.py +9 -0
  369. snowflake_ml_python-1.6.0/snowflake/ml/modeling/preprocessing/binarizer.py +164 -0
  370. snowflake_ml_python-1.6.0/snowflake/ml/modeling/preprocessing/k_bins_discretizer.py +509 -0
  371. snowflake_ml_python-1.6.0/snowflake/ml/modeling/preprocessing/label_encoder.py +177 -0
  372. snowflake_ml_python-1.6.0/snowflake/ml/modeling/preprocessing/max_abs_scaler.py +221 -0
  373. snowflake_ml_python-1.6.0/snowflake/ml/modeling/preprocessing/min_max_scaler.py +280 -0
  374. snowflake_ml_python-1.6.0/snowflake/ml/modeling/preprocessing/normalizer.py +177 -0
  375. snowflake_ml_python-1.6.0/snowflake/ml/modeling/preprocessing/one_hot_encoder.py +1590 -0
  376. snowflake_ml_python-1.6.0/snowflake/ml/modeling/preprocessing/ordinal_encoder.py +757 -0
  377. snowflake_ml_python-1.6.0/snowflake/ml/modeling/preprocessing/polynomial_features.py +1109 -0
  378. snowflake_ml_python-1.6.0/snowflake/ml/modeling/preprocessing/robust_scaler.py +286 -0
  379. snowflake_ml_python-1.6.0/snowflake/ml/modeling/preprocessing/standard_scaler.py +265 -0
  380. snowflake_ml_python-1.6.0/snowflake/ml/modeling/semi_supervised/__init__.py +8 -0
  381. snowflake_ml_python-1.6.0/snowflake/ml/modeling/semi_supervised/label_propagation.py +1117 -0
  382. snowflake_ml_python-1.6.0/snowflake/ml/modeling/semi_supervised/label_spreading.py +1126 -0
  383. snowflake_ml_python-1.6.0/snowflake/ml/modeling/svm/__init__.py +8 -0
  384. snowflake_ml_python-1.6.0/snowflake/ml/modeling/svm/linear_svc.py +1180 -0
  385. snowflake_ml_python-1.6.0/snowflake/ml/modeling/svm/linear_svr.py +1150 -0
  386. snowflake_ml_python-1.6.0/snowflake/ml/modeling/svm/nu_svc.py +1190 -0
  387. snowflake_ml_python-1.6.0/snowflake/ml/modeling/svm/nu_svr.py +1145 -0
  388. snowflake_ml_python-1.6.0/snowflake/ml/modeling/svm/svc.py +1193 -0
  389. snowflake_ml_python-1.6.0/snowflake/ml/modeling/svm/svr.py +1148 -0
  390. snowflake_ml_python-1.6.0/snowflake/ml/modeling/tree/__init__.py +8 -0
  391. snowflake_ml_python-1.6.0/snowflake/ml/modeling/tree/decision_tree_classifier.py +1219 -0
  392. snowflake_ml_python-1.6.0/snowflake/ml/modeling/tree/decision_tree_regressor.py +1197 -0
  393. snowflake_ml_python-1.6.0/snowflake/ml/modeling/tree/extra_tree_classifier.py +1211 -0
  394. snowflake_ml_python-1.6.0/snowflake/ml/modeling/tree/extra_tree_regressor.py +1189 -0
  395. snowflake_ml_python-1.6.0/snowflake/ml/modeling/xgboost/__init__.py +8 -0
  396. snowflake_ml_python-1.6.0/snowflake/ml/modeling/xgboost/xgb_classifier.py +1311 -0
  397. snowflake_ml_python-1.6.0/snowflake/ml/modeling/xgboost/xgb_regressor.py +1306 -0
  398. snowflake_ml_python-1.6.0/snowflake/ml/modeling/xgboost/xgbrf_classifier.py +1315 -0
  399. snowflake_ml_python-1.6.0/snowflake/ml/modeling/xgboost/xgbrf_regressor.py +1311 -0
  400. snowflake_ml_python-1.6.0/snowflake/ml/monitoring/monitor.py +203 -0
  401. snowflake_ml_python-1.6.0/snowflake/ml/monitoring/shap.py +101 -0
  402. snowflake_ml_python-1.6.0/snowflake/ml/registry/__init__.py +3 -0
  403. snowflake_ml_python-1.6.0/snowflake/ml/registry/_initial_schema.py +142 -0
  404. snowflake_ml_python-1.6.0/snowflake/ml/registry/_manager/model_manager.py +267 -0
  405. snowflake_ml_python-1.6.0/snowflake/ml/registry/_schema.py +82 -0
  406. snowflake_ml_python-1.6.0/snowflake/ml/registry/_schema_upgrade_plans.py +116 -0
  407. snowflake_ml_python-1.6.0/snowflake/ml/registry/_schema_version_manager.py +163 -0
  408. snowflake_ml_python-1.6.0/snowflake/ml/registry/model_registry.py +2048 -0
  409. snowflake_ml_python-1.6.0/snowflake/ml/registry/registry.py +336 -0
  410. snowflake_ml_python-1.6.0/snowflake/ml/utils/connection_params.py +193 -0
  411. snowflake_ml_python-1.6.0/snowflake/ml/utils/sparse.py +89 -0
  412. snowflake_ml_python-1.6.0/snowflake/ml/version.py +1 -0
  413. snowflake_ml_python-1.6.0/snowflake_ml_python.egg-info/PKG-INFO +1152 -0
  414. snowflake_ml_python-1.6.0/snowflake_ml_python.egg-info/SOURCES.txt +415 -0
  415. snowflake_ml_python-1.6.0/snowflake_ml_python.egg-info/dependency_links.txt +1 -0
  416. snowflake_ml_python-1.6.0/snowflake_ml_python.egg-info/requires.txt +63 -0
  417. snowflake_ml_python-1.6.0/snowflake_ml_python.egg-info/top_level.txt +1 -0
@@ -0,0 +1,779 @@
1
+ # Release History
2
+
3
+ ## 1.6.0
4
+
5
+ ### Bug Fixes
6
+
7
+ - Modeling: `SimpleImputer` can impute integer columns with integer values.
8
+ - Registry: Fix an issue when providing a pandas Dataframe whose index is not starting from 0 as the input to
9
+ the `ModelVersion.run`.
10
+
11
+ ### New Features
12
+
13
+ - Feature Store: Add overloads to APIs accept both object and name/version. Impacted APIs include read_feature_view(),
14
+ refresh_feature_view(), get_refresh_history(), resume_feature_view(), suspend_feature_view(), delete_feature_view().
15
+ - Feature Store: Add docstring inline examples for all public APIs.
16
+ - Feature Store: Add new utility class `ExampleHelper` to help with load source data to simplify public notebooks.
17
+ - Registry: Option to `enable_explainability` when registering XGBoost models as a pre-PuPr feature.
18
+ - Feature Store: add new API `update_entity()`.
19
+ - Registry: Option to `enable_explainability` when registering Catboost models as a pre-PuPr feature.
20
+ - Feature Store: Add new argument warehouse to FeatureView constructor to overwrite the default warehouse. Also add
21
+ a new column 'warehouse' to the output of list_feature_views().
22
+ - Registry: Add support for logging model from a model version.
23
+ - Modeling: Distributed Hyperparameter Optimization now announce GA refresh version. The latest memory efficient version
24
+ will not have the 10GB training limitation for dataset any more. To turn off, please run
25
+ `
26
+ from snowflake.ml.modeling._internal.snowpark_implementations import (
27
+ distributed_hpo_trainer,
28
+ )
29
+ distributed_hpo_trainer.ENABLE_EFFICIENT_MEMORY_USAGE = False
30
+ `
31
+ - Registry: Option to `enable_explainability` when registering LightGBM models as a pre-PuPr feature.
32
+
33
+ ### Behavior Changes
34
+
35
+ - Feature Store: change some positional parameters to keyword arguments in following APIs:
36
+ - Entity(): desc.
37
+ - FeatureView(): timestamp_col, refresh_freq, desc.
38
+ - FeatureStore(): creation_mode.
39
+ - update_entity(): desc.
40
+ - register_feature_view(): block, overwrite.
41
+ - list_feature_views(): entity_name, feature_view_name.
42
+ - get_refresh_history(): verbose.
43
+ - retrieve_feature_values(): spine_timestamp_col, exclude_columns, include_feature_view_timestamp_col.
44
+ - generate_training_set(): save_as, spine_timestamp_col, spine_label_cols, exclude_columns,
45
+ include_feature_view_timestamp_col.
46
+ - generate_dataset(): version, spine_timestamp_col, spine_label_cols, exclude_columns,
47
+ include_feature_view_timestamp_col, desc, output_type.
48
+
49
+ ## 1.5.4 (2024-07-11)
50
+
51
+ ### Bug Fixes
52
+
53
+ - Model Registry (PrPr): Fix 401 Unauthorized issue when deploying model to SPCS.
54
+ - Feature Store: Downgrades exceptions to warnings for few property setters in feature view. Now you can set
55
+ desc, refresh_freq and warehouse for draft feature views.
56
+ - Modeling: Fix an issue with calling `OrdinalEncoder` with `categories` as a dictionary and a pandas DataFrame
57
+ - Modeling: Fix an issue with calling `OneHotEncoder` with `categories` as a dictionary and a pandas DataFrame
58
+
59
+ ### New Features
60
+
61
+ - Registry: Allow overriding `device_map` and `device` when loading huggingface pipeline models.
62
+ - Registry: Add `set_alias` method to `ModelVersion` instance to set an alias to model version.
63
+ - Registry: Add `unset_alias` method to `ModelVersion` instance to unset an alias to model version.
64
+ - Registry: Add `partitioned_inference_api` allowing users to create partitioned inference functions in registered
65
+ models. Enable model inference methods with table functions with vectorized process methods in registered models.
66
+ - Feature Store: add 3 more columns: refresh_freq, refresh_mode and scheduling_state to the result of
67
+ `list_feature_views()`.
68
+ - Feature Store: `update_feature_view()` supports updating description.
69
+ - Feature Store: add new API `refresh_feature_view()`.
70
+ - Feature Store: add new API `get_refresh_history()`.
71
+ - Feature Store: Add `generate_training_set()` API for generating table-backed feature snapshots.
72
+ - Feature Store: Add `DeprecationWarning` for `generate_dataset(..., output_type="table")`.
73
+ - Feature Store: `update_feature_view()` supports updating description.
74
+ - Feature Store: add new API `refresh_feature_view()`.
75
+ - Feature Store: add new API `get_refresh_history()`.
76
+ - Model Development: OrdinalEncoder supports a list of array-likes for `categories` argument.
77
+ - Model Development: OneHotEncoder supports a list of array-likes for `categories` argument.
78
+
79
+ ## 1.5.3 (06-17-2024)
80
+
81
+ ### Bug Fixes
82
+
83
+ - Modeling: Fix an issue causing lineage information to be missing for
84
+ `Pipeline`, `GridSearchCV` , `SimpleImputer`, and `RandomizedSearchCV`
85
+ - Registry: Fix an issue that leads to incorrect result when using pandas Dataframe with over 100, 000 rows as the input
86
+ of `ModelVersion.run` method in Stored Procedure.
87
+
88
+ ### New Features
89
+
90
+ - Registry: Add support for TIMESTAMP_NTZ model signature data type, allowing timestamp input and output.
91
+ - Dataset: Add `DatasetVersion.label_cols` and `DatasetVersion.exclude_cols` properties.
92
+
93
+ ## 1.5.2 (06-10-2024)
94
+
95
+ ### Bug Fixes
96
+
97
+ - Registry: Fix an issue that leads to unable to log model in store procedure.
98
+ - Modeling: Quick fix `import snowflake.ml.modeling.parameters.enable_anonymous_sproc` cannot be imported due to package
99
+ dependency error.
100
+
101
+ ### Behavior Changes
102
+
103
+ ### New Features
104
+
105
+ ## 1.5.1 (05-22-2024)
106
+
107
+ ### Bug Fixes
108
+
109
+ - Dataset: Fix `snowflake.connector.errors.DataError: Query Result did not match expected number of rows` when accessing
110
+ DatasetVersion properties when case insensitive `SHOW VERSIONS IN DATASET` check matches multiple version names.
111
+ - Dataset: Fix bug in SnowFS bulk file read when used with DuckDB
112
+ - Registry: Fixed a bug when loading old models.
113
+ - Lineage: Fix Dataset source lineage propagation through `snowpark.DataFrame` transformations
114
+
115
+ ### Behavior Changes
116
+
117
+ - Feature Store: convert clear() into a private function. Also make it deletes feature views and entities only.
118
+ - Feature Store: Use NULL as default value for timestamp tag value.
119
+
120
+ ### New Features
121
+
122
+ - Feature Store: Added new `snowflake.ml.feature_store.setup_feature_store()` API to assist Feature Store RBAC setup.
123
+ - Feature Store: Add `output_type` argument to `FeatureStore.generate_dataset()` to allow generating data snapshots
124
+ as Datasets or Tables.
125
+ - Registry: `log_model`, `get_model`, `delete_model` now supports fully qualified name.
126
+ - Modeling: Supports anonymous stored procedure during fit calls so that modeling would not require sufficient
127
+ permissions to operate on schema. Please call
128
+ `import snowflake.ml.modeling.parameters.enable_anonymous_sproc # noqa: F401`
129
+
130
+ ## 1.5.0 (05-01-2024)
131
+
132
+ ### Bug Fixes
133
+
134
+ - Registry: Fix invalid parameter 'SHOW_MODEL_DETAILS_IN_SHOW_VERSIONS_IN_MODEL' error.
135
+
136
+ ### Behavior Changes
137
+
138
+ - Model Development: The behavior of `fit_transform` for all estimators is changed.
139
+ Firstly, it will cover all the estimator that contains this function,
140
+ secondly, the output would be the union of pandas DataFrame and snowpark DataFrame.
141
+
142
+ #### Model Registry (PrPr)
143
+
144
+ `snowflake.ml.registry.artifact` and related `snowflake.ml.model_registry.ModelRegistry` APIs have been removed.
145
+
146
+ - Removed `snowflake.ml.registry.artifact` module.
147
+ - Removed `ModelRegistry.log_artifact()`, `ModelRegistry.list_artifacts()`, `ModelRegistry.get_artifact()`
148
+ - Removed `artifacts` argument from `ModelRegistry.log_model()`
149
+
150
+ #### Dataset (PrPr)
151
+
152
+ `snowflake.ml.dataset.Dataset` has been redesigned to be backed by Snowflake Dataset entities.
153
+
154
+ - New `Dataset`s can be created with `Dataset.create()` and existing `Dataset`s may be loaded
155
+ with `Dataset.load()`.
156
+ - `Dataset`s now maintain an immutable `selected_version` state. The `Dataset.create_version()` and
157
+ `Dataset.load_version()` APIs return new `Dataset` objects with the requested `selected_version` state.
158
+ - Added `dataset.create_from_dataframe()` and `dataset.load_dataset()` convenience APIs as a shortcut
159
+ to creating and loading `Dataset`s with a pre-selected version.
160
+ - `Dataset.materialized_table` and `Dataset.snapshot_table` no longer exist with `Dataset.fully_qualified_name`
161
+ as the closest equivalent.
162
+ - `Dataset.df` no longer exists. Instead, use `DatasetReader.read.to_snowpark_dataframe()`.
163
+ - `Dataset.owner` has been moved to `Dataset.selected_version.owner`
164
+ - `Dataset.desc` has been moved to `DatasetVersion.selected_version.comment`
165
+ - `Dataset.timestamp_col`, `Dataset.label_cols`, `Dataset.feature_store_metadata`, and
166
+ `Dataset.schema_version` have been removed.
167
+
168
+ #### Feature Store (PrPr)
169
+
170
+ - `FeatureStore.generate_dataset` argument list has been changed to match the new
171
+ `snowflake.ml.dataset.Dataset` definition
172
+
173
+ - `materialized_table` has been removed and replaced with `name` and `version`.
174
+ - `name` moved to first positional argument
175
+ - `save_mode` has been removed as `merge` behavior is no longer supported. The new behavior is always `errorifexists`.
176
+
177
+ - Change feature view version type from str to `FeatureViewVersion`. It is a restricted string literal.
178
+
179
+ - Remove as_dataframe arg from FeatureStore.list_feature_views(), now always returns result as DataFrame.
180
+
181
+ - Combines few metadata tags into a new tag: SNOWML_FEATURE_VIEW_METADATA. This will make previously created feature views
182
+ not readable by new SDK.
183
+
184
+ ### New Features
185
+
186
+ - Registry: Add `export` method to `ModelVersion` instance to export model files.
187
+ - Registry: Add `load` method to `ModelVersion` instance to load the underlying object from the model.
188
+ - Registry: Add `Model.rename` method to `Model` instance to rename or move a model.
189
+
190
+ #### Dataset (PrPr)
191
+
192
+ - Added Snowpark DataFrame integration using `Dataset.read.to_snowpark_dataframe()`
193
+ - Added Pandas DataFrame integration using `Dataset.read.to_pandas()`
194
+ - Added PyTorch and TensorFlow integrations using `Dataset.read.to_torch_datapipe()`
195
+ and `Dataset.read.to_tf_dataset()` respectively.
196
+ - Added `fsspec` style file integration using `Dataset.read.files()` and `Dataset.read.filesystem()`
197
+
198
+ #### Feature Store
199
+
200
+ - use new tag_reference_internal to speed up metadata lookup.
201
+
202
+ ## 1.4.1 (2024-04-18)
203
+
204
+ ### New Features
205
+
206
+ - Registry: Add support for `catboost` model (`catboost.CatBoostClassifier`, `catboost.CatBoostRegressor`).
207
+ - Registry: Add support for `lightgbm` model (`lightgbm.Booster`, `lightgbm.LightGBMClassifier`, `lightgbm.LightGBMRegressor`).
208
+
209
+ ### Bug Fixes
210
+
211
+ - Registry: Fix a bug that leads to relax_version option is not working.
212
+
213
+ ### Behavior changes
214
+
215
+ - Feature Store: update_feature_view takes refresh_freq and warehouse as argument.
216
+
217
+ ## 1.4.0 (2024-04-08)
218
+
219
+ ### Bug Fixes
220
+
221
+ - Registry: Fix a bug when multiple models are being called from the same query, models other than the first one will
222
+ have incorrect result. This fix only works for newly logged model.
223
+ - Modeling: When registering a model, only method(s) that is mentioned in `save_model` would be added to model signature
224
+ in SnowML models.
225
+ - Modeling: Fix a bug that when n_jobs is not 1, model cannot execute methods such as
226
+ predict, predict_log_proba, and other batch inference methods. The n_jobs would automatically
227
+ set to 1 because vectorized udf currently doesn't support joblib parallel backend.
228
+ - Modeling: Fix a bug that batch inference methods cannot infer the datatype when the first row of data contains NULL.
229
+ - Modeling: Matches Distributed HPO output column names with the snowflake identifier.
230
+ - Modeling: Relax package versions for all Distributed HPO methods if the installed version
231
+ is not available in the Snowflake conda channel
232
+ - Modeling: Add sklearn as required dependency for LightGBM package.
233
+
234
+ ### Behavior Changes
235
+
236
+ - Registry: `apply` method is no longer by default logged when logging a xgboost model. If that is required, it could
237
+ be specified manually when logging the model by `log_model(..., options={"target_methods": ["apply", ...]})`.
238
+ - Feature Store: register_entity returns an entity object.
239
+ - Feature Store: register_feature_view `block=true` becomes default.
240
+
241
+ ### New Features
242
+
243
+ - Registry: Add support for `sentence-transformers` model (`sentence_transformers.SentenceTransformer`).
244
+ - Registry: Now version name is no longer required when logging a model. If not provided, a random human readable ID
245
+ will be generated.
246
+
247
+ ## 1.3.1 (2024-03-21)
248
+
249
+ ### New Features
250
+
251
+ - FileSet: `snowflake.ml.fileset.sfcfs.SFFileSystem` can now be used in UDFs and stored procedures.
252
+
253
+ ## 1.3.0 (2024-03-12)
254
+
255
+ ### Bug Fixes
256
+
257
+ - Registry: Fix a bug that leads to module in `code_paths` when `log_model` cannot be correctly imported.
258
+ - Registry: Fix incorrect error message when validating input Snowpark DataFrame with array feature.
259
+ - Model Registry: Fix an issue when deploying a model to SPCS that some files do not have proper permission.
260
+ - Model Development: Relax package versions for all inference methods if the installed version
261
+ is not available in the Snowflake conda channel
262
+
263
+ ### Behavior Changes
264
+
265
+ - Registry: When running the method of a model, the value range based input validation to avoid input from overflowing
266
+ is now optional rather than enforced, this should improve the performance and should not lead to problem for most
267
+ kinds of model. If you want to enable this check as previous, specify `strict_input_validation=True` when
268
+ calling `run`.
269
+ - Registry: By default `relax_version=True` when logging a model instead of using the specific local dependency versions.
270
+ This improves dependency versioning by using versions available in Snowflake. To switch back to the previous behavior
271
+ and use specific local dependency versions, specify `relax_version=False` when calling `log_model`.
272
+ - Model Development: The behavior of `fit_predict` for all estimators is changed.
273
+ Firstly, it will cover all the estimator that contains this function,
274
+ secondly, the output would be the union of pandas DataFrame and snowpark DataFrame.
275
+
276
+ ### New Features
277
+
278
+ - FileSet: `snowflake.ml.fileset.sfcfs.SFFileSystem` can now be serialized with `pickle`.
279
+
280
+ ## 1.2.3 (2024-02-26)
281
+
282
+ ### Bug Fixes
283
+
284
+ - Registry: Now when providing Decimal Type column to a DOUBLE or FLOAT feature will not error out but auto cast with
285
+ warnings.
286
+ - Registry: Improve the error message when specifying currently unsupported `pip_requirements` argument.
287
+ - Model Development: Fix precision_recall_fscore_support incorrect results when `average="samples"`.
288
+ - Model Registry: Fix an issue that leads to description, metrics or tags are not correctly returned in newly created
289
+ Model Registry (PrPr) due to Snowflake BCR [2024_01](https://docs.snowflake.com/en/release-notes/bcr-bundles/2024_01/bcr-1483)
290
+
291
+ ### Behavior Changes
292
+
293
+ - Feature Store: `FeatureStore.suspend_feature_view` and `FeatureStore.resume_feature_view` doesn't mutate input feature
294
+ view argument any more. The updated status only reflected in the returned feature view object.
295
+
296
+ ### New Features
297
+
298
+ - Model Development: support `score_samples` method for all the classes, including Pipeline,
299
+ GridSearchCV, RandomizedSearchCV, PCA, IsolationForest, ...
300
+ - Registry: Support deleting a version of a model.
301
+
302
+ ## 1.2.2 (2024-02-13)
303
+
304
+ ### New Features
305
+
306
+ - Model Registry: Support providing external access integrations when deploying a model to SPCS. This will help and be
307
+ required to make sure the deploying process work as long as SPCS will by default deny all network connections. The
308
+ following endpoints must be allowed to make deployment work: docker.com:80, docker.com:443, anaconda.com:80,
309
+ anaconda.com:443, anaconda.org:80, anaconda.org:443, pypi.org:80, pypi.org:443. If you are using
310
+ `snowflake.ml.model.models.huggingface_pipeline.HuggingFacePipelineModel` object, the following endpoints are required
311
+ to be allowed: huggingface.com:80, huggingface.com:443, huggingface.co:80, huggingface.co:443.
312
+
313
+ ## 1.2.1 (2024-01-25)
314
+
315
+ ### New Features
316
+
317
+ - Model Development: Infers output column data type for transformers when possible.
318
+ - Registry: `relax_version` option is available in the `options` argument when logging the model.
319
+
320
+ ## 1.2.0 (2024-01-11)
321
+
322
+ ### Bug Fixes
323
+
324
+ - Model Registry: Fix "XGBoost version not compiled with GPU support" error when running CPU inference against open-source
325
+ XGBoost models deployed to SPCS.
326
+ - Model Registry: Fix model deployment to SPCS on Windows machines.
327
+
328
+ ### New Features
329
+
330
+ - Model Development: Introduced XGBoost external memory training feature. This feature enables training XGBoost models
331
+ on large datasets that don't fit into memory.
332
+ - Registry: New Registry class named `snowflake.ml.registry.Registry` providing similar APIs as the old one but works
333
+ with new MODEL object in Snowflake SQL. Also, we are providing`snowflake.ml.model.Model` and
334
+ `snowflake.ml.model.ModelVersion` to represent a model and a specific version of a model.
335
+ - Model Development: Add support for `fit_predict` method in `AgglomerativeClustering`, `DBSCAN`, and `OPTICS` classes;
336
+ - Model Development: Add support for `fit_transform` method in `MDS`, `SpectralEmbedding` and `TSNE` class.
337
+
338
+ ### Additional Notes
339
+
340
+ - Model Registry: The `snowflake.ml.registry.model_registry.ModelRegistry` has been deprecated starting from version
341
+ 1.2.0. It will stay in the Private Preview phase. For future implementations, kindly utilize
342
+ `snowflake.ml.registry.Registry`, except when specifically required. The old model registry will be removed once all
343
+ its primary functionalities are fully integrated into the new registry.
344
+
345
+ ## 1.1.2 (2023-12-18)
346
+
347
+ ### Bug Fixes
348
+
349
+ - Generic: Fix the issue that stack trace is hidden by telemetry unexpectedly.
350
+ - Model Development: Execute model signature inference without materializing full dataframe in memory.
351
+ - Model Registry: Fix occasional 'snowflake-ml-python library does not exist' error when deploying to SPCS.
352
+
353
+ ### Behavior Changes
354
+
355
+ - Model Registry: When calling `predict` with Snowpark DataFrame, both inferred or normalized column names are accepted.
356
+ - Model Registry: When logging a Snowpark ML Modeling Model, sample input data or manually provided signature will be
357
+ ignored since they are not necessary.
358
+
359
+ ### New Features
360
+
361
+ - Model Development: SQL implementation of binary `precision_score` metric.
362
+
363
+ ## 1.1.1 (2023-12-05)
364
+
365
+ ### Bug Fixes
366
+
367
+ - Model Registry: The `predict` target method on registered models is now compatible with unsupervised estimators.
368
+ - Model Development: Fix confusion_matrix incorrect results when the row number cannot be divided by the batch size.
369
+
370
+ ### New Features
371
+
372
+ - Introduced passthrough_col param in Modeling API. This new param is helpful in scenarios
373
+ requiring automatic input_cols inference, but need to avoid using specific
374
+ columns, like index columns, during training or inference.
375
+
376
+ ## 1.1.0 (2023-12-01)
377
+
378
+ ### Bug Fixes
379
+
380
+ - Model Registry: Fix panda dataframe input not handling first row properly.
381
+ - Model Development: OrdinalEncoder and LabelEncoder output_columns do not need to be valid snowflake identifiers. They
382
+ would previously be excluded if the normalized name did not match the name specified in output_columns.
383
+
384
+ ### New Features
385
+
386
+ - Model Registry: Add support for invoking public endpoint on SPCS service, by providing a "enable_ingress" SPCS
387
+ deployment option.
388
+ - Model Development: Add support for distributed HPO - GridSearchCV and RandomizedSearchCV execution will be
389
+ distributed on multi-node warehouses.
390
+
391
+ ## 1.0.12 (2023-11-13)
392
+
393
+ ### Bug Fixes
394
+
395
+ - Model Registry: Fix regression issue that container logging is not shown during model deployment to SPCS.
396
+ - Model Development: Enhance the column capacity of OrdinalEncoder.
397
+ - Model Registry: Fix unbound `batch_size` error when deploying a model other than Hugging Face Pipeline
398
+ and LLM with GPU on SPCS.
399
+
400
+ ### Behavior Changes
401
+
402
+ - Model Registry: Raise early error when deploying to SPCS with db/schema that starts with underscore.
403
+ - Model Registry: `conda-forge` channel is now automatically added to channel lists when deploying to SPCS.
404
+ - Model Registry: `relax_version` will not strip all version specifier, instead it will relax `==x.y.z` specifier to
405
+ `>=x.y,<(x+1)`.
406
+ - Model Registry: Python with different patchlevel but the same major and minor will not result a warning when loading
407
+ the model via Model Registry and would be considered to use when deploying to SPCS.
408
+ - Model Registry: When logging a `snowflake.ml.model.models.huggingface_pipeline.HuggingFacePipelineModel` object,
409
+ versions of local installed libraries won't be picked as dependencies of models, instead it will pick up some pre-
410
+ defined dependencies to improve user experience.
411
+
412
+ ### New Features
413
+
414
+ - Model Registry: Enable best-effort SPCS job/service log streaming when logging level is set to INFO.
415
+
416
+ ## 1.0.11 (2023-10-27)
417
+
418
+ ### New Features
419
+
420
+ - Model Registry: Add log_artifact() public method.
421
+ - Model Development: Add support for `kneighbors`.
422
+
423
+ ### Behavior Changes
424
+
425
+ - Model Registry: Change log_model() argument from TrainingDataset to List of Artifact.
426
+ - Model Registry: Change get_training_dataset() to get_artifact().
427
+
428
+ ### Bug Fixes
429
+
430
+ - Model Development: Fix support for XGBoost and LightGBM models using SKLearn Grid Search and Randomized Search model selectors.
431
+ - Model Development: DecimalType is now supported as a DataType.
432
+ - Model Development: Fix metrics compatibility with Snowpark Dataframes that use Snowflake identifiers
433
+ - Model Registry: Resolve 'delete_deployment' not deleting the SPCS service in certain cases.
434
+
435
+ ## 1.0.10 (2023-10-13)
436
+
437
+ ### Behavior Changes
438
+
439
+ - Model Development: precision_score, recall_score, f1_score, fbeta_score, precision_recall_fscore_support,
440
+ mean_absolute_error, mean_squared_error, and mean_absolute_percentage_error metric calculations are now distributed.
441
+ - Model Registry: `deploy` will now return `Deployment` for deployment information.
442
+
443
+ ### New Features
444
+
445
+ - Model Registry: When the model signature is auto-inferred, it will be printed to the log for reference.
446
+ - Model Registry: For SPCS deployment, `Deployment` details will contains `image_name`, `service_spec` and `service_function_sql`.
447
+
448
+ ### Bug Fixes
449
+
450
+ - Model Development: Fix an issue that leading to UTF-8 decoding errors when using modeling modules on Windows.
451
+ - Model Development: Fix an issue that alias definitions cause `SnowparkSQLUnexpectedAliasException` in inference.
452
+ - Model Registry: Fix an issue that signature inference could be incorrect when using Snowpark DataFrame as sample input.
453
+ - Model Registry: Fix too strict data type validation when predicting. Now, for example, if you have a INT8
454
+ type feature in the signature, if providing a INT64 dataframe but all values are within the range, it would not fail.
455
+
456
+ ## 1.0.9 (2023-09-28)
457
+
458
+ ### Behavior Changes
459
+
460
+ - Model Development: log_loss metric calculation is now distributed.
461
+
462
+ ### Bug Fixes
463
+
464
+ - Model Registry: Fix an issue that building images fails with specific docker setup.
465
+ - Model Registry: Fix an issue that unable to embed local ML library when the library is imported by `zipimport`.
466
+ - Model Registry: Fix out-of-date doc about `platform` argument in the `deploy` function.
467
+ - Model Registry: Fix an issue that unable to deploy a GPU-trained PyTorch model to a platform where GPU is not available.
468
+
469
+ ## 1.0.8 (2023-09-15)
470
+
471
+ ### Bug Fixes
472
+
473
+ - Model Development: Ordinal encoder can be used with mixed input column types.
474
+ - Model Development: Fix an issue when the sklearn default value is `np.nan`.
475
+ - Model Registry: Fix an issue that incorrect docker executable is used when building images.
476
+ - Model Registry: Fix an issue that specifying `token` argument when using
477
+ `snowflake.ml.model.models.huggingface_pipeline.HuggingFacePipelineModel` with `transformers < 4.32.0` is not effective.
478
+ - Model Registry: Fix an issue that incorrect system function call is used when deploying to SPCS.
479
+ - Model Registry: Fix an issue when using a `transformers.pipeline` that does not have a `tokenizer`.
480
+ - Model Registry: Fix incorrectly-inferred image repository name during model deployment to SPCS.
481
+ - Model Registry: Fix GPU resource retention issue caused by failed or stuck previous deployments in SPCS.
482
+
483
+ ## 1.0.7 (2023-09-05)
484
+
485
+ ### Bug Fixes
486
+
487
+ - Model Development & Model Registry: Fix an error related to `pandas.io.json.json_normalize`.
488
+ - Allow disabling telemetry.
489
+
490
+ ## 1.0.6 (2023-09-01)
491
+
492
+ ### New Features
493
+
494
+ - Model Registry: add `create_if_not_exists` parameter in constructor.
495
+ - Model Registry: Added get_or_create_model_registry API.
496
+ - Model Registry: Added support for using GPU inference when deploying XGBoost (`xgboost.XGBModel` and `xgboost.Booster`
497
+ ), PyTorch (`torch.nn.Module` and `torch.jit.ScriptModule`) and TensorFlow (`tensorflow.Module` and
498
+ `tensorflow.keras.Model`) models to Snowpark Container Services.
499
+ - Model Registry: When inferring model signature, `Sequence` of built-in types, `Sequence` of `numpy.ndarray`,
500
+ `Sequence` of `torch.Tensor`, `Sequence` of `tensorflow.Tensor` and `Sequence` of `tensorflow.Tensor` can be used
501
+ instead of only `List` of them.
502
+ - Model Registry: Added `get_training_dataset` API.
503
+ - Model Development: Size of metrics result can exceed previous 8MB limit.
504
+ - Model Registry: Added support save/load/deploy HuggingFace pipeline object (`transformers.Pipeline`) and our wrapper
505
+ (`snowflake.ml.model.models.huggingface_pipeline.HuggingFacePipelineModel`) to it. Using the wrapper to specify
506
+ configurations and the model for the pipeline will be loaded dynamically when deploying. Currently, following tasks
507
+ are supported to log without manually specifying model signatures:
508
+ - "conversational"
509
+ - "fill-mask"
510
+ - "question-answering"
511
+ - "summarization"
512
+ - "table-question-answering"
513
+ - "text2text-generation"
514
+ - "text-classification" (alias "sentiment-analysis" available)
515
+ - "text-generation"
516
+ - "token-classification" (alias "ner" available)
517
+ - "translation"
518
+ - "translation_xx_to_yy"
519
+ - "zero-shot-classification"
520
+
521
+ ### Bug Fixes
522
+
523
+ - Model Development: Fixed a bug when using simple imputer with numpy >= 1.25.
524
+ - Model Development: Fixed a bug when inferring the type of label columns.
525
+
526
+ ### Behavior Changes
527
+
528
+ - Model Registry: `log_model()` now return a `ModelReference` object instead of a model ID.
529
+ - Model Registry: When deploying a model with 1 `target method` only, the `target_method` argument can be omitted.
530
+ - Model Registry: When using the snowflake-ml-python with version newer than what is available in Snowflake Anaconda
531
+ Channel, `embed_local_ml_library` option will be set as `True` automatically if not.
532
+ - Model Registry: When deploying a model to Snowpark Container Services and using GPU, the default value of num_workers
533
+ will be 1.
534
+ - Model Registry: `keep_order` and `output_with_input_features` in the deploy options have been removed. Now the
535
+ behavior is controlled by the type of the input when calling `model.predict()`. If the input is a `pandas.DataFrame`,
536
+ the behavior will be the same as `keep_order=True` and `output_with_input_features=False` before. If the input is a
537
+ `snowpark.DataFrame`, the behavior will be the same as `keep_order=False` and `output_with_input_features=True` before.
538
+ - Model Registry: When logging and deploying PyTorch (`torch.nn.Module` and `torch.jit.ScriptModule`) and TensorFlow
539
+ (`tensorflow.Module` and `tensorflow.keras.Model`) models, we no longer accept models whose input is a list of tensor
540
+ and output is a list of tensors. Instead, now we accept models whose input is 1 or more tensors as positional arguments,
541
+ and output is a tensor or a tuple of tensors. The input and output dataframe when predicting keep the same as before,
542
+ that is every column is an array feature and contains a tensor.
543
+
544
+ ## 1.0.5 (2023-08-17)
545
+
546
+ ### New Features
547
+
548
+ - Model Registry: Added support save/load/deploy xgboost Booster model.
549
+ - Model Registry: Added support to get the model name and the model version from model references.
550
+
551
+ ### Bug Fixes
552
+
553
+ - Model Registry: Restore the db/schema back to the session after `create_model_registry()`.
554
+ - Model Registry: Fixed an issue that the UDF name created when deploying a model is not identical to what is provided
555
+ and cannot be correctly dropped when deployment getting dropped.
556
+ - connection_params.SnowflakeLoginOptions(): Added support for `private_key_path`.
557
+
558
+ ## 1.0.4 (2023-07-28)
559
+
560
+ ### New Features
561
+
562
+ - Model Registry: Added support save/load/deploy Tensorflow models (`tensorflow.Module`).
563
+ - Model Registry: Added support save/load/deploy MLFlow PyFunc models (`mlflow.pyfunc.PyFuncModel`).
564
+ - Model Development: Input dataframes can now be joined against data loaded from staged files.
565
+ - Model Development: Added support for non-English languages.
566
+
567
+ ### Bug Fixes
568
+
569
+ - Model Registry: Fix an issue that model dependencies are incorrectly reported as unresolvable on certain platforms.
570
+
571
+ ## 1.0.3 (2023-07-14)
572
+
573
+ ### Behavior Changes
574
+
575
+ - Model Registry: When predicting a model whose output is a list of NumPy ndarray, the output would not be flattened,
576
+ instead, every ndarray will act as a feature(column) in the output.
577
+
578
+ ### New Features
579
+
580
+ - Model Registry: Added support save/load/deploy PyTorch models (`torch.nn.Module` and `torch.jit.ScriptModule`).
581
+
582
+ ### Bug Fixes
583
+
584
+ - Model Registry: Fix an issue that when database or schema name provided to `create_model_registry` contains special
585
+ characters, the model registry cannot be created.
586
+ - Model Registry: Fix an issue that `get_model_description` returns with additional quotes.
587
+ - Model Registry: Fix incorrect error message when attempting to remove a unset tag of a model.
588
+ - Model Registry: Fix a typo in the default deployment table name.
589
+ - Model Registry: Snowpark dataframe for sample input or input for `predict` method that contains a column with
590
+ Snowflake `NUMBER(precision, scale)` data type where `scale = 0` will not lead to error, and will now correctly
591
+ recognized as `INT64` data type in model signature.
592
+ - Model Registry: Fix an issue that prevent model logged in the system whose default encoding is not UTF-8 compatible
593
+ from deploying.
594
+ - Model Registry: Added earlier and better error message when any file name in the model or the file name of model
595
+ itself contains characters that are unable to be encoded using ASCII. It is currently not supported to deploy such a
596
+ model.
597
+
598
+ ## 1.0.2 (2023-06-22)
599
+
600
+ ### Behavior Changes
601
+
602
+ - Model Registry: Prohibit non-snowflake-native models from being logged.
603
+ - Model Registry: `_use_local_snowml` parameter in options of `deploy()` has been removed.
604
+ - Model Registry: A default `False` `embed_local_ml_library` parameter has been added to the options of `log_model()`.
605
+ With this set to `False` (default), the version of the local snowflake-ml-python library will be recorded and used when
606
+ deploying the model. With this set to `True`, local snowflake-ml-python library will be embedded into the logged model,
607
+ and will be used when you load or deploy the model.
608
+
609
+ ### New Features
610
+
611
+ - Model Registry: A new optional argument named `code_paths` has been added to the arguments of `log_model()` for users
612
+ to specify additional code paths to be imported when loading and deploying the model.
613
+ - Model Registry: A new optional argument named `options` has been added to the arguments of `log_model()` to specify
614
+ any additional options when saving the model.
615
+ - Model Development: Added metrics:
616
+ - d2_absolute_error_score
617
+ - d2_pinball_score
618
+ - explained_variance_score
619
+ - mean_absolute_error
620
+ - mean_absolute_percentage_error
621
+ - mean_squared_error
622
+
623
+ ### Bug Fixes
624
+
625
+ - Model Development: `accuracy_score()` now works when given label column names are lists of a single value.
626
+
627
+ ## 1.0.1 (2023-06-16)
628
+
629
+ ### Behavior Changes
630
+
631
+ - Model Development: Changed Metrics APIs to imitate sklearn metrics modules:
632
+ - `accuracy_score()`, `confusion_matrix()`, `precision_recall_fscore_support()`, `precision_score()` methods move from
633
+ respective modules to `metrics.classification`.
634
+ - Model Registry: The default table/stage created by the Registry now uses "_SYSTEM_" as a prefix.
635
+ - Model Registry: `get_model_history()` method as been enhanced to include the history of model deployment.
636
+
637
+ ### New Features
638
+
639
+ - Model Registry: A default `False` flag named `replace_udf` has been added to the options of `deploy()`. Setting this
640
+ to `True` will allow overwrite existing UDF with the same name when deploying.
641
+ - Model Development: Added metrics:
642
+ - f1_score
643
+ - fbeta_score
644
+ - recall_score
645
+ - roc_auc_score
646
+ - roc_curve
647
+ - log_loss
648
+ - precision_recall_curve
649
+ - Model Registry: A new argument named `permanent` has been added to the argument of `deploy()`. Setting this to `True`
650
+ allows the creation of a permanent deployment without needing to specify the UDF location.
651
+ - Model Registry: A new method `list_deployments()` has been added to enumerate all permanent deployments originating
652
+ from a specific model.
653
+ - Model Registry: A new method `get_deployment()` has been added to fetch a deployment by its deployment name.
654
+ - Model Registry: A new method `delete_deployment()` has been added to remove an existing permanent deployment.
655
+
656
+ ## 1.0.0 (2023-06-09)
657
+
658
+ ### Behavior Changes
659
+
660
+ - Model Registry: `predict()` method moves from Registry to ModelReference.
661
+ - Model Registry: `_snowml_wheel_path` parameter in options of `deploy()`, is replaced with `_use_local_snowml` with
662
+ default value of `False`. Setting this to `True` will have the same effect of uploading local SnowML code when executing
663
+ model in the warehouse.
664
+ - Model Registry: Removed `id` field from `ModelReference` constructor.
665
+ - Model Development: Preprocessing and Metrics move to the modeling package: `snowflake.ml.modeling.preprocessing` and
666
+ `snowflake.ml.modeling.metrics`.
667
+ - Model Development: `get_sklearn_object()` method is renamed to `to_sklearn()`, `to_xgboost()`, and `to_lightgbm()` for
668
+ respective native models.
669
+
670
+ ### New Features
671
+
672
+ - Added PolynomialFeatures transformer to the snowflake.ml.modeling.preprocessing module.
673
+ - Added metrics:
674
+ - accuracy_score
675
+ - confusion_matrix
676
+ - precision_recall_fscore_support
677
+ - precision_score
678
+
679
+ ### Bug Fixes
680
+
681
+ - Model Registry: Model version can now be any string (not required to be a valid identifier)
682
+ - Model Deployment: `deploy()` & `predict()` methods now correctly escapes identifiers
683
+
684
+ ## 0.3.2 (2023-05-23)
685
+
686
+ ### Behavior Changes
687
+
688
+ - Use cloudpickle to serialize and deserialize models throughout the codebase and removed dependency on joblib.
689
+
690
+ ### New Features
691
+
692
+ - Model Deployment: Added support for snowflake.ml models.
693
+
694
+ ## 0.3.1 (2023-05-18)
695
+
696
+ ### Behavior Changes
697
+
698
+ - Standardized registry API with following
699
+ - Create & open registry taking same set of arguments
700
+ - Create & Open can choose schema to use
701
+ - Set_tag, set_metric, etc now explicitly calls out arg name as metric_name, tag_name, metric_name, etc.
702
+
703
+ ### New Features
704
+
705
+ - Changes to support python 3.9, 3.10
706
+ - Added kBinsDiscretizer
707
+ - Support for deployment of XGBoost models & int8 types of data
708
+
709
+ ## 0.3.0 (2023-05-11)
710
+
711
+ ### Behavior Changes
712
+
713
+ - Big Model Registry Refresh
714
+ - Fixed API discrepancies between register_model & log_model.
715
+ - Model can be referred by Name + Version (no opaque internal id is required)
716
+
717
+ ### New Features
718
+
719
+ - Model Registry: Added support save/load/deploy SKL & XGB Models
720
+
721
+ ## 0.2.3 (2023-04-27)
722
+
723
+ ### Bug Fixes
724
+
725
+ - Allow using OneHotEncoder along with sklearn style estimators in a pipeline.
726
+
727
+ ### New Features
728
+
729
+ - Model Registry: Added support for delete_model. Use delete_artifact = False to not delete the underlying model data
730
+ but just unregister.
731
+
732
+ ## 0.2.2 (2023-04-11)
733
+
734
+ ### New Features
735
+
736
+ - Initial version of snowflake-ml modeling package.
737
+ - Provide support for training most of scikit-learn and xgboost estimators and transformers.
738
+
739
+ ### Bug Fixes
740
+
741
+ - Minor fixes in preprocessing package.
742
+
743
+ ## 0.2.1 (2023-03-23)
744
+
745
+ ### New Features
746
+
747
+ - New in Preprocessing:
748
+ - SimpleImputer
749
+ - Covariance Matrix
750
+ - Optimization of Ordinal Encoder client computations.
751
+
752
+ ### Bug Fixes
753
+
754
+ - Minor fixes in OneHotEncoder.
755
+
756
+ ## 0.2.0 (2023-02-27)
757
+
758
+ ### New Features
759
+
760
+ - Model Registry
761
+ - PyTorch & Tensorflow connector file generic FileSet API
762
+ - New to Preprocessing:
763
+ - Binarizer
764
+ - Normalizer
765
+ - Pearson correlation Matrix
766
+ - Optimization in Ordinal Encoder to cache vocabulary in temp tables.
767
+
768
+ ## 0.1.3 (2023-02-02)
769
+
770
+ ### New Features
771
+
772
+ - Initial version of transformers including:
773
+ - Label Encoder
774
+ - Max Abs Scaler
775
+ - Min Max Scaler
776
+ - One Hot Encoder
777
+ - Ordinal Encoder
778
+ - Robust Scaler
779
+ - Standard Scaler