oracle-ads 2.13.8__py3-none-any.whl → 2.13.9rc0__py3-none-any.whl
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.
- {oracle_ads-2.13.8.dist-info → oracle_ads-2.13.9rc0.dist-info}/METADATA +151 -151
- oracle_ads-2.13.9rc0.dist-info/RECORD +9 -0
- {oracle_ads-2.13.8.dist-info → oracle_ads-2.13.9rc0.dist-info}/WHEEL +2 -1
- {oracle_ads-2.13.8.dist-info → oracle_ads-2.13.9rc0.dist-info}/entry_points.txt +1 -2
- oracle_ads-2.13.9rc0.dist-info/top_level.txt +1 -0
- ads/aqua/__init__.py +0 -40
- ads/aqua/app.py +0 -506
- ads/aqua/cli.py +0 -96
- ads/aqua/client/__init__.py +0 -3
- ads/aqua/client/client.py +0 -836
- ads/aqua/client/openai_client.py +0 -305
- ads/aqua/common/__init__.py +0 -5
- ads/aqua/common/decorator.py +0 -125
- ads/aqua/common/entities.py +0 -266
- ads/aqua/common/enums.py +0 -122
- ads/aqua/common/errors.py +0 -109
- ads/aqua/common/utils.py +0 -1285
- ads/aqua/config/__init__.py +0 -4
- ads/aqua/config/container_config.py +0 -248
- ads/aqua/config/evaluation/__init__.py +0 -4
- ads/aqua/config/evaluation/evaluation_service_config.py +0 -147
- ads/aqua/config/utils/__init__.py +0 -4
- ads/aqua/config/utils/serializer.py +0 -339
- ads/aqua/constants.py +0 -114
- ads/aqua/data.py +0 -14
- ads/aqua/dummy_data/icon.txt +0 -1
- ads/aqua/dummy_data/oci_model_deployments.json +0 -56
- ads/aqua/dummy_data/oci_models.json +0 -1
- ads/aqua/dummy_data/readme.md +0 -26
- ads/aqua/evaluation/__init__.py +0 -8
- ads/aqua/evaluation/constants.py +0 -53
- ads/aqua/evaluation/entities.py +0 -186
- ads/aqua/evaluation/errors.py +0 -70
- ads/aqua/evaluation/evaluation.py +0 -1814
- ads/aqua/extension/__init__.py +0 -42
- ads/aqua/extension/aqua_ws_msg_handler.py +0 -76
- ads/aqua/extension/base_handler.py +0 -90
- ads/aqua/extension/common_handler.py +0 -121
- ads/aqua/extension/common_ws_msg_handler.py +0 -36
- ads/aqua/extension/deployment_handler.py +0 -298
- ads/aqua/extension/deployment_ws_msg_handler.py +0 -54
- ads/aqua/extension/errors.py +0 -30
- ads/aqua/extension/evaluation_handler.py +0 -129
- ads/aqua/extension/evaluation_ws_msg_handler.py +0 -61
- ads/aqua/extension/finetune_handler.py +0 -96
- ads/aqua/extension/model_handler.py +0 -390
- ads/aqua/extension/models/__init__.py +0 -0
- ads/aqua/extension/models/ws_models.py +0 -145
- ads/aqua/extension/models_ws_msg_handler.py +0 -50
- ads/aqua/extension/ui_handler.py +0 -282
- ads/aqua/extension/ui_websocket_handler.py +0 -130
- ads/aqua/extension/utils.py +0 -133
- ads/aqua/finetuning/__init__.py +0 -7
- ads/aqua/finetuning/constants.py +0 -23
- ads/aqua/finetuning/entities.py +0 -181
- ads/aqua/finetuning/finetuning.py +0 -731
- ads/aqua/model/__init__.py +0 -8
- ads/aqua/model/constants.py +0 -60
- ads/aqua/model/entities.py +0 -306
- ads/aqua/model/enums.py +0 -30
- ads/aqua/model/model.py +0 -2080
- ads/aqua/modeldeployment/__init__.py +0 -8
- ads/aqua/modeldeployment/constants.py +0 -10
- ads/aqua/modeldeployment/deployment.py +0 -1324
- ads/aqua/modeldeployment/entities.py +0 -653
- ads/aqua/modeldeployment/inference.py +0 -74
- ads/aqua/modeldeployment/utils.py +0 -543
- ads/aqua/resources/gpu_shapes_index.json +0 -94
- ads/aqua/server/__init__.py +0 -4
- ads/aqua/server/__main__.py +0 -24
- ads/aqua/server/app.py +0 -47
- ads/aqua/server/aqua_spec.yml +0 -1291
- ads/aqua/training/__init__.py +0 -4
- ads/aqua/training/exceptions.py +0 -476
- ads/aqua/ui.py +0 -499
- ads/automl/__init__.py +0 -9
- ads/automl/driver.py +0 -330
- ads/automl/provider.py +0 -975
- ads/bds/__init__.py +0 -5
- ads/bds/auth.py +0 -127
- ads/bds/big_data_service.py +0 -255
- ads/catalog/__init__.py +0 -19
- ads/catalog/model.py +0 -1576
- ads/catalog/notebook.py +0 -461
- ads/catalog/project.py +0 -468
- ads/catalog/summary.py +0 -178
- ads/common/__init__.py +0 -11
- ads/common/analyzer.py +0 -65
- ads/common/artifact/.model-ignore +0 -63
- ads/common/artifact/__init__.py +0 -10
- ads/common/auth.py +0 -1122
- ads/common/card_identifier.py +0 -83
- ads/common/config.py +0 -647
- ads/common/data.py +0 -165
- ads/common/decorator/__init__.py +0 -9
- ads/common/decorator/argument_to_case.py +0 -88
- ads/common/decorator/deprecate.py +0 -69
- ads/common/decorator/require_nonempty_arg.py +0 -65
- ads/common/decorator/runtime_dependency.py +0 -178
- ads/common/decorator/threaded.py +0 -97
- ads/common/decorator/utils.py +0 -35
- ads/common/dsc_file_system.py +0 -303
- ads/common/error.py +0 -14
- ads/common/extended_enum.py +0 -81
- ads/common/function/__init__.py +0 -5
- ads/common/function/fn_util.py +0 -142
- ads/common/function/func_conf.yaml +0 -25
- ads/common/ipython.py +0 -76
- ads/common/model.py +0 -679
- ads/common/model_artifact.py +0 -1759
- ads/common/model_artifact_schema.json +0 -107
- ads/common/model_export_util.py +0 -664
- ads/common/model_metadata.py +0 -24
- ads/common/object_storage_details.py +0 -296
- ads/common/oci_client.py +0 -175
- ads/common/oci_datascience.py +0 -46
- ads/common/oci_logging.py +0 -1144
- ads/common/oci_mixin.py +0 -957
- ads/common/oci_resource.py +0 -136
- ads/common/serializer.py +0 -559
- ads/common/utils.py +0 -1852
- ads/common/word_lists.py +0 -1491
- ads/common/work_request.py +0 -189
- ads/data_labeling/__init__.py +0 -13
- ads/data_labeling/boundingbox.py +0 -253
- ads/data_labeling/constants.py +0 -47
- ads/data_labeling/data_labeling_service.py +0 -244
- ads/data_labeling/interface/__init__.py +0 -5
- ads/data_labeling/interface/loader.py +0 -16
- ads/data_labeling/interface/parser.py +0 -16
- ads/data_labeling/interface/reader.py +0 -23
- ads/data_labeling/loader/__init__.py +0 -5
- ads/data_labeling/loader/file_loader.py +0 -241
- ads/data_labeling/metadata.py +0 -110
- ads/data_labeling/mixin/__init__.py +0 -5
- ads/data_labeling/mixin/data_labeling.py +0 -232
- ads/data_labeling/ner.py +0 -129
- ads/data_labeling/parser/__init__.py +0 -5
- ads/data_labeling/parser/dls_record_parser.py +0 -388
- ads/data_labeling/parser/export_metadata_parser.py +0 -94
- ads/data_labeling/parser/export_record_parser.py +0 -473
- ads/data_labeling/reader/__init__.py +0 -5
- ads/data_labeling/reader/dataset_reader.py +0 -574
- ads/data_labeling/reader/dls_record_reader.py +0 -121
- ads/data_labeling/reader/export_record_reader.py +0 -62
- ads/data_labeling/reader/jsonl_reader.py +0 -75
- ads/data_labeling/reader/metadata_reader.py +0 -203
- ads/data_labeling/reader/record_reader.py +0 -263
- ads/data_labeling/record.py +0 -52
- ads/data_labeling/visualizer/__init__.py +0 -5
- ads/data_labeling/visualizer/image_visualizer.py +0 -525
- ads/data_labeling/visualizer/text_visualizer.py +0 -357
- ads/database/__init__.py +0 -5
- ads/database/connection.py +0 -338
- ads/dataset/__init__.py +0 -10
- ads/dataset/capabilities.md +0 -51
- ads/dataset/classification_dataset.py +0 -339
- ads/dataset/correlation.py +0 -226
- ads/dataset/correlation_plot.py +0 -563
- ads/dataset/dask_series.py +0 -173
- ads/dataset/dataframe_transformer.py +0 -110
- ads/dataset/dataset.py +0 -1979
- ads/dataset/dataset_browser.py +0 -360
- ads/dataset/dataset_with_target.py +0 -995
- ads/dataset/exception.py +0 -25
- ads/dataset/factory.py +0 -987
- ads/dataset/feature_engineering_transformer.py +0 -35
- ads/dataset/feature_selection.py +0 -107
- ads/dataset/forecasting_dataset.py +0 -26
- ads/dataset/helper.py +0 -1450
- ads/dataset/label_encoder.py +0 -99
- ads/dataset/mixin/__init__.py +0 -5
- ads/dataset/mixin/dataset_accessor.py +0 -134
- ads/dataset/pipeline.py +0 -58
- ads/dataset/plot.py +0 -710
- ads/dataset/progress.py +0 -86
- ads/dataset/recommendation.py +0 -297
- ads/dataset/recommendation_transformer.py +0 -502
- ads/dataset/regression_dataset.py +0 -14
- ads/dataset/sampled_dataset.py +0 -1050
- ads/dataset/target.py +0 -98
- ads/dataset/timeseries.py +0 -18
- ads/dbmixin/__init__.py +0 -5
- ads/dbmixin/db_pandas_accessor.py +0 -153
- ads/environment/__init__.py +0 -9
- ads/environment/ml_runtime.py +0 -66
- ads/evaluations/README.md +0 -14
- ads/evaluations/__init__.py +0 -109
- ads/evaluations/evaluation_plot.py +0 -983
- ads/evaluations/evaluator.py +0 -1334
- ads/evaluations/statistical_metrics.py +0 -543
- ads/experiments/__init__.py +0 -9
- ads/experiments/capabilities.md +0 -0
- ads/explanations/__init__.py +0 -21
- ads/explanations/base_explainer.py +0 -142
- ads/explanations/capabilities.md +0 -83
- ads/explanations/explainer.py +0 -190
- ads/explanations/mlx_global_explainer.py +0 -1050
- ads/explanations/mlx_interface.py +0 -386
- ads/explanations/mlx_local_explainer.py +0 -287
- ads/explanations/mlx_whatif_explainer.py +0 -201
- ads/feature_engineering/__init__.py +0 -20
- ads/feature_engineering/accessor/__init__.py +0 -5
- ads/feature_engineering/accessor/dataframe_accessor.py +0 -535
- ads/feature_engineering/accessor/mixin/__init__.py +0 -5
- ads/feature_engineering/accessor/mixin/correlation.py +0 -166
- ads/feature_engineering/accessor/mixin/eda_mixin.py +0 -266
- ads/feature_engineering/accessor/mixin/eda_mixin_series.py +0 -85
- ads/feature_engineering/accessor/mixin/feature_types_mixin.py +0 -211
- ads/feature_engineering/accessor/mixin/utils.py +0 -65
- ads/feature_engineering/accessor/series_accessor.py +0 -431
- ads/feature_engineering/adsimage/__init__.py +0 -5
- ads/feature_engineering/adsimage/image.py +0 -192
- ads/feature_engineering/adsimage/image_reader.py +0 -170
- ads/feature_engineering/adsimage/interface/__init__.py +0 -5
- ads/feature_engineering/adsimage/interface/reader.py +0 -19
- ads/feature_engineering/adsstring/__init__.py +0 -7
- ads/feature_engineering/adsstring/oci_language/__init__.py +0 -8
- ads/feature_engineering/adsstring/string/__init__.py +0 -8
- ads/feature_engineering/data_schema.json +0 -57
- ads/feature_engineering/dataset/__init__.py +0 -5
- ads/feature_engineering/dataset/zip_code_data.py +0 -42062
- ads/feature_engineering/exceptions.py +0 -40
- ads/feature_engineering/feature_type/__init__.py +0 -133
- ads/feature_engineering/feature_type/address.py +0 -184
- ads/feature_engineering/feature_type/adsstring/__init__.py +0 -5
- ads/feature_engineering/feature_type/adsstring/common_regex_mixin.py +0 -164
- ads/feature_engineering/feature_type/adsstring/oci_language.py +0 -93
- ads/feature_engineering/feature_type/adsstring/parsers/__init__.py +0 -5
- ads/feature_engineering/feature_type/adsstring/parsers/base.py +0 -47
- ads/feature_engineering/feature_type/adsstring/parsers/nltk_parser.py +0 -96
- ads/feature_engineering/feature_type/adsstring/parsers/spacy_parser.py +0 -221
- ads/feature_engineering/feature_type/adsstring/string.py +0 -258
- ads/feature_engineering/feature_type/base.py +0 -58
- ads/feature_engineering/feature_type/boolean.py +0 -183
- ads/feature_engineering/feature_type/category.py +0 -146
- ads/feature_engineering/feature_type/constant.py +0 -137
- ads/feature_engineering/feature_type/continuous.py +0 -151
- ads/feature_engineering/feature_type/creditcard.py +0 -314
- ads/feature_engineering/feature_type/datetime.py +0 -190
- ads/feature_engineering/feature_type/discrete.py +0 -134
- ads/feature_engineering/feature_type/document.py +0 -43
- ads/feature_engineering/feature_type/gis.py +0 -251
- ads/feature_engineering/feature_type/handler/__init__.py +0 -5
- ads/feature_engineering/feature_type/handler/feature_validator.py +0 -524
- ads/feature_engineering/feature_type/handler/feature_warning.py +0 -319
- ads/feature_engineering/feature_type/handler/warnings.py +0 -128
- ads/feature_engineering/feature_type/integer.py +0 -142
- ads/feature_engineering/feature_type/ip_address.py +0 -144
- ads/feature_engineering/feature_type/ip_address_v4.py +0 -138
- ads/feature_engineering/feature_type/ip_address_v6.py +0 -138
- ads/feature_engineering/feature_type/lat_long.py +0 -256
- ads/feature_engineering/feature_type/object.py +0 -43
- ads/feature_engineering/feature_type/ordinal.py +0 -132
- ads/feature_engineering/feature_type/phone_number.py +0 -135
- ads/feature_engineering/feature_type/string.py +0 -171
- ads/feature_engineering/feature_type/text.py +0 -93
- ads/feature_engineering/feature_type/unknown.py +0 -43
- ads/feature_engineering/feature_type/zip_code.py +0 -164
- ads/feature_engineering/feature_type_manager.py +0 -406
- ads/feature_engineering/schema.py +0 -795
- ads/feature_engineering/utils.py +0 -245
- ads/feature_store/.readthedocs.yaml +0 -19
- ads/feature_store/README.md +0 -65
- ads/feature_store/__init__.py +0 -9
- ads/feature_store/common/__init__.py +0 -0
- ads/feature_store/common/enums.py +0 -339
- ads/feature_store/common/exceptions.py +0 -18
- ads/feature_store/common/spark_session_singleton.py +0 -125
- ads/feature_store/common/utils/__init__.py +0 -0
- ads/feature_store/common/utils/base64_encoder_decoder.py +0 -72
- ads/feature_store/common/utils/feature_schema_mapper.py +0 -283
- ads/feature_store/common/utils/transformation_utils.py +0 -82
- ads/feature_store/common/utils/utility.py +0 -403
- ads/feature_store/data_validation/__init__.py +0 -0
- ads/feature_store/data_validation/great_expectation.py +0 -129
- ads/feature_store/dataset.py +0 -1230
- ads/feature_store/dataset_job.py +0 -530
- ads/feature_store/docs/Dockerfile +0 -7
- ads/feature_store/docs/Makefile +0 -44
- ads/feature_store/docs/conf.py +0 -28
- ads/feature_store/docs/requirements.txt +0 -14
- ads/feature_store/docs/source/ads.feature_store.query.rst +0 -20
- ads/feature_store/docs/source/cicd.rst +0 -137
- ads/feature_store/docs/source/conf.py +0 -86
- ads/feature_store/docs/source/data_versioning.rst +0 -33
- ads/feature_store/docs/source/dataset.rst +0 -388
- ads/feature_store/docs/source/dataset_job.rst +0 -27
- ads/feature_store/docs/source/demo.rst +0 -70
- ads/feature_store/docs/source/entity.rst +0 -78
- ads/feature_store/docs/source/feature_group.rst +0 -624
- ads/feature_store/docs/source/feature_group_job.rst +0 -29
- ads/feature_store/docs/source/feature_store.rst +0 -122
- ads/feature_store/docs/source/feature_store_class.rst +0 -123
- ads/feature_store/docs/source/feature_validation.rst +0 -66
- ads/feature_store/docs/source/figures/cicd.png +0 -0
- ads/feature_store/docs/source/figures/data_validation.png +0 -0
- ads/feature_store/docs/source/figures/data_versioning.png +0 -0
- ads/feature_store/docs/source/figures/dataset.gif +0 -0
- ads/feature_store/docs/source/figures/dataset.png +0 -0
- ads/feature_store/docs/source/figures/dataset_lineage.png +0 -0
- ads/feature_store/docs/source/figures/dataset_statistics.png +0 -0
- ads/feature_store/docs/source/figures/dataset_statistics_viz.png +0 -0
- ads/feature_store/docs/source/figures/dataset_validation_results.png +0 -0
- ads/feature_store/docs/source/figures/dataset_validation_summary.png +0 -0
- ads/feature_store/docs/source/figures/drift_monitoring.png +0 -0
- ads/feature_store/docs/source/figures/entity.png +0 -0
- ads/feature_store/docs/source/figures/feature_group.png +0 -0
- ads/feature_store/docs/source/figures/feature_group_lineage.png +0 -0
- ads/feature_store/docs/source/figures/feature_group_statistics_viz.png +0 -0
- ads/feature_store/docs/source/figures/feature_store_deployment.png +0 -0
- ads/feature_store/docs/source/figures/feature_store_overview.png +0 -0
- ads/feature_store/docs/source/figures/featuregroup.gif +0 -0
- ads/feature_store/docs/source/figures/lineage_d1.png +0 -0
- ads/feature_store/docs/source/figures/lineage_d2.png +0 -0
- ads/feature_store/docs/source/figures/lineage_fg.png +0 -0
- ads/feature_store/docs/source/figures/logo-dark-mode.png +0 -0
- ads/feature_store/docs/source/figures/logo-light-mode.png +0 -0
- ads/feature_store/docs/source/figures/overview.png +0 -0
- ads/feature_store/docs/source/figures/resource_manager.png +0 -0
- ads/feature_store/docs/source/figures/resource_manager_feature_store_stack.png +0 -0
- ads/feature_store/docs/source/figures/resource_manager_home.png +0 -0
- ads/feature_store/docs/source/figures/stats_1.png +0 -0
- ads/feature_store/docs/source/figures/stats_2.png +0 -0
- ads/feature_store/docs/source/figures/stats_d.png +0 -0
- ads/feature_store/docs/source/figures/stats_fg.png +0 -0
- ads/feature_store/docs/source/figures/transformation.png +0 -0
- ads/feature_store/docs/source/figures/transformations.gif +0 -0
- ads/feature_store/docs/source/figures/validation.png +0 -0
- ads/feature_store/docs/source/figures/validation_fg.png +0 -0
- ads/feature_store/docs/source/figures/validation_results.png +0 -0
- ads/feature_store/docs/source/figures/validation_summary.png +0 -0
- ads/feature_store/docs/source/index.rst +0 -81
- ads/feature_store/docs/source/module.rst +0 -8
- ads/feature_store/docs/source/notebook.rst +0 -94
- ads/feature_store/docs/source/overview.rst +0 -47
- ads/feature_store/docs/source/quickstart.rst +0 -176
- ads/feature_store/docs/source/release_notes.rst +0 -194
- ads/feature_store/docs/source/setup_feature_store.rst +0 -81
- ads/feature_store/docs/source/statistics.rst +0 -58
- ads/feature_store/docs/source/transformation.rst +0 -199
- ads/feature_store/docs/source/ui.rst +0 -65
- ads/feature_store/docs/source/user_guides.setup.feature_store_operator.rst +0 -66
- ads/feature_store/docs/source/user_guides.setup.helm_chart.rst +0 -192
- ads/feature_store/docs/source/user_guides.setup.terraform.rst +0 -338
- ads/feature_store/entity.py +0 -718
- ads/feature_store/execution_strategy/__init__.py +0 -0
- ads/feature_store/execution_strategy/delta_lake/__init__.py +0 -0
- ads/feature_store/execution_strategy/delta_lake/delta_lake_service.py +0 -375
- ads/feature_store/execution_strategy/engine/__init__.py +0 -0
- ads/feature_store/execution_strategy/engine/spark_engine.py +0 -316
- ads/feature_store/execution_strategy/execution_strategy.py +0 -113
- ads/feature_store/execution_strategy/execution_strategy_provider.py +0 -47
- ads/feature_store/execution_strategy/spark/__init__.py +0 -0
- ads/feature_store/execution_strategy/spark/spark_execution.py +0 -618
- ads/feature_store/feature.py +0 -192
- ads/feature_store/feature_group.py +0 -1494
- ads/feature_store/feature_group_expectation.py +0 -346
- ads/feature_store/feature_group_job.py +0 -602
- ads/feature_store/feature_lineage/__init__.py +0 -0
- ads/feature_store/feature_lineage/graphviz_service.py +0 -180
- ads/feature_store/feature_option_details.py +0 -50
- ads/feature_store/feature_statistics/__init__.py +0 -0
- ads/feature_store/feature_statistics/statistics_service.py +0 -99
- ads/feature_store/feature_store.py +0 -699
- ads/feature_store/feature_store_registrar.py +0 -518
- ads/feature_store/input_feature_detail.py +0 -149
- ads/feature_store/mixin/__init__.py +0 -4
- ads/feature_store/mixin/oci_feature_store.py +0 -145
- ads/feature_store/model_details.py +0 -73
- ads/feature_store/query/__init__.py +0 -0
- ads/feature_store/query/filter.py +0 -266
- ads/feature_store/query/generator/__init__.py +0 -0
- ads/feature_store/query/generator/query_generator.py +0 -298
- ads/feature_store/query/join.py +0 -161
- ads/feature_store/query/query.py +0 -403
- ads/feature_store/query/validator/__init__.py +0 -0
- ads/feature_store/query/validator/query_validator.py +0 -57
- ads/feature_store/response/__init__.py +0 -0
- ads/feature_store/response/response_builder.py +0 -68
- ads/feature_store/service/__init__.py +0 -0
- ads/feature_store/service/oci_dataset.py +0 -139
- ads/feature_store/service/oci_dataset_job.py +0 -199
- ads/feature_store/service/oci_entity.py +0 -125
- ads/feature_store/service/oci_feature_group.py +0 -164
- ads/feature_store/service/oci_feature_group_job.py +0 -214
- ads/feature_store/service/oci_feature_store.py +0 -182
- ads/feature_store/service/oci_lineage.py +0 -87
- ads/feature_store/service/oci_transformation.py +0 -104
- ads/feature_store/statistics/__init__.py +0 -0
- ads/feature_store/statistics/abs_feature_value.py +0 -49
- ads/feature_store/statistics/charts/__init__.py +0 -0
- ads/feature_store/statistics/charts/abstract_feature_plot.py +0 -37
- ads/feature_store/statistics/charts/box_plot.py +0 -148
- ads/feature_store/statistics/charts/frequency_distribution.py +0 -65
- ads/feature_store/statistics/charts/probability_distribution.py +0 -68
- ads/feature_store/statistics/charts/top_k_frequent_elements.py +0 -98
- ads/feature_store/statistics/feature_stat.py +0 -126
- ads/feature_store/statistics/generic_feature_value.py +0 -33
- ads/feature_store/statistics/statistics.py +0 -41
- ads/feature_store/statistics_config.py +0 -101
- ads/feature_store/templates/feature_store_template.yaml +0 -45
- ads/feature_store/transformation.py +0 -499
- ads/feature_store/validation_output.py +0 -57
- ads/hpo/__init__.py +0 -9
- ads/hpo/_imports.py +0 -91
- ads/hpo/ads_search_space.py +0 -439
- ads/hpo/distributions.py +0 -325
- ads/hpo/objective.py +0 -280
- ads/hpo/search_cv.py +0 -1657
- ads/hpo/stopping_criterion.py +0 -75
- ads/hpo/tuner_artifact.py +0 -413
- ads/hpo/utils.py +0 -91
- ads/hpo/validation.py +0 -140
- ads/hpo/visualization/__init__.py +0 -5
- ads/hpo/visualization/_contour.py +0 -23
- ads/hpo/visualization/_edf.py +0 -20
- ads/hpo/visualization/_intermediate_values.py +0 -21
- ads/hpo/visualization/_optimization_history.py +0 -25
- ads/hpo/visualization/_parallel_coordinate.py +0 -169
- ads/hpo/visualization/_param_importances.py +0 -26
- ads/jobs/__init__.py +0 -53
- ads/jobs/ads_job.py +0 -663
- ads/jobs/builders/__init__.py +0 -5
- ads/jobs/builders/base.py +0 -156
- ads/jobs/builders/infrastructure/__init__.py +0 -6
- ads/jobs/builders/infrastructure/base.py +0 -165
- ads/jobs/builders/infrastructure/dataflow.py +0 -1252
- ads/jobs/builders/infrastructure/dsc_job.py +0 -1894
- ads/jobs/builders/infrastructure/dsc_job_runtime.py +0 -1233
- ads/jobs/builders/infrastructure/utils.py +0 -65
- ads/jobs/builders/runtimes/__init__.py +0 -5
- ads/jobs/builders/runtimes/artifact.py +0 -338
- ads/jobs/builders/runtimes/base.py +0 -325
- ads/jobs/builders/runtimes/container_runtime.py +0 -242
- ads/jobs/builders/runtimes/python_runtime.py +0 -1016
- ads/jobs/builders/runtimes/pytorch_runtime.py +0 -204
- ads/jobs/cli.py +0 -104
- ads/jobs/env_var_parser.py +0 -131
- ads/jobs/extension.py +0 -160
- ads/jobs/schema/__init__.py +0 -5
- ads/jobs/schema/infrastructure_schema.json +0 -116
- ads/jobs/schema/job_schema.json +0 -42
- ads/jobs/schema/runtime_schema.json +0 -183
- ads/jobs/schema/validator.py +0 -141
- ads/jobs/serializer.py +0 -296
- ads/jobs/templates/__init__.py +0 -5
- ads/jobs/templates/container.py +0 -6
- ads/jobs/templates/driver_notebook.py +0 -177
- ads/jobs/templates/driver_oci.py +0 -500
- ads/jobs/templates/driver_python.py +0 -48
- ads/jobs/templates/driver_pytorch.py +0 -852
- ads/jobs/templates/driver_utils.py +0 -615
- ads/jobs/templates/hostname_from_env.c +0 -55
- ads/jobs/templates/oci_metrics.py +0 -181
- ads/jobs/utils.py +0 -104
- ads/llm/__init__.py +0 -28
- ads/llm/autogen/__init__.py +0 -2
- ads/llm/autogen/constants.py +0 -15
- ads/llm/autogen/reports/__init__.py +0 -2
- ads/llm/autogen/reports/base.py +0 -67
- ads/llm/autogen/reports/data.py +0 -103
- ads/llm/autogen/reports/session.py +0 -526
- ads/llm/autogen/reports/templates/chat_box.html +0 -13
- ads/llm/autogen/reports/templates/chat_box_lt.html +0 -5
- ads/llm/autogen/reports/templates/chat_box_rt.html +0 -6
- ads/llm/autogen/reports/utils.py +0 -56
- ads/llm/autogen/v02/__init__.py +0 -4
- ads/llm/autogen/v02/client.py +0 -295
- ads/llm/autogen/v02/log_handlers/__init__.py +0 -2
- ads/llm/autogen/v02/log_handlers/oci_file_handler.py +0 -83
- ads/llm/autogen/v02/loggers/__init__.py +0 -6
- ads/llm/autogen/v02/loggers/metric_logger.py +0 -320
- ads/llm/autogen/v02/loggers/session_logger.py +0 -580
- ads/llm/autogen/v02/loggers/utils.py +0 -86
- ads/llm/autogen/v02/runtime_logging.py +0 -163
- ads/llm/chain.py +0 -268
- ads/llm/chat_template.py +0 -31
- ads/llm/deploy.py +0 -63
- ads/llm/guardrails/__init__.py +0 -5
- ads/llm/guardrails/base.py +0 -442
- ads/llm/guardrails/huggingface.py +0 -44
- ads/llm/langchain/__init__.py +0 -5
- ads/llm/langchain/plugins/__init__.py +0 -5
- ads/llm/langchain/plugins/chat_models/__init__.py +0 -5
- ads/llm/langchain/plugins/chat_models/oci_data_science.py +0 -1027
- ads/llm/langchain/plugins/embeddings/__init__.py +0 -4
- ads/llm/langchain/plugins/embeddings/oci_data_science_model_deployment_endpoint.py +0 -184
- ads/llm/langchain/plugins/llms/__init__.py +0 -5
- ads/llm/langchain/plugins/llms/oci_data_science_model_deployment_endpoint.py +0 -979
- ads/llm/requirements.txt +0 -3
- ads/llm/serialize.py +0 -219
- ads/llm/serializers/__init__.py +0 -0
- ads/llm/serializers/retrieval_qa.py +0 -153
- ads/llm/serializers/runnable_parallel.py +0 -27
- ads/llm/templates/score_chain.jinja2 +0 -155
- ads/llm/templates/tool_chat_template_hermes.jinja +0 -130
- ads/llm/templates/tool_chat_template_mistral_parallel.jinja +0 -94
- ads/model/__init__.py +0 -52
- ads/model/artifact.py +0 -573
- ads/model/artifact_downloader.py +0 -254
- ads/model/artifact_uploader.py +0 -267
- ads/model/base_properties.py +0 -238
- ads/model/common/.model-ignore +0 -66
- ads/model/common/__init__.py +0 -5
- ads/model/common/utils.py +0 -142
- ads/model/datascience_model.py +0 -2635
- ads/model/deployment/__init__.py +0 -20
- ads/model/deployment/common/__init__.py +0 -5
- ads/model/deployment/common/utils.py +0 -308
- ads/model/deployment/model_deployer.py +0 -466
- ads/model/deployment/model_deployment.py +0 -1846
- ads/model/deployment/model_deployment_infrastructure.py +0 -671
- ads/model/deployment/model_deployment_properties.py +0 -493
- ads/model/deployment/model_deployment_runtime.py +0 -838
- ads/model/extractor/__init__.py +0 -5
- ads/model/extractor/automl_extractor.py +0 -74
- ads/model/extractor/embedding_onnx_extractor.py +0 -80
- ads/model/extractor/huggingface_extractor.py +0 -88
- ads/model/extractor/keras_extractor.py +0 -84
- ads/model/extractor/lightgbm_extractor.py +0 -93
- ads/model/extractor/model_info_extractor.py +0 -114
- ads/model/extractor/model_info_extractor_factory.py +0 -105
- ads/model/extractor/pytorch_extractor.py +0 -87
- ads/model/extractor/sklearn_extractor.py +0 -112
- ads/model/extractor/spark_extractor.py +0 -89
- ads/model/extractor/tensorflow_extractor.py +0 -85
- ads/model/extractor/xgboost_extractor.py +0 -94
- ads/model/framework/__init__.py +0 -5
- ads/model/framework/automl_model.py +0 -178
- ads/model/framework/embedding_onnx_model.py +0 -438
- ads/model/framework/huggingface_model.py +0 -399
- ads/model/framework/lightgbm_model.py +0 -266
- ads/model/framework/pytorch_model.py +0 -266
- ads/model/framework/sklearn_model.py +0 -250
- ads/model/framework/spark_model.py +0 -326
- ads/model/framework/tensorflow_model.py +0 -254
- ads/model/framework/xgboost_model.py +0 -258
- ads/model/generic_model.py +0 -3518
- ads/model/model_artifact_boilerplate/README.md +0 -381
- ads/model/model_artifact_boilerplate/__init__.py +0 -5
- ads/model/model_artifact_boilerplate/artifact_introspection_test/__init__.py +0 -5
- ads/model/model_artifact_boilerplate/artifact_introspection_test/model_artifact_validate.py +0 -427
- ads/model/model_artifact_boilerplate/artifact_introspection_test/requirements.txt +0 -2
- ads/model/model_artifact_boilerplate/runtime.yaml +0 -7
- ads/model/model_artifact_boilerplate/score.py +0 -61
- ads/model/model_file_description_schema.json +0 -68
- ads/model/model_introspect.py +0 -331
- ads/model/model_metadata.py +0 -1810
- ads/model/model_metadata_mixin.py +0 -460
- ads/model/model_properties.py +0 -63
- ads/model/model_version_set.py +0 -739
- ads/model/runtime/__init__.py +0 -5
- ads/model/runtime/env_info.py +0 -306
- ads/model/runtime/model_deployment_details.py +0 -37
- ads/model/runtime/model_provenance_details.py +0 -58
- ads/model/runtime/runtime_info.py +0 -81
- ads/model/runtime/schemas/inference_env_info_schema.yaml +0 -16
- ads/model/runtime/schemas/model_provenance_schema.yaml +0 -36
- ads/model/runtime/schemas/training_env_info_schema.yaml +0 -16
- ads/model/runtime/utils.py +0 -201
- ads/model/serde/__init__.py +0 -5
- ads/model/serde/common.py +0 -40
- ads/model/serde/model_input.py +0 -547
- ads/model/serde/model_serializer.py +0 -1184
- ads/model/service/__init__.py +0 -5
- ads/model/service/oci_datascience_model.py +0 -1076
- ads/model/service/oci_datascience_model_deployment.py +0 -500
- ads/model/service/oci_datascience_model_version_set.py +0 -176
- ads/model/transformer/__init__.py +0 -5
- ads/model/transformer/onnx_transformer.py +0 -324
- ads/mysqldb/__init__.py +0 -5
- ads/mysqldb/mysql_db.py +0 -227
- ads/opctl/__init__.py +0 -18
- ads/opctl/anomaly_detection.py +0 -11
- ads/opctl/backend/__init__.py +0 -5
- ads/opctl/backend/ads_dataflow.py +0 -353
- ads/opctl/backend/ads_ml_job.py +0 -710
- ads/opctl/backend/ads_ml_pipeline.py +0 -164
- ads/opctl/backend/ads_model_deployment.py +0 -209
- ads/opctl/backend/base.py +0 -146
- ads/opctl/backend/local.py +0 -1053
- ads/opctl/backend/marketplace/__init__.py +0 -9
- ads/opctl/backend/marketplace/helm_helper.py +0 -173
- ads/opctl/backend/marketplace/local_marketplace.py +0 -271
- ads/opctl/backend/marketplace/marketplace_backend_runner.py +0 -71
- ads/opctl/backend/marketplace/marketplace_operator_interface.py +0 -44
- ads/opctl/backend/marketplace/marketplace_operator_runner.py +0 -24
- ads/opctl/backend/marketplace/marketplace_utils.py +0 -212
- ads/opctl/backend/marketplace/models/__init__.py +0 -5
- ads/opctl/backend/marketplace/models/bearer_token.py +0 -94
- ads/opctl/backend/marketplace/models/marketplace_type.py +0 -70
- ads/opctl/backend/marketplace/models/ocir_details.py +0 -56
- ads/opctl/backend/marketplace/prerequisite_checker.py +0 -238
- ads/opctl/cli.py +0 -707
- ads/opctl/cmds.py +0 -869
- ads/opctl/conda/__init__.py +0 -5
- ads/opctl/conda/cli.py +0 -193
- ads/opctl/conda/cmds.py +0 -749
- ads/opctl/conda/config.yaml +0 -34
- ads/opctl/conda/manifest_template.yaml +0 -13
- ads/opctl/conda/multipart_uploader.py +0 -188
- ads/opctl/conda/pack.py +0 -89
- ads/opctl/config/__init__.py +0 -5
- ads/opctl/config/base.py +0 -57
- ads/opctl/config/diagnostics/__init__.py +0 -5
- ads/opctl/config/diagnostics/distributed/default_requirements_config.yaml +0 -62
- ads/opctl/config/merger.py +0 -255
- ads/opctl/config/resolver.py +0 -297
- ads/opctl/config/utils.py +0 -79
- ads/opctl/config/validator.py +0 -17
- ads/opctl/config/versioner.py +0 -68
- ads/opctl/config/yaml_parsers/__init__.py +0 -7
- ads/opctl/config/yaml_parsers/base.py +0 -58
- ads/opctl/config/yaml_parsers/distributed/__init__.py +0 -7
- ads/opctl/config/yaml_parsers/distributed/yaml_parser.py +0 -201
- ads/opctl/constants.py +0 -66
- ads/opctl/decorator/__init__.py +0 -5
- ads/opctl/decorator/common.py +0 -129
- ads/opctl/diagnostics/__init__.py +0 -5
- ads/opctl/diagnostics/__main__.py +0 -25
- ads/opctl/diagnostics/check_distributed_job_requirements.py +0 -212
- ads/opctl/diagnostics/check_requirements.py +0 -144
- ads/opctl/diagnostics/requirement_exception.py +0 -9
- ads/opctl/distributed/README.md +0 -109
- ads/opctl/distributed/__init__.py +0 -5
- ads/opctl/distributed/certificates.py +0 -32
- ads/opctl/distributed/cli.py +0 -207
- ads/opctl/distributed/cmds.py +0 -731
- ads/opctl/distributed/common/__init__.py +0 -5
- ads/opctl/distributed/common/abstract_cluster_provider.py +0 -449
- ads/opctl/distributed/common/abstract_framework_spec_builder.py +0 -88
- ads/opctl/distributed/common/cluster_config_helper.py +0 -103
- ads/opctl/distributed/common/cluster_provider_factory.py +0 -21
- ads/opctl/distributed/common/cluster_runner.py +0 -54
- ads/opctl/distributed/common/framework_factory.py +0 -29
- ads/opctl/docker/Dockerfile.job +0 -103
- ads/opctl/docker/Dockerfile.job.arm +0 -107
- ads/opctl/docker/Dockerfile.job.gpu +0 -175
- ads/opctl/docker/base-env.yaml +0 -13
- ads/opctl/docker/cuda.repo +0 -6
- ads/opctl/docker/operator/.dockerignore +0 -0
- ads/opctl/docker/operator/Dockerfile +0 -41
- ads/opctl/docker/operator/Dockerfile.gpu +0 -85
- ads/opctl/docker/operator/cuda.repo +0 -6
- ads/opctl/docker/operator/environment.yaml +0 -8
- ads/opctl/forecast.py +0 -11
- ads/opctl/index.yaml +0 -3
- ads/opctl/model/__init__.py +0 -5
- ads/opctl/model/cli.py +0 -65
- ads/opctl/model/cmds.py +0 -73
- ads/opctl/operator/README.md +0 -4
- ads/opctl/operator/__init__.py +0 -31
- ads/opctl/operator/cli.py +0 -344
- ads/opctl/operator/cmd.py +0 -596
- ads/opctl/operator/common/__init__.py +0 -5
- ads/opctl/operator/common/backend_factory.py +0 -460
- ads/opctl/operator/common/const.py +0 -27
- ads/opctl/operator/common/data/synthetic.csv +0 -16001
- ads/opctl/operator/common/dictionary_merger.py +0 -148
- ads/opctl/operator/common/errors.py +0 -42
- ads/opctl/operator/common/operator_config.py +0 -99
- ads/opctl/operator/common/operator_loader.py +0 -811
- ads/opctl/operator/common/operator_schema.yaml +0 -130
- ads/opctl/operator/common/operator_yaml_generator.py +0 -152
- ads/opctl/operator/common/utils.py +0 -208
- ads/opctl/operator/lowcode/__init__.py +0 -5
- ads/opctl/operator/lowcode/anomaly/MLoperator +0 -16
- ads/opctl/operator/lowcode/anomaly/README.md +0 -207
- ads/opctl/operator/lowcode/anomaly/__init__.py +0 -5
- ads/opctl/operator/lowcode/anomaly/__main__.py +0 -103
- ads/opctl/operator/lowcode/anomaly/cmd.py +0 -35
- ads/opctl/operator/lowcode/anomaly/const.py +0 -167
- ads/opctl/operator/lowcode/anomaly/environment.yaml +0 -10
- ads/opctl/operator/lowcode/anomaly/model/__init__.py +0 -5
- ads/opctl/operator/lowcode/anomaly/model/anomaly_dataset.py +0 -146
- ads/opctl/operator/lowcode/anomaly/model/anomaly_merlion.py +0 -162
- ads/opctl/operator/lowcode/anomaly/model/automlx.py +0 -99
- ads/opctl/operator/lowcode/anomaly/model/autots.py +0 -115
- ads/opctl/operator/lowcode/anomaly/model/base_model.py +0 -404
- ads/opctl/operator/lowcode/anomaly/model/factory.py +0 -110
- ads/opctl/operator/lowcode/anomaly/model/isolationforest.py +0 -78
- ads/opctl/operator/lowcode/anomaly/model/oneclasssvm.py +0 -78
- ads/opctl/operator/lowcode/anomaly/model/randomcutforest.py +0 -120
- ads/opctl/operator/lowcode/anomaly/model/tods.py +0 -119
- ads/opctl/operator/lowcode/anomaly/operator_config.py +0 -127
- ads/opctl/operator/lowcode/anomaly/schema.yaml +0 -401
- ads/opctl/operator/lowcode/anomaly/utils.py +0 -88
- ads/opctl/operator/lowcode/common/__init__.py +0 -5
- ads/opctl/operator/lowcode/common/const.py +0 -10
- ads/opctl/operator/lowcode/common/data.py +0 -116
- ads/opctl/operator/lowcode/common/errors.py +0 -47
- ads/opctl/operator/lowcode/common/transformations.py +0 -296
- ads/opctl/operator/lowcode/common/utils.py +0 -293
- ads/opctl/operator/lowcode/feature_store_marketplace/MLoperator +0 -13
- ads/opctl/operator/lowcode/feature_store_marketplace/README.md +0 -30
- ads/opctl/operator/lowcode/feature_store_marketplace/__init__.py +0 -5
- ads/opctl/operator/lowcode/feature_store_marketplace/__main__.py +0 -116
- ads/opctl/operator/lowcode/feature_store_marketplace/cmd.py +0 -85
- ads/opctl/operator/lowcode/feature_store_marketplace/const.py +0 -15
- ads/opctl/operator/lowcode/feature_store_marketplace/environment.yaml +0 -0
- ads/opctl/operator/lowcode/feature_store_marketplace/models/__init__.py +0 -4
- ads/opctl/operator/lowcode/feature_store_marketplace/models/apigw_config.py +0 -32
- ads/opctl/operator/lowcode/feature_store_marketplace/models/db_config.py +0 -43
- ads/opctl/operator/lowcode/feature_store_marketplace/models/mysql_config.py +0 -120
- ads/opctl/operator/lowcode/feature_store_marketplace/models/serializable_yaml_model.py +0 -34
- ads/opctl/operator/lowcode/feature_store_marketplace/operator_utils.py +0 -386
- ads/opctl/operator/lowcode/feature_store_marketplace/schema.yaml +0 -160
- ads/opctl/operator/lowcode/forecast/MLoperator +0 -25
- ads/opctl/operator/lowcode/forecast/README.md +0 -209
- ads/opctl/operator/lowcode/forecast/__init__.py +0 -5
- ads/opctl/operator/lowcode/forecast/__main__.py +0 -89
- ads/opctl/operator/lowcode/forecast/cmd.py +0 -40
- ads/opctl/operator/lowcode/forecast/const.py +0 -92
- ads/opctl/operator/lowcode/forecast/environment.yaml +0 -20
- ads/opctl/operator/lowcode/forecast/errors.py +0 -26
- ads/opctl/operator/lowcode/forecast/model/__init__.py +0 -5
- ads/opctl/operator/lowcode/forecast/model/arima.py +0 -279
- ads/opctl/operator/lowcode/forecast/model/automlx.py +0 -542
- ads/opctl/operator/lowcode/forecast/model/autots.py +0 -312
- ads/opctl/operator/lowcode/forecast/model/base_model.py +0 -863
- ads/opctl/operator/lowcode/forecast/model/factory.py +0 -106
- ads/opctl/operator/lowcode/forecast/model/forecast_datasets.py +0 -492
- ads/opctl/operator/lowcode/forecast/model/ml_forecast.py +0 -243
- ads/opctl/operator/lowcode/forecast/model/neuralprophet.py +0 -486
- ads/opctl/operator/lowcode/forecast/model/prophet.py +0 -445
- ads/opctl/operator/lowcode/forecast/model_evaluator.py +0 -244
- ads/opctl/operator/lowcode/forecast/operator_config.py +0 -234
- ads/opctl/operator/lowcode/forecast/schema.yaml +0 -506
- ads/opctl/operator/lowcode/forecast/utils.py +0 -413
- ads/opctl/operator/lowcode/forecast/whatifserve/__init__.py +0 -7
- ads/opctl/operator/lowcode/forecast/whatifserve/deployment_manager.py +0 -285
- ads/opctl/operator/lowcode/forecast/whatifserve/score.py +0 -246
- ads/opctl/operator/lowcode/pii/MLoperator +0 -17
- ads/opctl/operator/lowcode/pii/README.md +0 -208
- ads/opctl/operator/lowcode/pii/__init__.py +0 -5
- ads/opctl/operator/lowcode/pii/__main__.py +0 -78
- ads/opctl/operator/lowcode/pii/cmd.py +0 -39
- ads/opctl/operator/lowcode/pii/constant.py +0 -84
- ads/opctl/operator/lowcode/pii/environment.yaml +0 -17
- ads/opctl/operator/lowcode/pii/errors.py +0 -27
- ads/opctl/operator/lowcode/pii/model/__init__.py +0 -5
- ads/opctl/operator/lowcode/pii/model/factory.py +0 -82
- ads/opctl/operator/lowcode/pii/model/guardrails.py +0 -167
- ads/opctl/operator/lowcode/pii/model/pii.py +0 -145
- ads/opctl/operator/lowcode/pii/model/processor/__init__.py +0 -34
- ads/opctl/operator/lowcode/pii/model/processor/email_replacer.py +0 -34
- ads/opctl/operator/lowcode/pii/model/processor/mbi_replacer.py +0 -35
- ads/opctl/operator/lowcode/pii/model/processor/name_replacer.py +0 -225
- ads/opctl/operator/lowcode/pii/model/processor/number_replacer.py +0 -73
- ads/opctl/operator/lowcode/pii/model/processor/remover.py +0 -26
- ads/opctl/operator/lowcode/pii/model/report.py +0 -487
- ads/opctl/operator/lowcode/pii/operator_config.py +0 -95
- ads/opctl/operator/lowcode/pii/schema.yaml +0 -108
- ads/opctl/operator/lowcode/pii/utils.py +0 -43
- ads/opctl/operator/lowcode/recommender/MLoperator +0 -16
- ads/opctl/operator/lowcode/recommender/README.md +0 -206
- ads/opctl/operator/lowcode/recommender/__init__.py +0 -5
- ads/opctl/operator/lowcode/recommender/__main__.py +0 -82
- ads/opctl/operator/lowcode/recommender/cmd.py +0 -33
- ads/opctl/operator/lowcode/recommender/constant.py +0 -30
- ads/opctl/operator/lowcode/recommender/environment.yaml +0 -11
- ads/opctl/operator/lowcode/recommender/model/base_model.py +0 -212
- ads/opctl/operator/lowcode/recommender/model/factory.py +0 -56
- ads/opctl/operator/lowcode/recommender/model/recommender_dataset.py +0 -25
- ads/opctl/operator/lowcode/recommender/model/svd.py +0 -106
- ads/opctl/operator/lowcode/recommender/operator_config.py +0 -81
- ads/opctl/operator/lowcode/recommender/schema.yaml +0 -265
- ads/opctl/operator/lowcode/recommender/utils.py +0 -13
- ads/opctl/operator/runtime/__init__.py +0 -5
- ads/opctl/operator/runtime/const.py +0 -17
- ads/opctl/operator/runtime/container_runtime_schema.yaml +0 -50
- ads/opctl/operator/runtime/marketplace_runtime.py +0 -50
- ads/opctl/operator/runtime/python_marketplace_runtime_schema.yaml +0 -21
- ads/opctl/operator/runtime/python_runtime_schema.yaml +0 -21
- ads/opctl/operator/runtime/runtime.py +0 -115
- ads/opctl/schema.yaml.yml +0 -36
- ads/opctl/script.py +0 -40
- ads/opctl/spark/__init__.py +0 -5
- ads/opctl/spark/cli.py +0 -43
- ads/opctl/spark/cmds.py +0 -147
- ads/opctl/templates/diagnostic_report_template.jinja2 +0 -102
- ads/opctl/utils.py +0 -344
- ads/oracledb/__init__.py +0 -5
- ads/oracledb/oracle_db.py +0 -346
- ads/pipeline/__init__.py +0 -39
- ads/pipeline/ads_pipeline.py +0 -2279
- ads/pipeline/ads_pipeline_run.py +0 -772
- ads/pipeline/ads_pipeline_step.py +0 -605
- ads/pipeline/builders/__init__.py +0 -5
- ads/pipeline/builders/infrastructure/__init__.py +0 -5
- ads/pipeline/builders/infrastructure/custom_script.py +0 -32
- ads/pipeline/cli.py +0 -119
- ads/pipeline/extension.py +0 -291
- ads/pipeline/schema/__init__.py +0 -5
- ads/pipeline/schema/cs_step_schema.json +0 -35
- ads/pipeline/schema/ml_step_schema.json +0 -31
- ads/pipeline/schema/pipeline_schema.json +0 -71
- ads/pipeline/visualizer/__init__.py +0 -5
- ads/pipeline/visualizer/base.py +0 -570
- ads/pipeline/visualizer/graph_renderer.py +0 -272
- ads/pipeline/visualizer/text_renderer.py +0 -84
- ads/secrets/__init__.py +0 -11
- ads/secrets/adb.py +0 -386
- ads/secrets/auth_token.py +0 -86
- ads/secrets/big_data_service.py +0 -365
- ads/secrets/mysqldb.py +0 -149
- ads/secrets/oracledb.py +0 -160
- ads/secrets/secrets.py +0 -407
- ads/telemetry/__init__.py +0 -7
- ads/telemetry/base.py +0 -69
- ads/telemetry/client.py +0 -125
- ads/telemetry/telemetry.py +0 -257
- ads/templates/dataflow_pyspark.jinja2 +0 -13
- ads/templates/dataflow_sparksql.jinja2 +0 -22
- ads/templates/func.jinja2 +0 -20
- ads/templates/schemas/openapi.json +0 -1740
- ads/templates/score-pkl.jinja2 +0 -173
- ads/templates/score.jinja2 +0 -322
- ads/templates/score_embedding_onnx.jinja2 +0 -202
- ads/templates/score_generic.jinja2 +0 -165
- ads/templates/score_huggingface_pipeline.jinja2 +0 -217
- ads/templates/score_lightgbm.jinja2 +0 -185
- ads/templates/score_onnx.jinja2 +0 -407
- ads/templates/score_onnx_new.jinja2 +0 -473
- ads/templates/score_oracle_automl.jinja2 +0 -185
- ads/templates/score_pyspark.jinja2 +0 -154
- ads/templates/score_pytorch.jinja2 +0 -219
- ads/templates/score_scikit-learn.jinja2 +0 -184
- ads/templates/score_tensorflow.jinja2 +0 -184
- ads/templates/score_xgboost.jinja2 +0 -178
- ads/text_dataset/__init__.py +0 -5
- ads/text_dataset/backends.py +0 -211
- ads/text_dataset/dataset.py +0 -445
- ads/text_dataset/extractor.py +0 -207
- ads/text_dataset/options.py +0 -53
- ads/text_dataset/udfs.py +0 -22
- ads/text_dataset/utils.py +0 -49
- ads/type_discovery/__init__.py +0 -9
- ads/type_discovery/abstract_detector.py +0 -21
- ads/type_discovery/constant_detector.py +0 -41
- ads/type_discovery/continuous_detector.py +0 -54
- ads/type_discovery/credit_card_detector.py +0 -99
- ads/type_discovery/datetime_detector.py +0 -92
- ads/type_discovery/discrete_detector.py +0 -118
- ads/type_discovery/document_detector.py +0 -146
- ads/type_discovery/ip_detector.py +0 -68
- ads/type_discovery/latlon_detector.py +0 -90
- ads/type_discovery/phone_number_detector.py +0 -63
- ads/type_discovery/type_discovery_driver.py +0 -87
- ads/type_discovery/typed_feature.py +0 -594
- ads/type_discovery/unknown_detector.py +0 -41
- ads/type_discovery/zipcode_detector.py +0 -48
- ads/vault/__init__.py +0 -7
- ads/vault/vault.py +0 -237
- oracle_ads-2.13.8.dist-info/RECORD +0 -858
- {oracle_ads-2.13.8.dist-info → oracle_ads-2.13.9rc0.dist-info}/licenses/LICENSE.txt +0 -0
@@ -1,285 +0,0 @@
|
|
1
|
-
#!/usr/bin/env python
|
2
|
-
# Copyright (c) 2023, 2024 Oracle and/or its affiliates.
|
3
|
-
# Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
|
4
|
-
|
5
|
-
import os
|
6
|
-
import pickle
|
7
|
-
import shutil
|
8
|
-
import sys
|
9
|
-
import tempfile
|
10
|
-
|
11
|
-
import cloudpickle
|
12
|
-
import oci
|
13
|
-
import pandas as pd
|
14
|
-
from oci.data_science import DataScienceClient, DataScienceClientCompositeOperations
|
15
|
-
from oci.data_science.models import (
|
16
|
-
CategoryLogDetails,
|
17
|
-
CreateModelDeploymentDetails,
|
18
|
-
FixedSizeScalingPolicy,
|
19
|
-
InstanceConfiguration,
|
20
|
-
LogDetails,
|
21
|
-
ModelConfigurationDetails,
|
22
|
-
SingleModelDeploymentConfigurationDetails,
|
23
|
-
)
|
24
|
-
|
25
|
-
from ads.common.model_export_util import prepare_generic_model
|
26
|
-
from ads.common.object_storage_details import ObjectStorageDetails
|
27
|
-
from ads.opctl import logger
|
28
|
-
from ads.opctl.operator.common.utils import create_log_in_log_group
|
29
|
-
from ads.opctl.operator.lowcode.common.utils import (
|
30
|
-
default_signer,
|
31
|
-
write_data,
|
32
|
-
write_simple_json,
|
33
|
-
)
|
34
|
-
|
35
|
-
from ..model.forecast_datasets import AdditionalData
|
36
|
-
from ..operator_config import ForecastOperatorSpec
|
37
|
-
|
38
|
-
|
39
|
-
class ModelDeploymentManager:
|
40
|
-
def __init__(
|
41
|
-
self,
|
42
|
-
spec: ForecastOperatorSpec,
|
43
|
-
additional_data: AdditionalData,
|
44
|
-
previous_model_version=None,
|
45
|
-
):
|
46
|
-
self.spec = spec
|
47
|
-
self.model_name = spec.model
|
48
|
-
self.horizon = spec.horizon
|
49
|
-
self.additional_data = additional_data.get_dict_by_series()
|
50
|
-
self.model_obj = {}
|
51
|
-
self.display_name = spec.what_if_analysis.model_display_name
|
52
|
-
self.project_id = (
|
53
|
-
spec.what_if_analysis.project_id
|
54
|
-
if spec.what_if_analysis.project_id
|
55
|
-
else os.environ.get("PROJECT_OCID")
|
56
|
-
)
|
57
|
-
self.compartment_id = (
|
58
|
-
spec.what_if_analysis.compartment_id
|
59
|
-
if spec.what_if_analysis.compartment_id
|
60
|
-
else os.environ.get("NB_SESSION_COMPARTMENT_OCID")
|
61
|
-
)
|
62
|
-
if self.project_id is None or self.compartment_id is None:
|
63
|
-
raise ValueError("Either project_id or compartment_id cannot be None.")
|
64
|
-
self.path_to_artifact = f"{self.spec.output_directory.url}/artifacts/"
|
65
|
-
self.pickle_file_path = f"{self.spec.output_directory.url}/model.pkl"
|
66
|
-
self.model_version = previous_model_version + 1 if previous_model_version else 1
|
67
|
-
self.catalog_id = None
|
68
|
-
self.test_mode = os.environ.get("TEST_MODE", False)
|
69
|
-
self.deployment_info = {}
|
70
|
-
|
71
|
-
def _sanity_test(self):
|
72
|
-
"""
|
73
|
-
Function perform sanity test for saved artifact
|
74
|
-
"""
|
75
|
-
org_sys_path = sys.path[:]
|
76
|
-
try:
|
77
|
-
sys.path.insert(0, f"{self.path_to_artifact}")
|
78
|
-
from score import load_model, predict
|
79
|
-
|
80
|
-
_ = load_model()
|
81
|
-
|
82
|
-
# Write additional data to tmp file and perform sanity check
|
83
|
-
with tempfile.NamedTemporaryFile(suffix=".csv") as temp_file:
|
84
|
-
one_series = next(iter(self.additional_data))
|
85
|
-
sample_prediction_data = self.additional_data[one_series].tail(
|
86
|
-
self.horizon
|
87
|
-
)
|
88
|
-
sample_prediction_data[self.spec.target_category_columns[0]] = (
|
89
|
-
one_series
|
90
|
-
)
|
91
|
-
date_col_name = self.spec.datetime_column.name
|
92
|
-
date_col_format = self.spec.datetime_column.format
|
93
|
-
sample_prediction_data[date_col_name] = sample_prediction_data[
|
94
|
-
date_col_name
|
95
|
-
].dt.strftime(date_col_format)
|
96
|
-
sample_prediction_data.to_csv(temp_file.name, index=False)
|
97
|
-
input_data = {"additional_data": {"url": temp_file.name}}
|
98
|
-
prediction_test = predict(input_data, _)
|
99
|
-
logger.info(f"prediction test completed with result :{prediction_test}")
|
100
|
-
except Exception as e:
|
101
|
-
logger.error(f"An error occurred during the sanity test: {e}")
|
102
|
-
raise
|
103
|
-
finally:
|
104
|
-
sys.path = org_sys_path
|
105
|
-
|
106
|
-
def _copy_score_file(self):
|
107
|
-
"""
|
108
|
-
Copies the score.py to the artifact_path.
|
109
|
-
"""
|
110
|
-
try:
|
111
|
-
current_dir = os.path.dirname(os.path.abspath(__file__))
|
112
|
-
score_file = os.path.join(current_dir, "score.py")
|
113
|
-
destination_file = os.path.join(
|
114
|
-
self.path_to_artifact, os.path.basename(score_file)
|
115
|
-
)
|
116
|
-
shutil.copy2(score_file, destination_file)
|
117
|
-
logger.info(f"score.py copied successfully to {self.path_to_artifact}")
|
118
|
-
except Exception as e:
|
119
|
-
logger.warning(f"Error copying file: {e}")
|
120
|
-
raise e
|
121
|
-
|
122
|
-
def save_to_catalog(self):
|
123
|
-
"""Save the model to a model catalog"""
|
124
|
-
with open(self.pickle_file_path, "rb") as file:
|
125
|
-
self.model_obj = pickle.load(file)
|
126
|
-
|
127
|
-
if not os.path.exists(self.path_to_artifact):
|
128
|
-
os.mkdir(self.path_to_artifact)
|
129
|
-
|
130
|
-
artifact_dict = {"spec": self.spec.to_dict(), "models": self.model_obj}
|
131
|
-
with open(f"{self.path_to_artifact}/models.pickle", "wb") as f:
|
132
|
-
cloudpickle.dump(artifact_dict, f)
|
133
|
-
artifact = prepare_generic_model(
|
134
|
-
self.path_to_artifact,
|
135
|
-
function_artifacts=False,
|
136
|
-
force_overwrite=True,
|
137
|
-
data_science_env=True,
|
138
|
-
)
|
139
|
-
|
140
|
-
self._copy_score_file()
|
141
|
-
self._sanity_test()
|
142
|
-
|
143
|
-
if isinstance(self.model_obj, dict):
|
144
|
-
series = self.model_obj.keys()
|
145
|
-
else:
|
146
|
-
series = self.additional_data.keys()
|
147
|
-
description = f"The object contains {len(series)} {self.model_name} models"
|
148
|
-
|
149
|
-
if not self.test_mode:
|
150
|
-
catalog_entry = artifact.save(
|
151
|
-
display_name=self.display_name,
|
152
|
-
compartment_id=self.compartment_id,
|
153
|
-
project_id=self.project_id,
|
154
|
-
description=description,
|
155
|
-
)
|
156
|
-
self.catalog_id = catalog_entry.id
|
157
|
-
|
158
|
-
logger.info(
|
159
|
-
f"Saved {self.model_name} version-v{self.model_version} to model catalog"
|
160
|
-
f" with model ocid : {self.catalog_id}"
|
161
|
-
)
|
162
|
-
|
163
|
-
self.deployment_info = {"model_ocid": self.catalog_id, "series": list(series)}
|
164
|
-
|
165
|
-
def create_deployment(self):
|
166
|
-
"""Create a model deployment serving"""
|
167
|
-
|
168
|
-
# create new model deployment
|
169
|
-
initial_shape = self.spec.what_if_analysis.model_deployment.initial_shape
|
170
|
-
name = self.spec.what_if_analysis.model_deployment.display_name
|
171
|
-
description = self.spec.what_if_analysis.model_deployment.description
|
172
|
-
auto_scaling_config = self.spec.what_if_analysis.model_deployment.auto_scaling
|
173
|
-
|
174
|
-
# if auto_scaling_config is defined
|
175
|
-
if auto_scaling_config:
|
176
|
-
scaling_policy = oci.data_science.models.AutoScalingPolicy(
|
177
|
-
policy_type="AUTOSCALING",
|
178
|
-
auto_scaling_policies=[
|
179
|
-
oci.data_science.models.ThresholdBasedAutoScalingPolicyDetails(
|
180
|
-
auto_scaling_policy_type="THRESHOLD",
|
181
|
-
rules=[
|
182
|
-
oci.data_science.models.PredefinedMetricExpressionRule(
|
183
|
-
metric_expression_rule_type="PREDEFINED_EXPRESSION",
|
184
|
-
metric_type=auto_scaling_config.scaling_metric,
|
185
|
-
scale_in_configuration=oci.data_science.models.PredefinedExpressionThresholdScalingConfiguration(
|
186
|
-
scaling_configuration_type="THRESHOLD",
|
187
|
-
threshold=auto_scaling_config.scale_in_threshold,
|
188
|
-
),
|
189
|
-
scale_out_configuration=oci.data_science.models.PredefinedExpressionThresholdScalingConfiguration(
|
190
|
-
scaling_configuration_type="THRESHOLD",
|
191
|
-
threshold=auto_scaling_config.scale_out_threshold,
|
192
|
-
),
|
193
|
-
)
|
194
|
-
],
|
195
|
-
maximum_instance_count=auto_scaling_config.maximum_instance,
|
196
|
-
minimum_instance_count=auto_scaling_config.minimum_instance,
|
197
|
-
initial_instance_count=auto_scaling_config.minimum_instance,
|
198
|
-
)
|
199
|
-
],
|
200
|
-
cool_down_in_seconds=auto_scaling_config.cool_down_in_seconds,
|
201
|
-
is_enabled=True,
|
202
|
-
)
|
203
|
-
logger.info(
|
204
|
-
f"Using autoscaling {auto_scaling_config.scaling_metric} for creating MD"
|
205
|
-
)
|
206
|
-
else:
|
207
|
-
scaling_policy = FixedSizeScalingPolicy(instance_count=1)
|
208
|
-
logger.info("Using fixed size policy for creating MD")
|
209
|
-
|
210
|
-
model_configuration_details_object = ModelConfigurationDetails(
|
211
|
-
model_id=self.catalog_id,
|
212
|
-
instance_configuration=InstanceConfiguration(
|
213
|
-
instance_shape_name=initial_shape
|
214
|
-
),
|
215
|
-
scaling_policy=scaling_policy,
|
216
|
-
bandwidth_mbps=20,
|
217
|
-
)
|
218
|
-
|
219
|
-
single_model_config = SingleModelDeploymentConfigurationDetails(
|
220
|
-
deployment_type="SINGLE_MODEL",
|
221
|
-
model_configuration_details=model_configuration_details_object,
|
222
|
-
)
|
223
|
-
|
224
|
-
log_group = self.spec.what_if_analysis.model_deployment.log_group
|
225
|
-
log_id = self.spec.what_if_analysis.model_deployment.log_id
|
226
|
-
if not log_id and not self.test_mode:
|
227
|
-
signer = oci.auth.signers.get_resource_principals_signer()
|
228
|
-
auth = {"signer": signer, "config": {}}
|
229
|
-
log_id = create_log_in_log_group(self.compartment_id, log_group, auth)
|
230
|
-
|
231
|
-
logs_configuration_details_object = CategoryLogDetails(
|
232
|
-
access=LogDetails(log_group_id=log_group, log_id=log_id),
|
233
|
-
predict=LogDetails(log_group_id=log_group, log_id=log_id),
|
234
|
-
)
|
235
|
-
|
236
|
-
model_deploy_configuration = CreateModelDeploymentDetails(
|
237
|
-
display_name=name,
|
238
|
-
description=description,
|
239
|
-
project_id=self.project_id,
|
240
|
-
compartment_id=self.compartment_id,
|
241
|
-
model_deployment_configuration_details=single_model_config,
|
242
|
-
category_log_details=logs_configuration_details_object,
|
243
|
-
)
|
244
|
-
|
245
|
-
if not self.test_mode:
|
246
|
-
auth = oci.auth.signers.get_resource_principals_signer()
|
247
|
-
data_science = DataScienceClient({}, signer=auth)
|
248
|
-
data_science_composite = DataScienceClientCompositeOperations(data_science)
|
249
|
-
model_deployment = (
|
250
|
-
data_science_composite.create_model_deployment_and_wait_for_state(
|
251
|
-
model_deploy_configuration, wait_for_states=["SUCCEEDED", "FAILED"]
|
252
|
-
)
|
253
|
-
)
|
254
|
-
self.deployment_info["work_request"] = model_deployment.data.id
|
255
|
-
logger.info(f"deployment metadata :{model_deployment.data}")
|
256
|
-
md = data_science.get_model_deployment(
|
257
|
-
model_deployment_id=model_deployment.data.resources[0].identifier
|
258
|
-
)
|
259
|
-
self.deployment_info["model_deployment_ocid"] = md.data.id
|
260
|
-
self.deployment_info["status"] = md.data.lifecycle_state
|
261
|
-
endpoint_url = md.data.model_deployment_url
|
262
|
-
self.deployment_info["model_deployment_endpoint"] = (
|
263
|
-
f"{endpoint_url}/predict"
|
264
|
-
)
|
265
|
-
self.deployment_info["log_id"] = log_id
|
266
|
-
|
267
|
-
def save_deployment_info(self):
|
268
|
-
output_dir = self.spec.output_directory.url
|
269
|
-
if ObjectStorageDetails.is_oci_path(output_dir):
|
270
|
-
storage_options = default_signer()
|
271
|
-
else:
|
272
|
-
storage_options = {}
|
273
|
-
write_data(
|
274
|
-
data=pd.DataFrame.from_dict(self.deployment_info),
|
275
|
-
filename=os.path.join(output_dir, "deployment_info.json"),
|
276
|
-
format="json",
|
277
|
-
storage_options=storage_options,
|
278
|
-
index=False,
|
279
|
-
indent=4,
|
280
|
-
orient="records",
|
281
|
-
)
|
282
|
-
write_simple_json(
|
283
|
-
self.deployment_info, os.path.join(output_dir, "deployment_info.json")
|
284
|
-
)
|
285
|
-
logger.info(f"Saved deployment info to {output_dir}")
|
@@ -1,246 +0,0 @@
|
|
1
|
-
#!/usr/bin/env python
|
2
|
-
|
3
|
-
# Copyright (c) 2023, 2024 Oracle and/or its affiliates.
|
4
|
-
# Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
|
5
|
-
|
6
|
-
import json
|
7
|
-
import os
|
8
|
-
from joblib import load
|
9
|
-
import pandas as pd
|
10
|
-
import numpy as np
|
11
|
-
from functools import lru_cache
|
12
|
-
import logging
|
13
|
-
import ads
|
14
|
-
from ads.opctl.operator.lowcode.common.utils import load_data
|
15
|
-
from ads.opctl.operator.common.operator_config import InputData
|
16
|
-
from ads.opctl.operator.lowcode.forecast.const import SupportedModels
|
17
|
-
|
18
|
-
ads.set_auth("resource_principal")
|
19
|
-
|
20
|
-
logging.basicConfig(format='%(name)s - %(levelname)s - %(message)s', level=logging.INFO)
|
21
|
-
logger_pred = logging.getLogger('model-prediction')
|
22
|
-
logger_pred.setLevel(logging.INFO)
|
23
|
-
logger_feat = logging.getLogger('input-features')
|
24
|
-
logger_feat.setLevel(logging.INFO)
|
25
|
-
|
26
|
-
"""
|
27
|
-
Inference script. This script is used for prediction by scoring server when schema is known.
|
28
|
-
"""
|
29
|
-
|
30
|
-
|
31
|
-
@lru_cache(maxsize=10)
|
32
|
-
def load_model():
|
33
|
-
"""
|
34
|
-
Loads model from the serialized format
|
35
|
-
|
36
|
-
Returns
|
37
|
-
-------
|
38
|
-
model: a model instance on which predict API can be invoked
|
39
|
-
"""
|
40
|
-
model_dir = os.path.dirname(os.path.realpath(__file__))
|
41
|
-
contents = os.listdir(model_dir)
|
42
|
-
model_file_name = "models.pickle"
|
43
|
-
if model_file_name in contents:
|
44
|
-
with open(os.path.join(os.path.dirname(os.path.realpath(__file__)), model_file_name), "rb") as file:
|
45
|
-
model = load(file)
|
46
|
-
else:
|
47
|
-
raise Exception('{0} is not found in model directory {1}'.format(model_file_name, model_dir))
|
48
|
-
return model
|
49
|
-
|
50
|
-
|
51
|
-
@lru_cache(maxsize=1)
|
52
|
-
def fetch_data_type_from_schema(
|
53
|
-
input_schema_path=os.path.join(os.path.dirname(os.path.realpath(__file__)), "input_schema.json")):
|
54
|
-
"""
|
55
|
-
Returns data type information fetch from input_schema.json.
|
56
|
-
|
57
|
-
Parameters
|
58
|
-
----------
|
59
|
-
input_schema_path: path of input schema.
|
60
|
-
|
61
|
-
Returns
|
62
|
-
-------
|
63
|
-
data_type: data type fetch from input_schema.json.
|
64
|
-
|
65
|
-
"""
|
66
|
-
data_type = {}
|
67
|
-
if os.path.exists(input_schema_path):
|
68
|
-
schema = json.load(open(input_schema_path))
|
69
|
-
for col in schema['schema']:
|
70
|
-
data_type[col['name']] = col['dtype']
|
71
|
-
else:
|
72
|
-
print(
|
73
|
-
"input_schema has to be passed in in order to recover the same data type. pass `X_sample` in `ads.model.framework.sklearn_model.SklearnModel.prepare` function to generate the input_schema. Otherwise, the data type might be changed after serialization/deserialization.")
|
74
|
-
return data_type
|
75
|
-
|
76
|
-
|
77
|
-
def deserialize(data, input_schema_path):
|
78
|
-
"""
|
79
|
-
Deserialize json serialization data to data in original type when sent to predict.
|
80
|
-
|
81
|
-
Parameters
|
82
|
-
----------
|
83
|
-
data: serialized input data.
|
84
|
-
input_schema_path: path of input schema.
|
85
|
-
|
86
|
-
Returns
|
87
|
-
-------
|
88
|
-
data: deserialized input data.
|
89
|
-
|
90
|
-
"""
|
91
|
-
|
92
|
-
# Add further data deserialization if needed
|
93
|
-
return data
|
94
|
-
|
95
|
-
|
96
|
-
def pre_inference(data, input_schema_path):
|
97
|
-
"""
|
98
|
-
Preprocess data
|
99
|
-
|
100
|
-
Parameters
|
101
|
-
----------
|
102
|
-
data: Data format as expected by the predict API of the core estimator.
|
103
|
-
input_schema_path: path of input schema.
|
104
|
-
|
105
|
-
Returns
|
106
|
-
-------
|
107
|
-
data: Data format after any processing.
|
108
|
-
|
109
|
-
"""
|
110
|
-
return deserialize(data, input_schema_path)
|
111
|
-
|
112
|
-
|
113
|
-
def post_inference(yhat):
|
114
|
-
"""
|
115
|
-
Post-process the model results
|
116
|
-
|
117
|
-
Parameters
|
118
|
-
----------
|
119
|
-
yhat: Data format after calling model.predict.
|
120
|
-
|
121
|
-
Returns
|
122
|
-
-------
|
123
|
-
yhat: Data format after any processing.
|
124
|
-
|
125
|
-
"""
|
126
|
-
if isinstance(yhat, pd.core.frame.DataFrame):
|
127
|
-
yhat = yhat.values
|
128
|
-
if isinstance(yhat, np.ndarray):
|
129
|
-
yhat = yhat.tolist()
|
130
|
-
return yhat
|
131
|
-
|
132
|
-
|
133
|
-
def get_forecast(future_df, model_name, series_id, model_object, date_col, target_column, target_cat_col, horizon):
|
134
|
-
date_col_name = date_col["name"]
|
135
|
-
date_col_format = date_col["format"]
|
136
|
-
future_df[target_cat_col] = future_df[target_cat_col].astype("str")
|
137
|
-
future_df[date_col_name] = pd.to_datetime(
|
138
|
-
future_df[date_col_name], format=date_col_format
|
139
|
-
)
|
140
|
-
if model_name == SupportedModels.AutoTS:
|
141
|
-
series_id_col = "Series"
|
142
|
-
full_data_indexed = future_df.rename(columns={target_cat_col: series_id_col})
|
143
|
-
additional_regressors = list(
|
144
|
-
set(full_data_indexed.columns) - {target_column, series_id_col, date_col_name}
|
145
|
-
)
|
146
|
-
future_reg = full_data_indexed.reset_index().pivot(
|
147
|
-
index=date_col_name,
|
148
|
-
columns=series_id_col,
|
149
|
-
values=additional_regressors,
|
150
|
-
)
|
151
|
-
pred_obj = model_object.predict(future_regressor=future_reg)
|
152
|
-
return pred_obj.forecast[series_id].tolist()
|
153
|
-
elif model_name == SupportedModels.Prophet and series_id in model_object:
|
154
|
-
model = model_object[series_id]['model']
|
155
|
-
label_encoder = model_object[series_id]['le']
|
156
|
-
processed = future_df.rename(columns={date_col_name: 'ds', target_column: 'y'})
|
157
|
-
encoded_df = label_encoder.transform(processed)
|
158
|
-
forecast = model.predict(encoded_df)
|
159
|
-
return forecast['yhat'].tolist()
|
160
|
-
elif model_name == SupportedModels.NeuralProphet and series_id in model_object:
|
161
|
-
model = model_object[series_id]['model']
|
162
|
-
label_encoder = model_object[series_id]['le']
|
163
|
-
model.restore_trainer()
|
164
|
-
accepted_regressors = list(model.config_regressors.regressors.keys())
|
165
|
-
data = future_df.rename(columns={date_col_name: 'ds', target_column: 'y'})
|
166
|
-
encoded_df = label_encoder.transform(data)
|
167
|
-
future = encoded_df[accepted_regressors + ["ds"]].reset_index(drop=True)
|
168
|
-
future["y"] = None
|
169
|
-
forecast = model.predict(future)
|
170
|
-
return forecast['yhat1'].tolist()
|
171
|
-
elif model_name == SupportedModels.Arima and series_id in model_object:
|
172
|
-
model = model_object[series_id]['model']
|
173
|
-
label_encoder = model_object[series_id]['le']
|
174
|
-
predict_cols = model_object[series_id]["predict_component_cols"]
|
175
|
-
encoded_df = label_encoder.transform(future_df)
|
176
|
-
x_pred = encoded_df.set_index(date_col_name)
|
177
|
-
x_pred = x_pred.drop(target_cat_col, axis=1)
|
178
|
-
yhat, conf_int = model.predict(
|
179
|
-
n_periods=horizon,
|
180
|
-
X=x_pred[predict_cols],
|
181
|
-
return_conf_int=True
|
182
|
-
)
|
183
|
-
yhat_clean = pd.DataFrame(yhat, index=yhat.index, columns=["yhat"])
|
184
|
-
return yhat_clean['yhat'].tolist()
|
185
|
-
elif model_name == SupportedModels.AutoMLX and series_id in model_object:
|
186
|
-
model = model_object[series_id]['model']
|
187
|
-
label_encoder = model_object[series_id]['le']
|
188
|
-
encoded_df = label_encoder.transform(future_df)
|
189
|
-
x_pred = encoded_df.drop(target_cat_col, axis=1)
|
190
|
-
x_pred = x_pred.set_index(date_col_name)
|
191
|
-
forecast = model.forecast(
|
192
|
-
X=x_pred,
|
193
|
-
periods=horizon
|
194
|
-
)
|
195
|
-
return forecast[target_column].tolist()
|
196
|
-
else:
|
197
|
-
raise Exception(f"Invalid model object type: {type(model_object).__name__}.")
|
198
|
-
|
199
|
-
|
200
|
-
def predict(data, model=load_model()) -> dict:
|
201
|
-
"""
|
202
|
-
Returns prediction given the model and data to predict
|
203
|
-
|
204
|
-
Parameters
|
205
|
-
----------
|
206
|
-
model: Model instance returned by load_model API
|
207
|
-
data: Data format as expected by the predict API of the core estimator. For eg. in case of sckit models it could be numpy array/List of list/Panda DataFrame
|
208
|
-
|
209
|
-
Returns
|
210
|
-
-------
|
211
|
-
predictions: Output from scoring server
|
212
|
-
Format: { 'prediction': output from `model.predict` method }
|
213
|
-
|
214
|
-
"""
|
215
|
-
assert model is not None, "Model is not loaded"
|
216
|
-
|
217
|
-
models = model["models"]
|
218
|
-
specs = model["spec"]
|
219
|
-
horizon = specs["horizon"]
|
220
|
-
model_name = specs["model"]
|
221
|
-
date_col = specs["datetime_column"]
|
222
|
-
target_column = specs["target_column"]
|
223
|
-
target_category_column = specs["target_category_columns"][0]
|
224
|
-
|
225
|
-
try:
|
226
|
-
input_data = InputData(**data["additional_data"])
|
227
|
-
except TypeError as e:
|
228
|
-
raise ValueError(f"Validation error: {e}")
|
229
|
-
additional_data = load_data(input_data)
|
230
|
-
|
231
|
-
unique_values = additional_data[target_category_column].unique()
|
232
|
-
forecasts = {}
|
233
|
-
for key in unique_values:
|
234
|
-
try:
|
235
|
-
s_id = str(key)
|
236
|
-
filtered = additional_data[additional_data[target_category_column] == key]
|
237
|
-
future = filtered.tail(horizon)
|
238
|
-
forecast = get_forecast(future, model_name, s_id, models, date_col,
|
239
|
-
target_column, target_category_column, horizon)
|
240
|
-
forecasts[s_id] = json.dumps(forecast)
|
241
|
-
except Exception as e:
|
242
|
-
raise RuntimeError(
|
243
|
-
f"An error occurred during prediction: {e}."
|
244
|
-
f" Please ensure the input data matches the format and structure of the data used during training.")
|
245
|
-
|
246
|
-
return {'prediction': json.dumps(forecasts)}
|
@@ -1,17 +0,0 @@
|
|
1
|
-
type: pii
|
2
|
-
version: v1
|
3
|
-
name: PII Operator
|
4
|
-
conda_type: published
|
5
|
-
conda: pii_v1
|
6
|
-
gpu: no
|
7
|
-
keywords:
|
8
|
-
- PII
|
9
|
-
- Spacy
|
10
|
-
backends:
|
11
|
-
- job
|
12
|
-
- operator.local
|
13
|
-
description: |
|
14
|
-
PII operator, that detects detect and redact Personally Identifiable Information
|
15
|
-
(PII) data in datasets by combining pattern match and machine learning solution.
|
16
|
-
Use `ads operator info -t pii` to get more details about the pii operator."
|
17
|
-
logo: data:image/svg+xml;base64,PHN2ZyBpZD0ic3ZnIiB2ZXJzaW9uPSIxLjEiIHhtbG5zPSJodHRwOi8vd3d3LnczLm9yZy8yMDAwL3N2ZyIgeG1sbnM6eGxpbms9Imh0dHA6Ly93d3cudzMub3JnLzE5OTkveGxpbmsiIHdpZHRoPSI1MCIKICBoZWlnaHQ9IjUwLjk0ODQ1MzYwODI0NzQ2IiB2aWV3Qm94PSIwLCAwLCA0MDAsMzY0Ljk0ODQ1MzYwODI0NzQ2Ij4KICA8ZyBpZD0ic3ZnZyI+CiAgICA8cGF0aCBpZD0icGF0aDAiCiAgICAgIGQ9Ik0xODIuNDc0IDQuMTk0IEMgMTI2LjI1NyAxMC4yODIsNjkuOTExIDQ0LjA5MCwyNy4zNzggOTcuMjUxIEMgMjIuMjM3IDEwMy42NzcsMTUuMDY1IDExMi4yOTgsMTEuNDQwIDExNi40MDkgQyAtMy4wMjcgMTMyLjgxOSwtMi41MTYgMTM2Ljk2MywxNi43OTIgMTU5Ljc5NCBDIDc5LjAzMCAyMzMuMzg3LDEzNS43MjMgMjY2LjI4NiwyMDAuMzQ0IDI2Ni4zMDcgQyAyNjIuNTYyIDI2Ni4zMjcsMzI0LjI1NSAyMzEuNjc1LDM3NS45NDkgMTY3LjY3MyBDIDM3OS4xNDQgMTYzLjcxOCwzODQuNDc5IDE1Ny4zODgsMzg3LjgwNCAxNTMuNjA4IEMgNDA0LjEwNSAxMzUuMDc3LDQwMi41OTggMTMxLjEwMSwzNjIuMTUzIDg1LjkxMSBDIDMwOS4zNjEgMjYuOTI1LDI0NC41OTAgLTIuNTMyLDE4Mi40NzQgNC4xOTQgTTIxNC40MzcgMzQuMDA4IEMgMjM0LjkwOSAzNi40MzYsMjYzLjkxOCA0NC43MDcsMjYzLjkxOCA0OC4xMTUgQyAyNjMuOTE4IDQ4LjY3MiwyNDQuNDg1IDcwLjI0OCwyMzIuNjYxIDgyLjgxOCBDIDIyNy42ODMgODguMTEwLDIxOS45MjEgOTYuNDUwLDIxNS40MTMgMTAxLjM1MiBDIDIxMC45MDUgMTA2LjI1MywyMDQuNTg4IDExMy4wNjIsMjAxLjM3NSAxMTYuNDgzIEMgMTk4LjE2MiAxMTkuOTAzLDE5Mi4yODUgMTI2LjI1MSwxODguMzE2IDEzMC41ODggQyAxODQuMzQ3IDEzNC45MjYsMTc3LjA3OSAxNDIuNzU4LDE3Mi4xNjUgMTQ3Ljk5NCBDIDE2MS4zMDcgMTU5LjU2MiwxNDEuOTEwIDE4MC40NDMsMTMyLjQ3NCAxOTAuNzIyIEMgMTI4LjY1NyAxOTQuODgwLDEyMi44MzMgMjAxLjA5NywxMTkuNTMxIDIwNC41MzcgTCAxMTMuNTI5IDIxMC43OTMgMTA0LjcwMiAyMDQuMjYzIEMgODcuMDMxIDE5MS4xODksNjYuNzk0IDE3Mi4wMDksNDguNzE4IDE1MS4yMDMgQyAzNC43NjMgMTM1LjE0MCwzNC42MjggMTM0Ljk1NCwzNS44NTggMTMzLjUxMCBDIDM2LjQyNCAxMzIuODQ2LDQwLjg3OSAxMjcuNTA5LDQ1Ljc1NyAxMjEuNjQ5IEMgNjAuMDIzIDEwNC41MTUsNzUuOTI3IDg3Ljk0Myw4Ni45NDIgNzguNzM0IEMgOTUuNjI4IDcxLjQ3MiwxMDkuMTU0IDYxLjE2OCwxMTAuMDAwIDYxLjE2OCBDIDExMC4zMTYgNjEuMTY4LDExNi45MDkgNjcuNTM5LDEyNC42NTMgNzUuMzI2IEwgMTM4LjczMiA4OS40ODMgMTM2LjM3OCA5Mi43NTkgQyAxMjQuMzU5IDEwOS40ODksMTIwLjExOSAxMzguMzMyLDEyNi41ODYgMTU5LjM3MiBDIDEyOC4xNzcgMTY0LjU1MCwxMzIuNjA1IDE3NC41NzAsMTMzLjMwMSAxNzQuNTcwIEMgMTMzLjU1NiAxNzQuNTcwLDEzOS4wMDIgMTY4Ljk0MCwxNDUuNDA1IDE2Mi4wNTkgQyAxNTEuODA3IDE1NS4xNzgsMTU5LjEzMiAxNDcuMzQ1LDE2MS42ODQgMTQ0LjY1MyBDIDE2NS42MTUgMTQwLjUwNiwxNjYuMzIzIDEzOS4zODcsMTY2LjMyMyAxMzcuMzI3IEMgMTY2LjMyMyAxMjEuOTgwLDE3Ny41NTkgMTA4LjYzOCwxOTIuNDYwIDEwNi4yOTIgTCAxOTcuMzY3IDEwNS41MTkgMjA1LjcyOCA5Ni44OTggQyAyMTkuOTEzIDgyLjI3MywyMzEuNjE1IDY5LjYyOCwyMzEuNjE1IDY4LjkyNyBDIDIzMS42MTUgNjguNTU5LDIyOC45MDkgNjcuNDQwLDIyNS42MDEgNjYuNDQwIEMgMjA4LjExNyA2MS4xNTMsMTgyLjgzMyA2Mi4zMzUsMTY3LjY1MSA2OS4xNDggTCAxNjMuNDcwIDcxLjAyNSAxNTEuMzIyIDU4Ljg2NCBDIDE0MS41NzkgNDkuMTEwLDEzOS4zNzkgNDYuNTcyLDE0MC4yMDYgNDYuMDM1IEMgMTUzLjc5NSAzNy4yMjcsMTkwLjg5NCAzMS4yMTYsMjE0LjQzNyAzNC4wMDggTTI5Ny40NTIgNjYuMDk4IEMgMzEyLjY0MSA3Ny4yMDEsMzM3LjU1MCAxMDEuOTgxLDM1Ny42MzYgMTI1Ljk2OSBMIDM2NC45OTQgMTM0Ljc1NiAzNjEuMTA5IDEzOS4zNzEgQyAzNTguOTcyIDE0MS45MDksMzU2LjA5NiAxNDUuMjIzLDM1NC43MTcgMTQ2LjczNSBDIDM1My4zMzkgMTQ4LjI0NywzNDguNjMwIDE1My41MDUsMzQ0LjI1NCAxNTguNDE5IEMgMzIyLjA1MSAxODMuMzUxLDI5MC4wNjkgMjExLjQ4MCwyODcuMzI2IDIwOC40ODggQyAyODYuNjg1IDIwNy43ODksMjgwLjY4MSAyMDEuNjE1LDI3My45ODUgMTk0Ljc2OCBMIDI2MS44MTAgMTgyLjMyMCAyNjQuMjcwIDE3OC42MTcgQyAyNzguMjU3IDE1Ny41NjYsMjgwLjg1MiAxMjUuNDI1LDI3MC4yMzYgMTA0LjczMyBMIDI2OS4wMjkgMTAyLjM4MCAyNTEuNTU4IDEyMS4zNDcgQyAyMzQuNDI1IDEzOS45NDgsMjM0LjA2NiAxNDAuMzk4LDIzMy4wMTAgMTQ0LjU5MSBDIDIzMC42MDMgMTU0LjE1NywyMjEuMDI0IDE2My41NDAsMjEzLjY0MSAxNjMuNTY3IEMgMjEyLjQxMiAxNjMuNTcyLDIxMS4xNzkgMTY0LjMzOCwyMDkuODYwIDE2NS45MTYgQyAyMDguNzg0IDE2Ny4yMDQsMjAyLjgwMSAxNzMuODkwLDE5Ni41NjQgMTgwLjc3NCBDIDE3My42MjAgMjA2LjA5OCwxNzQuNzQ5IDIwNC4yNTAsMTgxLjI3MSAyMDUuODA1IEMgMTk3Ljg4OSAyMDkuNzY4LDIxNi4wMTYgMjA4LjMxNSwyMzMuMjQxIDIwMS42NDAgTCAyMzYuNTg2IDIwMC4zNDQgMjQ3Ljg3OSAyMTEuOTI5IEMgMjYwLjU2NyAyMjQuOTQ1LDI2MC4zMTYgMjI0LjM0MSwyNTQuMDg3IDIyNi44OTQgQyAyMjYuODU5IDIzOC4wNTMsMTkxLjU5NyAyMzkuOTkzLDE2My4yMzAgMjMxLjg5MiBDIDE1Ni4xNzQgMjI5Ljg3NywxNDQuMjk0IDIyNS41NzUsMTQzLjM2MCAyMjQuNjk3IEMgMTQyLjk5NyAyMjQuMzU1LDE0Ni43NDAgMjE5Ljg4MCwxNTIuMjk1IDIxNC4wMTggQyAxNjEuODI5IDIwMy45NTYsMTYzLjQyMCAyMDIuMjM0LDE4MS43MjcgMTgyLjE3NiBDIDIwMC45MzUgMTYxLjEzMSwyMTUuNTg3IDE0NS4yMzQsMjIxLjk5NCAxMzguNDkwIEMgMjI1LjU4NiAxMzQuNzA5LDIzMC4wMzggMTI5LjkxNCwyMzEuODg4IDEyNy44MzUgQyAyMzMuNzM4IDEyNS43NTYsMjQwLjEwMiAxMTguOTUyLDI0Ni4wMzEgMTEyLjcxNSBDIDI1MS45NjAgMTA2LjQ3OCwyNTguNzk3IDk5LjIxNiwyNjEuMjIzIDk2LjU3OCBDIDI2My42NTAgOTMuOTQwLDI2OS4wMzggODguMjE4LDI3My4xOTYgODMuODYzIEMgMjc3LjM1NCA3OS41MDgsMjgzLjM2NCA3My4wODQsMjg2LjU1MSA2OS41ODggQyAyOTMuMjk4IDYyLjE4NSwyOTIuNDY1IDYyLjQ1MiwyOTcuNDUyIDY2LjA5OCBNMjAyLjQwNSAyOTYuMDI1IEMgMjAxLjY0OSAyOTYuMjA4LDIwMC4zMDYgMjk3LjAyMiwxOTkuNDE5IDI5Ny44MzUgQyAxOTUuOTMzIDMwMS4wMzEsMTk1LjkxNyAzMDEuMTc1LDE5NS43MzUgMzI5LjczNyBMIDE5NS41NjQgMzU2LjUwNiAxOTcuMzc3IDM1OS4yNDUgQyAyMDEuNzkxIDM2NS45MTUsMjA5Ljk5NSAzNjUuNjQzLDIxNC4wODkgMzU4LjY5MSBDIDIxNS4zOTkgMzU2LjQ2OCwyMTUuNDY0IDM1NS4xMjEsMjE1LjQ2NCAzMzAuMjQxIEMgMjE1LjQ2NCAzMDUuNDk1LDIxNS4zOTIgMzAzLjk4MCwyMTQuMDg5IDMwMS4zODQgQyAyMTEuNzYzIDI5Ni43NDcsMjA3LjUwMSAyOTQuNzkyLDIwMi40MDUgMjk2LjAyNSBNMjM5Ljg4NiAyOTYuNzM2IEMgMjM0Ljc3NSAyOTkuMzE1LDIzNC43MzQgMjk5LjU1OCwyMzQuNDkyIDMyOC45NTUgQyAyMzQuMjUyIDM1OC4xNDgsMjM0LjI5MCAzNTguNDM4LDIzOC43MzMgMzYxLjgyNyBDIDI0My42ODUgMzY1LjYwNCwyNDkuNjU0IDM2NC40MTQsMjUyLjczOCAzNTkuMDM1IEMgMjU1LjE0NyAzNTQuODMyLDI1NS4xNTcgMzA1Ljc1MSwyNTIuNzQ5IDMwMS4wNTYgQyAyNTAuMTU4IDI5Ni4wMDIsMjQ0Ljg2NSAyOTQuMjI1LDIzOS44ODYgMjk2LjczNiBNMTQxLjA1OSAyOTcuMTMxIEMgMTM3LjA5NSAyOTkuMzMyLDEzNy4xMzIgMjk5LjA1MSwxMzYuODkyIDMyOC4zMTIgQyAxMzYuNjQzIDM1OC42NjksMTM2LjgxMiAzNTkuODgyLDE0MS43MDYgMzYyLjg2NiBDIDE0OS43OTMgMzY3Ljc5NiwxNTUuMzI2IDM2Mi4xNDEsMTU1LjMyNiAzNDguOTQ2IEwgMTU1LjMyNiAzNDEuMDU1IDE2MC42NTMgMzQwLjU5NCBDIDE4NC40NzIgMzM4LjUzMSwxOTAuMjYxIDMwOC4zNTgsMTY4Ljc4NyAyOTguMTk5IEMgMTY0LjAyNyAyOTUuOTQ4LDE0NC41NDIgMjk1LjE5NywxNDEuMDU5IDI5Ny4xMzEgTTE2My4yNjEgMzE1LjA4OSBDIDE2Ni44NjcgMzE4LjY5NSwxNjQuNTM1IDMyMy4yMzcsMTU5LjAzNiAzMjMuMzE4IEwgMTU1LjY3MCAzMjMuMzY4IDE1NS40NjEgMzE5LjMyMSBDIDE1NS4xMzEgMzEyLjkzMCwxNTkuMDAyIDMxMC44MzAsMTYzLjI2MSAzMTUuMDg5ICIKICAgICAgc3Ryb2tlPSJub25lIiBmaWxsPSIjMjk0NDY1IiBmaWxsLXJ1bGU9ImV2ZW5vZGQiPjwvcGF0aD4KICA8L2c+Cjwvc3ZnPg==
|