oracle-ads 2.13.9rc0__py3-none-any.whl → 2.13.10__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.
- ads/aqua/__init__.py +40 -0
- ads/aqua/app.py +507 -0
- ads/aqua/cli.py +96 -0
- ads/aqua/client/__init__.py +3 -0
- ads/aqua/client/client.py +836 -0
- ads/aqua/client/openai_client.py +305 -0
- ads/aqua/common/__init__.py +5 -0
- ads/aqua/common/decorator.py +125 -0
- ads/aqua/common/entities.py +274 -0
- ads/aqua/common/enums.py +134 -0
- ads/aqua/common/errors.py +109 -0
- ads/aqua/common/utils.py +1295 -0
- ads/aqua/config/__init__.py +4 -0
- ads/aqua/config/container_config.py +246 -0
- ads/aqua/config/evaluation/__init__.py +4 -0
- ads/aqua/config/evaluation/evaluation_service_config.py +147 -0
- ads/aqua/config/utils/__init__.py +4 -0
- ads/aqua/config/utils/serializer.py +339 -0
- ads/aqua/constants.py +116 -0
- ads/aqua/data.py +14 -0
- ads/aqua/dummy_data/icon.txt +1 -0
- ads/aqua/dummy_data/oci_model_deployments.json +56 -0
- ads/aqua/dummy_data/oci_models.json +1 -0
- ads/aqua/dummy_data/readme.md +26 -0
- ads/aqua/evaluation/__init__.py +8 -0
- ads/aqua/evaluation/constants.py +53 -0
- ads/aqua/evaluation/entities.py +186 -0
- ads/aqua/evaluation/errors.py +70 -0
- ads/aqua/evaluation/evaluation.py +1814 -0
- ads/aqua/extension/__init__.py +42 -0
- ads/aqua/extension/aqua_ws_msg_handler.py +76 -0
- ads/aqua/extension/base_handler.py +90 -0
- ads/aqua/extension/common_handler.py +121 -0
- ads/aqua/extension/common_ws_msg_handler.py +36 -0
- ads/aqua/extension/deployment_handler.py +381 -0
- ads/aqua/extension/deployment_ws_msg_handler.py +54 -0
- ads/aqua/extension/errors.py +30 -0
- ads/aqua/extension/evaluation_handler.py +129 -0
- ads/aqua/extension/evaluation_ws_msg_handler.py +61 -0
- ads/aqua/extension/finetune_handler.py +96 -0
- ads/aqua/extension/model_handler.py +390 -0
- ads/aqua/extension/models/__init__.py +0 -0
- ads/aqua/extension/models/ws_models.py +145 -0
- ads/aqua/extension/models_ws_msg_handler.py +50 -0
- ads/aqua/extension/ui_handler.py +300 -0
- ads/aqua/extension/ui_websocket_handler.py +130 -0
- ads/aqua/extension/utils.py +133 -0
- ads/aqua/finetuning/__init__.py +7 -0
- ads/aqua/finetuning/constants.py +23 -0
- ads/aqua/finetuning/entities.py +181 -0
- ads/aqua/finetuning/finetuning.py +749 -0
- ads/aqua/model/__init__.py +8 -0
- ads/aqua/model/constants.py +60 -0
- ads/aqua/model/entities.py +385 -0
- ads/aqua/model/enums.py +32 -0
- ads/aqua/model/model.py +2134 -0
- ads/aqua/model/utils.py +52 -0
- ads/aqua/modeldeployment/__init__.py +6 -0
- ads/aqua/modeldeployment/constants.py +10 -0
- ads/aqua/modeldeployment/deployment.py +1315 -0
- ads/aqua/modeldeployment/entities.py +653 -0
- ads/aqua/modeldeployment/utils.py +543 -0
- ads/aqua/resources/gpu_shapes_index.json +94 -0
- ads/aqua/server/__init__.py +4 -0
- ads/aqua/server/__main__.py +24 -0
- ads/aqua/server/app.py +47 -0
- ads/aqua/server/aqua_spec.yml +1291 -0
- ads/aqua/training/__init__.py +4 -0
- ads/aqua/training/exceptions.py +476 -0
- ads/aqua/ui.py +519 -0
- ads/automl/__init__.py +9 -0
- ads/automl/driver.py +330 -0
- ads/automl/provider.py +975 -0
- ads/bds/__init__.py +5 -0
- ads/bds/auth.py +127 -0
- ads/bds/big_data_service.py +255 -0
- ads/catalog/__init__.py +19 -0
- ads/catalog/model.py +1576 -0
- ads/catalog/notebook.py +461 -0
- ads/catalog/project.py +468 -0
- ads/catalog/summary.py +178 -0
- ads/common/__init__.py +11 -0
- ads/common/analyzer.py +65 -0
- ads/common/artifact/.model-ignore +63 -0
- ads/common/artifact/__init__.py +10 -0
- ads/common/auth.py +1122 -0
- ads/common/card_identifier.py +83 -0
- ads/common/config.py +647 -0
- ads/common/data.py +165 -0
- ads/common/decorator/__init__.py +9 -0
- ads/common/decorator/argument_to_case.py +88 -0
- ads/common/decorator/deprecate.py +69 -0
- ads/common/decorator/require_nonempty_arg.py +65 -0
- ads/common/decorator/runtime_dependency.py +178 -0
- ads/common/decorator/threaded.py +97 -0
- ads/common/decorator/utils.py +35 -0
- ads/common/dsc_file_system.py +303 -0
- ads/common/error.py +14 -0
- ads/common/extended_enum.py +81 -0
- ads/common/function/__init__.py +5 -0
- ads/common/function/fn_util.py +142 -0
- ads/common/function/func_conf.yaml +25 -0
- ads/common/ipython.py +76 -0
- ads/common/model.py +679 -0
- ads/common/model_artifact.py +1759 -0
- ads/common/model_artifact_schema.json +107 -0
- ads/common/model_export_util.py +664 -0
- ads/common/model_metadata.py +24 -0
- ads/common/object_storage_details.py +296 -0
- ads/common/oci_client.py +179 -0
- ads/common/oci_datascience.py +46 -0
- ads/common/oci_logging.py +1144 -0
- ads/common/oci_mixin.py +957 -0
- ads/common/oci_resource.py +136 -0
- ads/common/serializer.py +559 -0
- ads/common/utils.py +1852 -0
- ads/common/word_lists.py +1491 -0
- ads/common/work_request.py +189 -0
- ads/config.py +1 -0
- ads/data_labeling/__init__.py +13 -0
- ads/data_labeling/boundingbox.py +253 -0
- ads/data_labeling/constants.py +47 -0
- ads/data_labeling/data_labeling_service.py +244 -0
- ads/data_labeling/interface/__init__.py +5 -0
- ads/data_labeling/interface/loader.py +16 -0
- ads/data_labeling/interface/parser.py +16 -0
- ads/data_labeling/interface/reader.py +23 -0
- ads/data_labeling/loader/__init__.py +5 -0
- ads/data_labeling/loader/file_loader.py +241 -0
- ads/data_labeling/metadata.py +110 -0
- ads/data_labeling/mixin/__init__.py +5 -0
- ads/data_labeling/mixin/data_labeling.py +232 -0
- ads/data_labeling/ner.py +129 -0
- ads/data_labeling/parser/__init__.py +5 -0
- ads/data_labeling/parser/dls_record_parser.py +388 -0
- ads/data_labeling/parser/export_metadata_parser.py +94 -0
- ads/data_labeling/parser/export_record_parser.py +473 -0
- ads/data_labeling/reader/__init__.py +5 -0
- ads/data_labeling/reader/dataset_reader.py +574 -0
- ads/data_labeling/reader/dls_record_reader.py +121 -0
- ads/data_labeling/reader/export_record_reader.py +62 -0
- ads/data_labeling/reader/jsonl_reader.py +75 -0
- ads/data_labeling/reader/metadata_reader.py +203 -0
- ads/data_labeling/reader/record_reader.py +263 -0
- ads/data_labeling/record.py +52 -0
- ads/data_labeling/visualizer/__init__.py +5 -0
- ads/data_labeling/visualizer/image_visualizer.py +525 -0
- ads/data_labeling/visualizer/text_visualizer.py +357 -0
- ads/database/__init__.py +5 -0
- ads/database/connection.py +338 -0
- ads/dataset/__init__.py +10 -0
- ads/dataset/capabilities.md +51 -0
- ads/dataset/classification_dataset.py +339 -0
- ads/dataset/correlation.py +226 -0
- ads/dataset/correlation_plot.py +563 -0
- ads/dataset/dask_series.py +173 -0
- ads/dataset/dataframe_transformer.py +110 -0
- ads/dataset/dataset.py +1979 -0
- ads/dataset/dataset_browser.py +360 -0
- ads/dataset/dataset_with_target.py +995 -0
- ads/dataset/exception.py +25 -0
- ads/dataset/factory.py +987 -0
- ads/dataset/feature_engineering_transformer.py +35 -0
- ads/dataset/feature_selection.py +107 -0
- ads/dataset/forecasting_dataset.py +26 -0
- ads/dataset/helper.py +1450 -0
- ads/dataset/label_encoder.py +99 -0
- ads/dataset/mixin/__init__.py +5 -0
- ads/dataset/mixin/dataset_accessor.py +134 -0
- ads/dataset/pipeline.py +58 -0
- ads/dataset/plot.py +710 -0
- ads/dataset/progress.py +86 -0
- ads/dataset/recommendation.py +297 -0
- ads/dataset/recommendation_transformer.py +502 -0
- ads/dataset/regression_dataset.py +14 -0
- ads/dataset/sampled_dataset.py +1050 -0
- ads/dataset/target.py +98 -0
- ads/dataset/timeseries.py +18 -0
- ads/dbmixin/__init__.py +5 -0
- ads/dbmixin/db_pandas_accessor.py +153 -0
- ads/environment/__init__.py +9 -0
- ads/environment/ml_runtime.py +66 -0
- ads/evaluations/README.md +14 -0
- ads/evaluations/__init__.py +109 -0
- ads/evaluations/evaluation_plot.py +983 -0
- ads/evaluations/evaluator.py +1334 -0
- ads/evaluations/statistical_metrics.py +543 -0
- ads/experiments/__init__.py +9 -0
- ads/experiments/capabilities.md +0 -0
- ads/explanations/__init__.py +21 -0
- ads/explanations/base_explainer.py +142 -0
- ads/explanations/capabilities.md +83 -0
- ads/explanations/explainer.py +190 -0
- ads/explanations/mlx_global_explainer.py +1050 -0
- ads/explanations/mlx_interface.py +386 -0
- ads/explanations/mlx_local_explainer.py +287 -0
- ads/explanations/mlx_whatif_explainer.py +201 -0
- ads/feature_engineering/__init__.py +20 -0
- ads/feature_engineering/accessor/__init__.py +5 -0
- ads/feature_engineering/accessor/dataframe_accessor.py +535 -0
- ads/feature_engineering/accessor/mixin/__init__.py +5 -0
- ads/feature_engineering/accessor/mixin/correlation.py +166 -0
- ads/feature_engineering/accessor/mixin/eda_mixin.py +266 -0
- ads/feature_engineering/accessor/mixin/eda_mixin_series.py +85 -0
- ads/feature_engineering/accessor/mixin/feature_types_mixin.py +211 -0
- ads/feature_engineering/accessor/mixin/utils.py +65 -0
- ads/feature_engineering/accessor/series_accessor.py +431 -0
- ads/feature_engineering/adsimage/__init__.py +5 -0
- ads/feature_engineering/adsimage/image.py +192 -0
- ads/feature_engineering/adsimage/image_reader.py +170 -0
- ads/feature_engineering/adsimage/interface/__init__.py +5 -0
- ads/feature_engineering/adsimage/interface/reader.py +19 -0
- ads/feature_engineering/adsstring/__init__.py +7 -0
- ads/feature_engineering/adsstring/oci_language/__init__.py +8 -0
- ads/feature_engineering/adsstring/string/__init__.py +8 -0
- ads/feature_engineering/data_schema.json +57 -0
- ads/feature_engineering/dataset/__init__.py +5 -0
- ads/feature_engineering/dataset/zip_code_data.py +42062 -0
- ads/feature_engineering/exceptions.py +40 -0
- ads/feature_engineering/feature_type/__init__.py +133 -0
- ads/feature_engineering/feature_type/address.py +184 -0
- ads/feature_engineering/feature_type/adsstring/__init__.py +5 -0
- ads/feature_engineering/feature_type/adsstring/common_regex_mixin.py +164 -0
- ads/feature_engineering/feature_type/adsstring/oci_language.py +93 -0
- ads/feature_engineering/feature_type/adsstring/parsers/__init__.py +5 -0
- ads/feature_engineering/feature_type/adsstring/parsers/base.py +47 -0
- ads/feature_engineering/feature_type/adsstring/parsers/nltk_parser.py +96 -0
- ads/feature_engineering/feature_type/adsstring/parsers/spacy_parser.py +221 -0
- ads/feature_engineering/feature_type/adsstring/string.py +258 -0
- ads/feature_engineering/feature_type/base.py +58 -0
- ads/feature_engineering/feature_type/boolean.py +183 -0
- ads/feature_engineering/feature_type/category.py +146 -0
- ads/feature_engineering/feature_type/constant.py +137 -0
- ads/feature_engineering/feature_type/continuous.py +151 -0
- ads/feature_engineering/feature_type/creditcard.py +314 -0
- ads/feature_engineering/feature_type/datetime.py +190 -0
- ads/feature_engineering/feature_type/discrete.py +134 -0
- ads/feature_engineering/feature_type/document.py +43 -0
- ads/feature_engineering/feature_type/gis.py +251 -0
- ads/feature_engineering/feature_type/handler/__init__.py +5 -0
- ads/feature_engineering/feature_type/handler/feature_validator.py +524 -0
- ads/feature_engineering/feature_type/handler/feature_warning.py +319 -0
- ads/feature_engineering/feature_type/handler/warnings.py +128 -0
- ads/feature_engineering/feature_type/integer.py +142 -0
- ads/feature_engineering/feature_type/ip_address.py +144 -0
- ads/feature_engineering/feature_type/ip_address_v4.py +138 -0
- ads/feature_engineering/feature_type/ip_address_v6.py +138 -0
- ads/feature_engineering/feature_type/lat_long.py +256 -0
- ads/feature_engineering/feature_type/object.py +43 -0
- ads/feature_engineering/feature_type/ordinal.py +132 -0
- ads/feature_engineering/feature_type/phone_number.py +135 -0
- ads/feature_engineering/feature_type/string.py +171 -0
- ads/feature_engineering/feature_type/text.py +93 -0
- ads/feature_engineering/feature_type/unknown.py +43 -0
- ads/feature_engineering/feature_type/zip_code.py +164 -0
- ads/feature_engineering/feature_type_manager.py +406 -0
- ads/feature_engineering/schema.py +795 -0
- ads/feature_engineering/utils.py +245 -0
- ads/feature_store/.readthedocs.yaml +19 -0
- ads/feature_store/README.md +65 -0
- ads/feature_store/__init__.py +9 -0
- ads/feature_store/common/__init__.py +0 -0
- ads/feature_store/common/enums.py +339 -0
- ads/feature_store/common/exceptions.py +18 -0
- ads/feature_store/common/spark_session_singleton.py +125 -0
- ads/feature_store/common/utils/__init__.py +0 -0
- ads/feature_store/common/utils/base64_encoder_decoder.py +72 -0
- ads/feature_store/common/utils/feature_schema_mapper.py +283 -0
- ads/feature_store/common/utils/transformation_utils.py +82 -0
- ads/feature_store/common/utils/utility.py +403 -0
- ads/feature_store/data_validation/__init__.py +0 -0
- ads/feature_store/data_validation/great_expectation.py +129 -0
- ads/feature_store/dataset.py +1230 -0
- ads/feature_store/dataset_job.py +530 -0
- ads/feature_store/docs/Dockerfile +7 -0
- ads/feature_store/docs/Makefile +44 -0
- ads/feature_store/docs/conf.py +28 -0
- ads/feature_store/docs/requirements.txt +14 -0
- ads/feature_store/docs/source/ads.feature_store.query.rst +20 -0
- ads/feature_store/docs/source/cicd.rst +137 -0
- ads/feature_store/docs/source/conf.py +86 -0
- ads/feature_store/docs/source/data_versioning.rst +33 -0
- ads/feature_store/docs/source/dataset.rst +388 -0
- ads/feature_store/docs/source/dataset_job.rst +27 -0
- ads/feature_store/docs/source/demo.rst +70 -0
- ads/feature_store/docs/source/entity.rst +78 -0
- ads/feature_store/docs/source/feature_group.rst +624 -0
- ads/feature_store/docs/source/feature_group_job.rst +29 -0
- ads/feature_store/docs/source/feature_store.rst +122 -0
- ads/feature_store/docs/source/feature_store_class.rst +123 -0
- ads/feature_store/docs/source/feature_validation.rst +66 -0
- 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 +81 -0
- ads/feature_store/docs/source/module.rst +8 -0
- ads/feature_store/docs/source/notebook.rst +94 -0
- ads/feature_store/docs/source/overview.rst +47 -0
- ads/feature_store/docs/source/quickstart.rst +176 -0
- ads/feature_store/docs/source/release_notes.rst +194 -0
- ads/feature_store/docs/source/setup_feature_store.rst +81 -0
- ads/feature_store/docs/source/statistics.rst +58 -0
- ads/feature_store/docs/source/transformation.rst +199 -0
- ads/feature_store/docs/source/ui.rst +65 -0
- ads/feature_store/docs/source/user_guides.setup.feature_store_operator.rst +66 -0
- ads/feature_store/docs/source/user_guides.setup.helm_chart.rst +192 -0
- ads/feature_store/docs/source/user_guides.setup.terraform.rst +338 -0
- ads/feature_store/entity.py +718 -0
- 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 +375 -0
- ads/feature_store/execution_strategy/engine/__init__.py +0 -0
- ads/feature_store/execution_strategy/engine/spark_engine.py +316 -0
- ads/feature_store/execution_strategy/execution_strategy.py +113 -0
- ads/feature_store/execution_strategy/execution_strategy_provider.py +47 -0
- ads/feature_store/execution_strategy/spark/__init__.py +0 -0
- ads/feature_store/execution_strategy/spark/spark_execution.py +618 -0
- ads/feature_store/feature.py +192 -0
- ads/feature_store/feature_group.py +1494 -0
- ads/feature_store/feature_group_expectation.py +346 -0
- ads/feature_store/feature_group_job.py +602 -0
- ads/feature_store/feature_lineage/__init__.py +0 -0
- ads/feature_store/feature_lineage/graphviz_service.py +180 -0
- ads/feature_store/feature_option_details.py +50 -0
- ads/feature_store/feature_statistics/__init__.py +0 -0
- ads/feature_store/feature_statistics/statistics_service.py +99 -0
- ads/feature_store/feature_store.py +699 -0
- ads/feature_store/feature_store_registrar.py +518 -0
- ads/feature_store/input_feature_detail.py +149 -0
- ads/feature_store/mixin/__init__.py +4 -0
- ads/feature_store/mixin/oci_feature_store.py +145 -0
- ads/feature_store/model_details.py +73 -0
- ads/feature_store/query/__init__.py +0 -0
- ads/feature_store/query/filter.py +266 -0
- ads/feature_store/query/generator/__init__.py +0 -0
- ads/feature_store/query/generator/query_generator.py +298 -0
- ads/feature_store/query/join.py +161 -0
- ads/feature_store/query/query.py +403 -0
- ads/feature_store/query/validator/__init__.py +0 -0
- ads/feature_store/query/validator/query_validator.py +57 -0
- ads/feature_store/response/__init__.py +0 -0
- ads/feature_store/response/response_builder.py +68 -0
- ads/feature_store/service/__init__.py +0 -0
- ads/feature_store/service/oci_dataset.py +139 -0
- ads/feature_store/service/oci_dataset_job.py +199 -0
- ads/feature_store/service/oci_entity.py +125 -0
- ads/feature_store/service/oci_feature_group.py +164 -0
- ads/feature_store/service/oci_feature_group_job.py +214 -0
- ads/feature_store/service/oci_feature_store.py +182 -0
- ads/feature_store/service/oci_lineage.py +87 -0
- ads/feature_store/service/oci_transformation.py +104 -0
- ads/feature_store/statistics/__init__.py +0 -0
- ads/feature_store/statistics/abs_feature_value.py +49 -0
- ads/feature_store/statistics/charts/__init__.py +0 -0
- ads/feature_store/statistics/charts/abstract_feature_plot.py +37 -0
- ads/feature_store/statistics/charts/box_plot.py +148 -0
- ads/feature_store/statistics/charts/frequency_distribution.py +65 -0
- ads/feature_store/statistics/charts/probability_distribution.py +68 -0
- ads/feature_store/statistics/charts/top_k_frequent_elements.py +98 -0
- ads/feature_store/statistics/feature_stat.py +126 -0
- ads/feature_store/statistics/generic_feature_value.py +33 -0
- ads/feature_store/statistics/statistics.py +41 -0
- ads/feature_store/statistics_config.py +101 -0
- ads/feature_store/templates/feature_store_template.yaml +45 -0
- ads/feature_store/transformation.py +499 -0
- ads/feature_store/validation_output.py +57 -0
- ads/hpo/__init__.py +9 -0
- ads/hpo/_imports.py +91 -0
- ads/hpo/ads_search_space.py +439 -0
- ads/hpo/distributions.py +325 -0
- ads/hpo/objective.py +280 -0
- ads/hpo/search_cv.py +1657 -0
- ads/hpo/stopping_criterion.py +75 -0
- ads/hpo/tuner_artifact.py +413 -0
- ads/hpo/utils.py +91 -0
- ads/hpo/validation.py +140 -0
- ads/hpo/visualization/__init__.py +5 -0
- ads/hpo/visualization/_contour.py +23 -0
- ads/hpo/visualization/_edf.py +20 -0
- ads/hpo/visualization/_intermediate_values.py +21 -0
- ads/hpo/visualization/_optimization_history.py +25 -0
- ads/hpo/visualization/_parallel_coordinate.py +169 -0
- ads/hpo/visualization/_param_importances.py +26 -0
- ads/jobs/__init__.py +53 -0
- ads/jobs/ads_job.py +663 -0
- ads/jobs/builders/__init__.py +5 -0
- ads/jobs/builders/base.py +156 -0
- ads/jobs/builders/infrastructure/__init__.py +6 -0
- ads/jobs/builders/infrastructure/base.py +165 -0
- ads/jobs/builders/infrastructure/dataflow.py +1252 -0
- ads/jobs/builders/infrastructure/dsc_job.py +1894 -0
- ads/jobs/builders/infrastructure/dsc_job_runtime.py +1233 -0
- ads/jobs/builders/infrastructure/utils.py +65 -0
- ads/jobs/builders/runtimes/__init__.py +5 -0
- ads/jobs/builders/runtimes/artifact.py +338 -0
- ads/jobs/builders/runtimes/base.py +325 -0
- ads/jobs/builders/runtimes/container_runtime.py +242 -0
- ads/jobs/builders/runtimes/python_runtime.py +1016 -0
- ads/jobs/builders/runtimes/pytorch_runtime.py +204 -0
- ads/jobs/cli.py +104 -0
- ads/jobs/env_var_parser.py +131 -0
- ads/jobs/extension.py +160 -0
- ads/jobs/schema/__init__.py +5 -0
- ads/jobs/schema/infrastructure_schema.json +116 -0
- ads/jobs/schema/job_schema.json +42 -0
- ads/jobs/schema/runtime_schema.json +183 -0
- ads/jobs/schema/validator.py +141 -0
- ads/jobs/serializer.py +296 -0
- ads/jobs/templates/__init__.py +5 -0
- ads/jobs/templates/container.py +6 -0
- ads/jobs/templates/driver_notebook.py +177 -0
- ads/jobs/templates/driver_oci.py +500 -0
- ads/jobs/templates/driver_python.py +48 -0
- ads/jobs/templates/driver_pytorch.py +852 -0
- ads/jobs/templates/driver_utils.py +615 -0
- ads/jobs/templates/hostname_from_env.c +55 -0
- ads/jobs/templates/oci_metrics.py +181 -0
- ads/jobs/utils.py +104 -0
- ads/llm/__init__.py +28 -0
- ads/llm/autogen/__init__.py +2 -0
- ads/llm/autogen/constants.py +15 -0
- ads/llm/autogen/reports/__init__.py +2 -0
- ads/llm/autogen/reports/base.py +67 -0
- ads/llm/autogen/reports/data.py +103 -0
- ads/llm/autogen/reports/session.py +526 -0
- ads/llm/autogen/reports/templates/chat_box.html +13 -0
- ads/llm/autogen/reports/templates/chat_box_lt.html +5 -0
- ads/llm/autogen/reports/templates/chat_box_rt.html +6 -0
- ads/llm/autogen/reports/utils.py +56 -0
- ads/llm/autogen/v02/__init__.py +4 -0
- ads/llm/autogen/v02/client.py +295 -0
- ads/llm/autogen/v02/log_handlers/__init__.py +2 -0
- ads/llm/autogen/v02/log_handlers/oci_file_handler.py +83 -0
- ads/llm/autogen/v02/loggers/__init__.py +6 -0
- ads/llm/autogen/v02/loggers/metric_logger.py +320 -0
- ads/llm/autogen/v02/loggers/session_logger.py +580 -0
- ads/llm/autogen/v02/loggers/utils.py +86 -0
- ads/llm/autogen/v02/runtime_logging.py +163 -0
- ads/llm/chain.py +268 -0
- ads/llm/chat_template.py +31 -0
- ads/llm/deploy.py +63 -0
- ads/llm/guardrails/__init__.py +5 -0
- ads/llm/guardrails/base.py +442 -0
- ads/llm/guardrails/huggingface.py +44 -0
- ads/llm/langchain/__init__.py +5 -0
- ads/llm/langchain/plugins/__init__.py +5 -0
- ads/llm/langchain/plugins/chat_models/__init__.py +5 -0
- ads/llm/langchain/plugins/chat_models/oci_data_science.py +1027 -0
- ads/llm/langchain/plugins/embeddings/__init__.py +4 -0
- ads/llm/langchain/plugins/embeddings/oci_data_science_model_deployment_endpoint.py +184 -0
- ads/llm/langchain/plugins/llms/__init__.py +5 -0
- ads/llm/langchain/plugins/llms/oci_data_science_model_deployment_endpoint.py +979 -0
- ads/llm/requirements.txt +3 -0
- ads/llm/serialize.py +219 -0
- ads/llm/serializers/__init__.py +0 -0
- ads/llm/serializers/retrieval_qa.py +153 -0
- ads/llm/serializers/runnable_parallel.py +27 -0
- ads/llm/templates/score_chain.jinja2 +155 -0
- ads/llm/templates/tool_chat_template_hermes.jinja +130 -0
- ads/llm/templates/tool_chat_template_mistral_parallel.jinja +94 -0
- ads/model/__init__.py +52 -0
- ads/model/artifact.py +573 -0
- ads/model/artifact_downloader.py +254 -0
- ads/model/artifact_uploader.py +267 -0
- ads/model/base_properties.py +238 -0
- ads/model/common/.model-ignore +66 -0
- ads/model/common/__init__.py +5 -0
- ads/model/common/utils.py +142 -0
- ads/model/datascience_model.py +2635 -0
- ads/model/deployment/__init__.py +20 -0
- ads/model/deployment/common/__init__.py +5 -0
- ads/model/deployment/common/utils.py +308 -0
- ads/model/deployment/model_deployer.py +466 -0
- ads/model/deployment/model_deployment.py +1846 -0
- ads/model/deployment/model_deployment_infrastructure.py +671 -0
- ads/model/deployment/model_deployment_properties.py +493 -0
- ads/model/deployment/model_deployment_runtime.py +838 -0
- ads/model/extractor/__init__.py +5 -0
- ads/model/extractor/automl_extractor.py +74 -0
- ads/model/extractor/embedding_onnx_extractor.py +80 -0
- ads/model/extractor/huggingface_extractor.py +88 -0
- ads/model/extractor/keras_extractor.py +84 -0
- ads/model/extractor/lightgbm_extractor.py +93 -0
- ads/model/extractor/model_info_extractor.py +114 -0
- ads/model/extractor/model_info_extractor_factory.py +105 -0
- ads/model/extractor/pytorch_extractor.py +87 -0
- ads/model/extractor/sklearn_extractor.py +112 -0
- ads/model/extractor/spark_extractor.py +89 -0
- ads/model/extractor/tensorflow_extractor.py +85 -0
- ads/model/extractor/xgboost_extractor.py +94 -0
- ads/model/framework/__init__.py +5 -0
- ads/model/framework/automl_model.py +178 -0
- ads/model/framework/embedding_onnx_model.py +438 -0
- ads/model/framework/huggingface_model.py +399 -0
- ads/model/framework/lightgbm_model.py +266 -0
- ads/model/framework/pytorch_model.py +266 -0
- ads/model/framework/sklearn_model.py +250 -0
- ads/model/framework/spark_model.py +326 -0
- ads/model/framework/tensorflow_model.py +254 -0
- ads/model/framework/xgboost_model.py +258 -0
- ads/model/generic_model.py +3518 -0
- ads/model/model_artifact_boilerplate/README.md +381 -0
- ads/model/model_artifact_boilerplate/__init__.py +5 -0
- ads/model/model_artifact_boilerplate/artifact_introspection_test/__init__.py +5 -0
- ads/model/model_artifact_boilerplate/artifact_introspection_test/model_artifact_validate.py +427 -0
- ads/model/model_artifact_boilerplate/artifact_introspection_test/requirements.txt +2 -0
- ads/model/model_artifact_boilerplate/runtime.yaml +7 -0
- ads/model/model_artifact_boilerplate/score.py +61 -0
- ads/model/model_file_description_schema.json +68 -0
- ads/model/model_introspect.py +331 -0
- ads/model/model_metadata.py +1810 -0
- ads/model/model_metadata_mixin.py +460 -0
- ads/model/model_properties.py +63 -0
- ads/model/model_version_set.py +739 -0
- ads/model/runtime/__init__.py +5 -0
- ads/model/runtime/env_info.py +306 -0
- ads/model/runtime/model_deployment_details.py +37 -0
- ads/model/runtime/model_provenance_details.py +58 -0
- ads/model/runtime/runtime_info.py +81 -0
- ads/model/runtime/schemas/inference_env_info_schema.yaml +16 -0
- ads/model/runtime/schemas/model_provenance_schema.yaml +36 -0
- ads/model/runtime/schemas/training_env_info_schema.yaml +16 -0
- ads/model/runtime/utils.py +201 -0
- ads/model/serde/__init__.py +5 -0
- ads/model/serde/common.py +40 -0
- ads/model/serde/model_input.py +547 -0
- ads/model/serde/model_serializer.py +1184 -0
- ads/model/service/__init__.py +5 -0
- ads/model/service/oci_datascience_model.py +1076 -0
- ads/model/service/oci_datascience_model_deployment.py +500 -0
- ads/model/service/oci_datascience_model_version_set.py +176 -0
- ads/model/transformer/__init__.py +5 -0
- ads/model/transformer/onnx_transformer.py +324 -0
- ads/mysqldb/__init__.py +5 -0
- ads/mysqldb/mysql_db.py +227 -0
- ads/opctl/__init__.py +18 -0
- ads/opctl/anomaly_detection.py +11 -0
- ads/opctl/backend/__init__.py +5 -0
- ads/opctl/backend/ads_dataflow.py +353 -0
- ads/opctl/backend/ads_ml_job.py +710 -0
- ads/opctl/backend/ads_ml_pipeline.py +164 -0
- ads/opctl/backend/ads_model_deployment.py +209 -0
- ads/opctl/backend/base.py +146 -0
- ads/opctl/backend/local.py +1053 -0
- ads/opctl/backend/marketplace/__init__.py +9 -0
- ads/opctl/backend/marketplace/helm_helper.py +173 -0
- ads/opctl/backend/marketplace/local_marketplace.py +271 -0
- ads/opctl/backend/marketplace/marketplace_backend_runner.py +71 -0
- ads/opctl/backend/marketplace/marketplace_operator_interface.py +44 -0
- ads/opctl/backend/marketplace/marketplace_operator_runner.py +24 -0
- ads/opctl/backend/marketplace/marketplace_utils.py +212 -0
- ads/opctl/backend/marketplace/models/__init__.py +5 -0
- ads/opctl/backend/marketplace/models/bearer_token.py +94 -0
- ads/opctl/backend/marketplace/models/marketplace_type.py +70 -0
- ads/opctl/backend/marketplace/models/ocir_details.py +56 -0
- ads/opctl/backend/marketplace/prerequisite_checker.py +238 -0
- ads/opctl/cli.py +707 -0
- ads/opctl/cmds.py +869 -0
- ads/opctl/conda/__init__.py +5 -0
- ads/opctl/conda/cli.py +193 -0
- ads/opctl/conda/cmds.py +749 -0
- ads/opctl/conda/config.yaml +34 -0
- ads/opctl/conda/manifest_template.yaml +13 -0
- ads/opctl/conda/multipart_uploader.py +188 -0
- ads/opctl/conda/pack.py +89 -0
- ads/opctl/config/__init__.py +5 -0
- ads/opctl/config/base.py +57 -0
- ads/opctl/config/diagnostics/__init__.py +5 -0
- ads/opctl/config/diagnostics/distributed/default_requirements_config.yaml +62 -0
- ads/opctl/config/merger.py +255 -0
- ads/opctl/config/resolver.py +297 -0
- ads/opctl/config/utils.py +79 -0
- ads/opctl/config/validator.py +17 -0
- ads/opctl/config/versioner.py +68 -0
- ads/opctl/config/yaml_parsers/__init__.py +7 -0
- ads/opctl/config/yaml_parsers/base.py +58 -0
- ads/opctl/config/yaml_parsers/distributed/__init__.py +7 -0
- ads/opctl/config/yaml_parsers/distributed/yaml_parser.py +201 -0
- ads/opctl/constants.py +66 -0
- ads/opctl/decorator/__init__.py +5 -0
- ads/opctl/decorator/common.py +129 -0
- ads/opctl/diagnostics/__init__.py +5 -0
- ads/opctl/diagnostics/__main__.py +25 -0
- ads/opctl/diagnostics/check_distributed_job_requirements.py +212 -0
- ads/opctl/diagnostics/check_requirements.py +144 -0
- ads/opctl/diagnostics/requirement_exception.py +9 -0
- ads/opctl/distributed/README.md +109 -0
- ads/opctl/distributed/__init__.py +5 -0
- ads/opctl/distributed/certificates.py +32 -0
- ads/opctl/distributed/cli.py +207 -0
- ads/opctl/distributed/cmds.py +731 -0
- ads/opctl/distributed/common/__init__.py +5 -0
- ads/opctl/distributed/common/abstract_cluster_provider.py +449 -0
- ads/opctl/distributed/common/abstract_framework_spec_builder.py +88 -0
- ads/opctl/distributed/common/cluster_config_helper.py +103 -0
- ads/opctl/distributed/common/cluster_provider_factory.py +21 -0
- ads/opctl/distributed/common/cluster_runner.py +54 -0
- ads/opctl/distributed/common/framework_factory.py +29 -0
- ads/opctl/docker/Dockerfile.job +103 -0
- ads/opctl/docker/Dockerfile.job.arm +107 -0
- ads/opctl/docker/Dockerfile.job.gpu +175 -0
- ads/opctl/docker/base-env.yaml +13 -0
- ads/opctl/docker/cuda.repo +6 -0
- ads/opctl/docker/operator/.dockerignore +0 -0
- ads/opctl/docker/operator/Dockerfile +41 -0
- ads/opctl/docker/operator/Dockerfile.gpu +85 -0
- ads/opctl/docker/operator/cuda.repo +6 -0
- ads/opctl/docker/operator/environment.yaml +8 -0
- ads/opctl/forecast.py +11 -0
- ads/opctl/index.yaml +3 -0
- ads/opctl/model/__init__.py +5 -0
- ads/opctl/model/cli.py +65 -0
- ads/opctl/model/cmds.py +73 -0
- ads/opctl/operator/README.md +4 -0
- ads/opctl/operator/__init__.py +31 -0
- ads/opctl/operator/cli.py +344 -0
- ads/opctl/operator/cmd.py +596 -0
- ads/opctl/operator/common/__init__.py +5 -0
- ads/opctl/operator/common/backend_factory.py +460 -0
- ads/opctl/operator/common/const.py +27 -0
- ads/opctl/operator/common/data/synthetic.csv +16001 -0
- ads/opctl/operator/common/dictionary_merger.py +148 -0
- ads/opctl/operator/common/errors.py +42 -0
- ads/opctl/operator/common/operator_config.py +99 -0
- ads/opctl/operator/common/operator_loader.py +811 -0
- ads/opctl/operator/common/operator_schema.yaml +130 -0
- ads/opctl/operator/common/operator_yaml_generator.py +152 -0
- ads/opctl/operator/common/utils.py +208 -0
- ads/opctl/operator/lowcode/__init__.py +5 -0
- ads/opctl/operator/lowcode/anomaly/MLoperator +16 -0
- ads/opctl/operator/lowcode/anomaly/README.md +207 -0
- ads/opctl/operator/lowcode/anomaly/__init__.py +5 -0
- ads/opctl/operator/lowcode/anomaly/__main__.py +103 -0
- ads/opctl/operator/lowcode/anomaly/cmd.py +35 -0
- ads/opctl/operator/lowcode/anomaly/const.py +167 -0
- ads/opctl/operator/lowcode/anomaly/environment.yaml +10 -0
- ads/opctl/operator/lowcode/anomaly/model/__init__.py +5 -0
- ads/opctl/operator/lowcode/anomaly/model/anomaly_dataset.py +146 -0
- ads/opctl/operator/lowcode/anomaly/model/anomaly_merlion.py +162 -0
- ads/opctl/operator/lowcode/anomaly/model/automlx.py +99 -0
- ads/opctl/operator/lowcode/anomaly/model/autots.py +115 -0
- ads/opctl/operator/lowcode/anomaly/model/base_model.py +404 -0
- ads/opctl/operator/lowcode/anomaly/model/factory.py +110 -0
- ads/opctl/operator/lowcode/anomaly/model/isolationforest.py +78 -0
- ads/opctl/operator/lowcode/anomaly/model/oneclasssvm.py +78 -0
- ads/opctl/operator/lowcode/anomaly/model/randomcutforest.py +120 -0
- ads/opctl/operator/lowcode/anomaly/model/tods.py +119 -0
- ads/opctl/operator/lowcode/anomaly/operator_config.py +127 -0
- ads/opctl/operator/lowcode/anomaly/schema.yaml +401 -0
- ads/opctl/operator/lowcode/anomaly/utils.py +88 -0
- ads/opctl/operator/lowcode/common/__init__.py +5 -0
- ads/opctl/operator/lowcode/common/const.py +10 -0
- ads/opctl/operator/lowcode/common/data.py +116 -0
- ads/opctl/operator/lowcode/common/errors.py +47 -0
- ads/opctl/operator/lowcode/common/transformations.py +296 -0
- ads/opctl/operator/lowcode/common/utils.py +384 -0
- ads/opctl/operator/lowcode/feature_store_marketplace/MLoperator +13 -0
- ads/opctl/operator/lowcode/feature_store_marketplace/README.md +30 -0
- ads/opctl/operator/lowcode/feature_store_marketplace/__init__.py +5 -0
- ads/opctl/operator/lowcode/feature_store_marketplace/__main__.py +116 -0
- ads/opctl/operator/lowcode/feature_store_marketplace/cmd.py +85 -0
- ads/opctl/operator/lowcode/feature_store_marketplace/const.py +15 -0
- ads/opctl/operator/lowcode/feature_store_marketplace/environment.yaml +0 -0
- ads/opctl/operator/lowcode/feature_store_marketplace/models/__init__.py +4 -0
- ads/opctl/operator/lowcode/feature_store_marketplace/models/apigw_config.py +32 -0
- ads/opctl/operator/lowcode/feature_store_marketplace/models/db_config.py +43 -0
- ads/opctl/operator/lowcode/feature_store_marketplace/models/mysql_config.py +120 -0
- ads/opctl/operator/lowcode/feature_store_marketplace/models/serializable_yaml_model.py +34 -0
- ads/opctl/operator/lowcode/feature_store_marketplace/operator_utils.py +386 -0
- ads/opctl/operator/lowcode/feature_store_marketplace/schema.yaml +160 -0
- ads/opctl/operator/lowcode/forecast/MLoperator +25 -0
- ads/opctl/operator/lowcode/forecast/README.md +209 -0
- ads/opctl/operator/lowcode/forecast/__init__.py +5 -0
- ads/opctl/operator/lowcode/forecast/__main__.py +89 -0
- ads/opctl/operator/lowcode/forecast/cmd.py +40 -0
- ads/opctl/operator/lowcode/forecast/const.py +92 -0
- ads/opctl/operator/lowcode/forecast/environment.yaml +20 -0
- ads/opctl/operator/lowcode/forecast/errors.py +26 -0
- ads/opctl/operator/lowcode/forecast/model/__init__.py +5 -0
- ads/opctl/operator/lowcode/forecast/model/arima.py +279 -0
- ads/opctl/operator/lowcode/forecast/model/automlx.py +553 -0
- ads/opctl/operator/lowcode/forecast/model/autots.py +312 -0
- ads/opctl/operator/lowcode/forecast/model/base_model.py +875 -0
- ads/opctl/operator/lowcode/forecast/model/factory.py +106 -0
- ads/opctl/operator/lowcode/forecast/model/forecast_datasets.py +492 -0
- ads/opctl/operator/lowcode/forecast/model/ml_forecast.py +243 -0
- ads/opctl/operator/lowcode/forecast/model/neuralprophet.py +482 -0
- ads/opctl/operator/lowcode/forecast/model/prophet.py +450 -0
- ads/opctl/operator/lowcode/forecast/model_evaluator.py +244 -0
- ads/opctl/operator/lowcode/forecast/operator_config.py +234 -0
- ads/opctl/operator/lowcode/forecast/schema.yaml +506 -0
- ads/opctl/operator/lowcode/forecast/utils.py +397 -0
- ads/opctl/operator/lowcode/forecast/whatifserve/__init__.py +7 -0
- ads/opctl/operator/lowcode/forecast/whatifserve/deployment_manager.py +285 -0
- ads/opctl/operator/lowcode/forecast/whatifserve/score.py +246 -0
- ads/opctl/operator/lowcode/pii/MLoperator +17 -0
- ads/opctl/operator/lowcode/pii/README.md +208 -0
- ads/opctl/operator/lowcode/pii/__init__.py +5 -0
- ads/opctl/operator/lowcode/pii/__main__.py +78 -0
- ads/opctl/operator/lowcode/pii/cmd.py +39 -0
- ads/opctl/operator/lowcode/pii/constant.py +84 -0
- ads/opctl/operator/lowcode/pii/environment.yaml +17 -0
- ads/opctl/operator/lowcode/pii/errors.py +27 -0
- ads/opctl/operator/lowcode/pii/model/__init__.py +5 -0
- ads/opctl/operator/lowcode/pii/model/factory.py +82 -0
- ads/opctl/operator/lowcode/pii/model/guardrails.py +167 -0
- ads/opctl/operator/lowcode/pii/model/pii.py +145 -0
- ads/opctl/operator/lowcode/pii/model/processor/__init__.py +34 -0
- ads/opctl/operator/lowcode/pii/model/processor/email_replacer.py +34 -0
- ads/opctl/operator/lowcode/pii/model/processor/mbi_replacer.py +35 -0
- ads/opctl/operator/lowcode/pii/model/processor/name_replacer.py +225 -0
- ads/opctl/operator/lowcode/pii/model/processor/number_replacer.py +73 -0
- ads/opctl/operator/lowcode/pii/model/processor/remover.py +26 -0
- ads/opctl/operator/lowcode/pii/model/report.py +487 -0
- ads/opctl/operator/lowcode/pii/operator_config.py +95 -0
- ads/opctl/operator/lowcode/pii/schema.yaml +108 -0
- ads/opctl/operator/lowcode/pii/utils.py +43 -0
- ads/opctl/operator/lowcode/recommender/MLoperator +16 -0
- ads/opctl/operator/lowcode/recommender/README.md +206 -0
- ads/opctl/operator/lowcode/recommender/__init__.py +5 -0
- ads/opctl/operator/lowcode/recommender/__main__.py +82 -0
- ads/opctl/operator/lowcode/recommender/cmd.py +33 -0
- ads/opctl/operator/lowcode/recommender/constant.py +30 -0
- ads/opctl/operator/lowcode/recommender/environment.yaml +11 -0
- ads/opctl/operator/lowcode/recommender/model/base_model.py +212 -0
- ads/opctl/operator/lowcode/recommender/model/factory.py +56 -0
- ads/opctl/operator/lowcode/recommender/model/recommender_dataset.py +25 -0
- ads/opctl/operator/lowcode/recommender/model/svd.py +106 -0
- ads/opctl/operator/lowcode/recommender/operator_config.py +81 -0
- ads/opctl/operator/lowcode/recommender/schema.yaml +265 -0
- ads/opctl/operator/lowcode/recommender/utils.py +13 -0
- ads/opctl/operator/runtime/__init__.py +5 -0
- ads/opctl/operator/runtime/const.py +17 -0
- ads/opctl/operator/runtime/container_runtime_schema.yaml +50 -0
- ads/opctl/operator/runtime/marketplace_runtime.py +50 -0
- ads/opctl/operator/runtime/python_marketplace_runtime_schema.yaml +21 -0
- ads/opctl/operator/runtime/python_runtime_schema.yaml +21 -0
- ads/opctl/operator/runtime/runtime.py +115 -0
- ads/opctl/schema.yaml.yml +36 -0
- ads/opctl/script.py +40 -0
- ads/opctl/spark/__init__.py +5 -0
- ads/opctl/spark/cli.py +43 -0
- ads/opctl/spark/cmds.py +147 -0
- ads/opctl/templates/diagnostic_report_template.jinja2 +102 -0
- ads/opctl/utils.py +344 -0
- ads/oracledb/__init__.py +5 -0
- ads/oracledb/oracle_db.py +346 -0
- ads/pipeline/__init__.py +39 -0
- ads/pipeline/ads_pipeline.py +2279 -0
- ads/pipeline/ads_pipeline_run.py +772 -0
- ads/pipeline/ads_pipeline_step.py +605 -0
- ads/pipeline/builders/__init__.py +5 -0
- ads/pipeline/builders/infrastructure/__init__.py +5 -0
- ads/pipeline/builders/infrastructure/custom_script.py +32 -0
- ads/pipeline/cli.py +119 -0
- ads/pipeline/extension.py +291 -0
- ads/pipeline/schema/__init__.py +5 -0
- ads/pipeline/schema/cs_step_schema.json +35 -0
- ads/pipeline/schema/ml_step_schema.json +31 -0
- ads/pipeline/schema/pipeline_schema.json +71 -0
- ads/pipeline/visualizer/__init__.py +5 -0
- ads/pipeline/visualizer/base.py +570 -0
- ads/pipeline/visualizer/graph_renderer.py +272 -0
- ads/pipeline/visualizer/text_renderer.py +84 -0
- ads/secrets/__init__.py +11 -0
- ads/secrets/adb.py +386 -0
- ads/secrets/auth_token.py +86 -0
- ads/secrets/big_data_service.py +365 -0
- ads/secrets/mysqldb.py +149 -0
- ads/secrets/oracledb.py +160 -0
- ads/secrets/secrets.py +407 -0
- ads/telemetry/__init__.py +7 -0
- ads/telemetry/base.py +69 -0
- ads/telemetry/client.py +122 -0
- ads/telemetry/telemetry.py +257 -0
- ads/templates/dataflow_pyspark.jinja2 +13 -0
- ads/templates/dataflow_sparksql.jinja2 +22 -0
- ads/templates/func.jinja2 +20 -0
- ads/templates/schemas/openapi.json +1740 -0
- ads/templates/score-pkl.jinja2 +173 -0
- ads/templates/score.jinja2 +322 -0
- ads/templates/score_embedding_onnx.jinja2 +202 -0
- ads/templates/score_generic.jinja2 +165 -0
- ads/templates/score_huggingface_pipeline.jinja2 +217 -0
- ads/templates/score_lightgbm.jinja2 +185 -0
- ads/templates/score_onnx.jinja2 +407 -0
- ads/templates/score_onnx_new.jinja2 +473 -0
- ads/templates/score_oracle_automl.jinja2 +185 -0
- ads/templates/score_pyspark.jinja2 +154 -0
- ads/templates/score_pytorch.jinja2 +219 -0
- ads/templates/score_scikit-learn.jinja2 +184 -0
- ads/templates/score_tensorflow.jinja2 +184 -0
- ads/templates/score_xgboost.jinja2 +178 -0
- ads/text_dataset/__init__.py +5 -0
- ads/text_dataset/backends.py +211 -0
- ads/text_dataset/dataset.py +445 -0
- ads/text_dataset/extractor.py +207 -0
- ads/text_dataset/options.py +53 -0
- ads/text_dataset/udfs.py +22 -0
- ads/text_dataset/utils.py +49 -0
- ads/type_discovery/__init__.py +9 -0
- ads/type_discovery/abstract_detector.py +21 -0
- ads/type_discovery/constant_detector.py +41 -0
- ads/type_discovery/continuous_detector.py +54 -0
- ads/type_discovery/credit_card_detector.py +99 -0
- ads/type_discovery/datetime_detector.py +92 -0
- ads/type_discovery/discrete_detector.py +118 -0
- ads/type_discovery/document_detector.py +146 -0
- ads/type_discovery/ip_detector.py +68 -0
- ads/type_discovery/latlon_detector.py +90 -0
- ads/type_discovery/phone_number_detector.py +63 -0
- ads/type_discovery/type_discovery_driver.py +87 -0
- ads/type_discovery/typed_feature.py +594 -0
- ads/type_discovery/unknown_detector.py +41 -0
- ads/type_discovery/zipcode_detector.py +48 -0
- ads/vault/__init__.py +7 -0
- ads/vault/vault.py +237 -0
- {oracle_ads-2.13.9rc0.dist-info → oracle_ads-2.13.10.dist-info}/METADATA +150 -149
- oracle_ads-2.13.10.dist-info/RECORD +858 -0
- {oracle_ads-2.13.9rc0.dist-info → oracle_ads-2.13.10.dist-info}/WHEEL +1 -2
- {oracle_ads-2.13.9rc0.dist-info → oracle_ads-2.13.10.dist-info}/entry_points.txt +2 -1
- oracle_ads-2.13.9rc0.dist-info/RECORD +0 -9
- oracle_ads-2.13.9rc0.dist-info/top_level.txt +0 -1
- {oracle_ads-2.13.9rc0.dist-info → oracle_ads-2.13.10.dist-info}/licenses/LICENSE.txt +0 -0
@@ -0,0 +1,543 @@
|
|
1
|
+
#!/usr/bin/env python
|
2
|
+
# -*- coding: utf-8; -*-
|
3
|
+
|
4
|
+
# Copyright (c) 2020, 2022 Oracle and/or its affiliates.
|
5
|
+
# Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
|
6
|
+
|
7
|
+
|
8
|
+
from __future__ import print_function, absolute_import, division
|
9
|
+
|
10
|
+
import numpy as np
|
11
|
+
import pandas as pd
|
12
|
+
from sklearn import metrics
|
13
|
+
from sklearn.preprocessing import LabelEncoder
|
14
|
+
from ads.common import logger
|
15
|
+
from ads.common.decorator.runtime_dependency import (
|
16
|
+
runtime_dependency,
|
17
|
+
OptionalDependency,
|
18
|
+
)
|
19
|
+
|
20
|
+
__all__ = ["ModelEvaluator"]
|
21
|
+
|
22
|
+
DEFAULT_BIN_CLASS_METRICS = [
|
23
|
+
"accuracy",
|
24
|
+
"hamming_loss",
|
25
|
+
"precision",
|
26
|
+
"recall",
|
27
|
+
"f1",
|
28
|
+
"auc",
|
29
|
+
]
|
30
|
+
DEFAULT_MULTI_CLASS_METRICS = [
|
31
|
+
"accuracy",
|
32
|
+
"hamming_loss",
|
33
|
+
"precision_weighted",
|
34
|
+
"precision_micro",
|
35
|
+
"recall_weighted",
|
36
|
+
"recall_micro",
|
37
|
+
"f1_weighted",
|
38
|
+
"f1_micro",
|
39
|
+
]
|
40
|
+
DEFAULT_REG_METRICS = ["r2_score", "mse", "mae"]
|
41
|
+
DEFAULT_BIN_CLASS_LABELS_MAP = {
|
42
|
+
"accuracy": "Accuracy",
|
43
|
+
"hamming_loss": "Hamming distance",
|
44
|
+
"kappa_score": "Cohen's kappa coefficient",
|
45
|
+
"precision": "Precision",
|
46
|
+
"recall": "Recall",
|
47
|
+
"f1": "F1",
|
48
|
+
"auc": "ROC AUC",
|
49
|
+
}
|
50
|
+
DEFAULT_MULTI_CLASS_LABELS_MAP = {
|
51
|
+
"accuracy": "Accuracy",
|
52
|
+
"hamming_loss": "Hamming distance",
|
53
|
+
"precision_weighted": "Precision Weighted Average",
|
54
|
+
"precision_micro": "Precision Micro Average",
|
55
|
+
"recall_weighted": "Recall Weighted Average",
|
56
|
+
"recall_micro": "Recall Micro Average",
|
57
|
+
"f1_weighted": "F1 Weighted Average",
|
58
|
+
"f1_micro": "F1 Micro Average",
|
59
|
+
}
|
60
|
+
DEFAULT_REG_LABELS_MAP = {
|
61
|
+
"r2_score": "r-Squared Score",
|
62
|
+
"root_mean_squared_error": "Root Mean Squared Error",
|
63
|
+
"median_absolute_error": "Median Absolute Error",
|
64
|
+
}
|
65
|
+
|
66
|
+
|
67
|
+
class ModelEvaluator:
|
68
|
+
"""
|
69
|
+
ModelEvaluator takes in the true and predicted values and returns a pandas dataframe
|
70
|
+
|
71
|
+
Attributes
|
72
|
+
----------
|
73
|
+
y_true : array-like object holding the true values for the model
|
74
|
+
y_pred : array-like object holding the predicted values for the model
|
75
|
+
model_name (str): the name of the model
|
76
|
+
classes (list): list of target classes
|
77
|
+
positive_class (str): label for positive outcome from model
|
78
|
+
y_score : array-like object holding the scores for true values for the model
|
79
|
+
metrics (dict): dictionary object holding model data
|
80
|
+
|
81
|
+
Methods
|
82
|
+
-------
|
83
|
+
get_metrics()
|
84
|
+
Gets the metrics information in a dataframe based on the number of classes
|
85
|
+
safe_metrics_call(scoring_functions, *args)
|
86
|
+
Applies sklearn scoring functions to parameters in args
|
87
|
+
|
88
|
+
"""
|
89
|
+
|
90
|
+
def __init__(
|
91
|
+
self,
|
92
|
+
y_true,
|
93
|
+
y_pred,
|
94
|
+
model_name,
|
95
|
+
classes=None,
|
96
|
+
positive_class=None,
|
97
|
+
y_score=None,
|
98
|
+
):
|
99
|
+
self.y_true = y_true
|
100
|
+
self.y_pred = np.squeeze(
|
101
|
+
y_pred
|
102
|
+
) # This is a bug where y_pred is shape (1,n) from AutoML, rather than (n)
|
103
|
+
self.y_score = y_score
|
104
|
+
self.model_name = model_name
|
105
|
+
self.classes = classes
|
106
|
+
self.positive_class = positive_class
|
107
|
+
self.metrics = {
|
108
|
+
"model_name": model_name,
|
109
|
+
"classes": classes,
|
110
|
+
"positive_class": positive_class,
|
111
|
+
"y_true": y_true,
|
112
|
+
"y_pred": y_pred,
|
113
|
+
"y_score": y_score,
|
114
|
+
}
|
115
|
+
|
116
|
+
def get_metrics(self):
|
117
|
+
"""
|
118
|
+
Gets the metrics information in a dataframe based on the number of classes
|
119
|
+
|
120
|
+
Parameters
|
121
|
+
----------
|
122
|
+
self: (`ModelEvaluator` instance)
|
123
|
+
The `ModelEvaluator` instance with the metrics.
|
124
|
+
|
125
|
+
Returns
|
126
|
+
-------
|
127
|
+
:class:`pandas.DataFrame`
|
128
|
+
Pandas dataframe containing the metrics
|
129
|
+
"""
|
130
|
+
|
131
|
+
if self.classes is None:
|
132
|
+
self._get_regression_metrics()
|
133
|
+
return pd.DataFrame.from_dict(
|
134
|
+
self.metrics, orient="index", columns=[self.model_name]
|
135
|
+
)
|
136
|
+
else:
|
137
|
+
if len(self.classes) == 2: # 'binary'
|
138
|
+
self.positive_class = (
|
139
|
+
self.positive_class if self.positive_class else self.classes[0]
|
140
|
+
)
|
141
|
+
self._get_binary_metrics()
|
142
|
+
return pd.DataFrame.from_dict(
|
143
|
+
self.metrics, orient="index", columns=[self.model_name]
|
144
|
+
)
|
145
|
+
else: # 'multiclass'
|
146
|
+
self._get_multiclass_metrics()
|
147
|
+
return pd.DataFrame.from_dict(
|
148
|
+
self.metrics, orient="index", columns=[self.model_name]
|
149
|
+
)
|
150
|
+
|
151
|
+
def _get_general_metrics(self):
|
152
|
+
try:
|
153
|
+
args = [self.y_true, self.y_pred]
|
154
|
+
kwargs = {"labels": self.classes}
|
155
|
+
|
156
|
+
scoring_functions = {
|
157
|
+
"classification_report": ("classification_report", len(args)),
|
158
|
+
"kappa_score": ("cohen_kappa_score", len(args)),
|
159
|
+
"raw_confusion_matrix": ("confusion_matrix", len(args)),
|
160
|
+
"hinge_loss": ("hinge_loss", len(args)),
|
161
|
+
}
|
162
|
+
self.safe_metrics_call(scoring_functions, *args, **kwargs)
|
163
|
+
|
164
|
+
args = [self.y_true, self.y_pred]
|
165
|
+
scoring_functions_without_labs = {
|
166
|
+
"accuracy": ("accuracy_score", len(args)),
|
167
|
+
"zero_one_loss": ("zero_one_loss", len(args)),
|
168
|
+
"hamming_loss": ("hamming_loss", len(args)),
|
169
|
+
}
|
170
|
+
self.safe_metrics_call(scoring_functions_without_labs, *args)
|
171
|
+
|
172
|
+
cm = self.metrics["raw_confusion_matrix"]
|
173
|
+
a = cm.astype("float")
|
174
|
+
b = cm.sum(axis=1)[:, np.newaxis]
|
175
|
+
normalized_cm = np.divide(a, b, out=np.zeros_like(a), where=b != 0).tolist()
|
176
|
+
self.metrics["confusion_matrix"] = normalized_cm
|
177
|
+
|
178
|
+
vc = pd.DataFrame(self.y_true).value_counts()
|
179
|
+
self.metrics["class_balance"] = min(vc) / max(vc)
|
180
|
+
except:
|
181
|
+
raise ValueError(
|
182
|
+
f"Errors arose when attempting to compute metrics. Metrics broke in the state: {self.metrics}"
|
183
|
+
)
|
184
|
+
|
185
|
+
def _get_binary_metrics(self):
|
186
|
+
self._get_general_metrics()
|
187
|
+
# add a metric to determine imbalance
|
188
|
+
self.metrics["balanced accuracy"] = metrics.balanced_accuracy_score(
|
189
|
+
self.y_true, self.y_pred
|
190
|
+
)
|
191
|
+
self.metrics["precision"] = metrics.precision_score(
|
192
|
+
self.y_true, self.y_pred, pos_label=self.positive_class
|
193
|
+
)
|
194
|
+
self.metrics["recall"] = metrics.recall_score(
|
195
|
+
self.y_true, self.y_pred, pos_label=self.positive_class
|
196
|
+
)
|
197
|
+
self.metrics["f1"] = metrics.f1_score(
|
198
|
+
self.y_true, self.y_pred, pos_label=self.positive_class, average="binary"
|
199
|
+
)
|
200
|
+
|
201
|
+
if self.y_score is not None:
|
202
|
+
if not all(0 >= x >= 1 for x in self.y_score):
|
203
|
+
self.y_score = np.asarray(
|
204
|
+
[0 if x < 0 else 1 if x > 1 else x for x in self.y_score]
|
205
|
+
)
|
206
|
+
if len(np.asarray(self.y_score).shape) > 1:
|
207
|
+
# If the SKLearn classifier doesn't correctly identify the problem as
|
208
|
+
# binary classification, y_score may be of shape (n_rows, 2)
|
209
|
+
# instead of (n_rows,)
|
210
|
+
pos_class_idx = self.classes.index(self.positive_class)
|
211
|
+
positive_class_scores = self.y_score[:, pos_class_idx]
|
212
|
+
else:
|
213
|
+
positive_class_scores = self.y_score
|
214
|
+
(
|
215
|
+
self.metrics["false_positive_rate"],
|
216
|
+
self.metrics["true_positive_rate"],
|
217
|
+
_,
|
218
|
+
) = metrics.roc_curve(y_true=self.y_true, y_score=positive_class_scores, pos_label=self.positive_class)
|
219
|
+
self.metrics["auc"] = metrics.auc(
|
220
|
+
self.metrics["false_positive_rate"], self.metrics["true_positive_rate"]
|
221
|
+
)
|
222
|
+
self.y_score = list(self.y_score)
|
223
|
+
self.metrics["youden_j"] = (
|
224
|
+
self.metrics["true_positive_rate"] - self.metrics["false_positive_rate"]
|
225
|
+
)
|
226
|
+
best_idx = np.argmax(self.metrics["youden_j"])
|
227
|
+
self.metrics["roc_best_model_score"] = (
|
228
|
+
self.metrics["false_positive_rate"][best_idx],
|
229
|
+
self.metrics["true_positive_rate"][best_idx],
|
230
|
+
)
|
231
|
+
(
|
232
|
+
self.metrics["precision_values"],
|
233
|
+
self.metrics["recall_values"],
|
234
|
+
_,
|
235
|
+
) = metrics.precision_recall_curve(
|
236
|
+
self.y_true, self.y_score, pos_label=self.positive_class
|
237
|
+
)
|
238
|
+
pr_best_idx = np.argmax(
|
239
|
+
self.metrics["precision_values"] + self.metrics["recall_values"]
|
240
|
+
)
|
241
|
+
self.metrics["pr_best_idx"] = pr_best_idx
|
242
|
+
self.metrics["pr_best_model_score"] = (
|
243
|
+
self.metrics["recall_values"][pr_best_idx],
|
244
|
+
self.metrics["precision_values"][pr_best_idx],
|
245
|
+
)
|
246
|
+
self.metrics["average_precision_score"] = metrics.average_precision_score(
|
247
|
+
self.y_true, self.y_score, pos_label=self.positive_class
|
248
|
+
)
|
249
|
+
|
250
|
+
self.metrics["brier score"] = metrics.brier_score_loss(
|
251
|
+
self.y_true, self.y_score
|
252
|
+
)
|
253
|
+
self._get_lift_and_gain()
|
254
|
+
# Compute KS Statistic curves
|
255
|
+
self._binary_ks_curve()
|
256
|
+
|
257
|
+
def _get_lift_and_gain(self):
|
258
|
+
# make y_true a boolean vector
|
259
|
+
y_true, y_score = np.asarray(self.y_true), np.asarray(self.y_score)
|
260
|
+
y_true = y_true == self.positive_class
|
261
|
+
|
262
|
+
sorted_indices = np.argsort(y_score)[::-1]
|
263
|
+
y_true = y_true[sorted_indices]
|
264
|
+
gains = np.cumsum(y_true)
|
265
|
+
|
266
|
+
percentages = np.arange(start=1, stop=len(y_true) + 1)
|
267
|
+
tp = sum(y_true)
|
268
|
+
perfect = np.append(
|
269
|
+
np.arange(start=1, stop=tp + 1) / float(tp), np.ones(len(y_true) - tp)
|
270
|
+
)
|
271
|
+
|
272
|
+
gains = gains / float(np.sum(y_true))
|
273
|
+
percentages = percentages / float(len(y_true))
|
274
|
+
|
275
|
+
self.metrics["cumulative_gain"] = np.insert(gains, 0, [0]) * 100
|
276
|
+
self.metrics["percentages"] = np.insert(percentages, 0, [0]) * 100
|
277
|
+
self.metrics["perfect_gain"] = np.insert(perfect, 0, [0]) * 100
|
278
|
+
|
279
|
+
percentages = percentages[1:]
|
280
|
+
self.metrics["lift"] = gains[1:] / percentages
|
281
|
+
self.metrics["perfect_lift"] = perfect[1:] / percentages
|
282
|
+
|
283
|
+
def _binary_ks_curve(self):
|
284
|
+
"""This function generates the points necessary to calculate the KS
|
285
|
+
Statistic curve.
|
286
|
+
Args:
|
287
|
+
y_true (array-like, shape (n_samples)): True labels of the data.
|
288
|
+
y_probas (array-like, shape (n_samples)): Probability predictions of
|
289
|
+
the positive class.
|
290
|
+
Returns:
|
291
|
+
thresholds (numpy.ndarray): An array containing the X-axis values for
|
292
|
+
plotting the KS Statistic plot.
|
293
|
+
pct1 (numpy.ndarray): An array containing the Y-axis values for one
|
294
|
+
curve of the KS Statistic plot.
|
295
|
+
pct2 (numpy.ndarray): An array containing the Y-axis values for one
|
296
|
+
curve of the KS Statistic plot.
|
297
|
+
ks_statistic (float): The KS Statistic, or the maximum vertical
|
298
|
+
distance between the two curves.
|
299
|
+
max_distance_at (float): The X-axis value at which the maximum vertical
|
300
|
+
distance between the two curves is seen.
|
301
|
+
classes (np.ndarray, shape (2)): An array containing the labels of the
|
302
|
+
two classes making up `y_true`.
|
303
|
+
Raises:
|
304
|
+
ValueError: If `y_true` is not composed of 2 classes. The KS Statistic
|
305
|
+
is only relevant in binary classification.
|
306
|
+
"""
|
307
|
+
y_true, y_probas = np.asarray(self.y_true), np.asarray(self.y_score)
|
308
|
+
lb = LabelEncoder()
|
309
|
+
encoded_labels = lb.fit_transform(y_true)
|
310
|
+
if len(lb.classes_) != 2:
|
311
|
+
raise ValueError(
|
312
|
+
"Cannot calculate KS statistic for data with "
|
313
|
+
"{} category/ies".format(len(lb.classes_))
|
314
|
+
)
|
315
|
+
idx = encoded_labels == 0
|
316
|
+
data1 = np.sort(y_probas[idx])
|
317
|
+
data2 = np.sort(y_probas[np.logical_not(idx)])
|
318
|
+
|
319
|
+
ctr1, ctr2 = 0, 0
|
320
|
+
thresholds, pct1, pct2 = [], [], []
|
321
|
+
while ctr1 < len(data1) or ctr2 < len(data2):
|
322
|
+
|
323
|
+
# Check if data1 has no more elements
|
324
|
+
if ctr1 >= len(data1):
|
325
|
+
current = data2[ctr2]
|
326
|
+
while ctr2 < len(data2) and current == data2[ctr2]:
|
327
|
+
ctr2 += 1
|
328
|
+
|
329
|
+
# Check if data2 has no more elements
|
330
|
+
elif ctr2 >= len(data2):
|
331
|
+
current = data1[ctr1]
|
332
|
+
while ctr1 < len(data1) and current == data1[ctr1]:
|
333
|
+
ctr1 += 1
|
334
|
+
|
335
|
+
else:
|
336
|
+
if data1[ctr1] > data2[ctr2]:
|
337
|
+
current = data2[ctr2]
|
338
|
+
while ctr2 < len(data2) and current == data2[ctr2]:
|
339
|
+
ctr2 += 1
|
340
|
+
|
341
|
+
elif data1[ctr1] < data2[ctr2]:
|
342
|
+
current = data1[ctr1]
|
343
|
+
while ctr1 < len(data1) and current == data1[ctr1]:
|
344
|
+
ctr1 += 1
|
345
|
+
|
346
|
+
else:
|
347
|
+
current = data2[ctr2]
|
348
|
+
while ctr2 < len(data2) and current == data2[ctr2]:
|
349
|
+
ctr2 += 1
|
350
|
+
while ctr1 < len(data1) and current == data1[ctr1]:
|
351
|
+
ctr1 += 1
|
352
|
+
|
353
|
+
thresholds.append(current)
|
354
|
+
pct1.append(ctr1)
|
355
|
+
pct2.append(ctr2)
|
356
|
+
|
357
|
+
thresholds = np.asarray(thresholds)
|
358
|
+
pct1 = np.asarray(pct1) / float(len(data1))
|
359
|
+
pct2 = np.asarray(pct2) / float(len(data2))
|
360
|
+
|
361
|
+
if thresholds[0] != 0:
|
362
|
+
thresholds = np.insert(thresholds, 0, [0.0])
|
363
|
+
pct1 = np.insert(pct1, 0, [0.0])
|
364
|
+
pct2 = np.insert(pct2, 0, [0.0])
|
365
|
+
if thresholds[-1] != 1:
|
366
|
+
thresholds = np.append(thresholds, [1.0])
|
367
|
+
pct1 = np.append(pct1, [1.0])
|
368
|
+
pct2 = np.append(pct2, [1.0])
|
369
|
+
|
370
|
+
differences = pct1 - pct2
|
371
|
+
self.metrics["ks_statistic"], self.metrics["max_distance_at"] = (
|
372
|
+
np.max(differences),
|
373
|
+
thresholds[np.argmax(differences)],
|
374
|
+
)
|
375
|
+
|
376
|
+
(
|
377
|
+
self.metrics["ks_thresholds"],
|
378
|
+
self.metrics["ks_pct1"],
|
379
|
+
self.metrics["ks_pct2"],
|
380
|
+
self.metrics["ks_labels"],
|
381
|
+
) = (thresholds, pct1, pct2, lb.classes_)
|
382
|
+
|
383
|
+
def _get_multiclass_metrics(self):
|
384
|
+
self._get_general_metrics()
|
385
|
+
y_true, y_pred, y_score = (
|
386
|
+
np.asarray(self.y_true),
|
387
|
+
np.asarray(self.y_pred),
|
388
|
+
np.asarray(self.y_score),
|
389
|
+
)
|
390
|
+
if not all([x in self.classes for x in np.unique(y_true)]):
|
391
|
+
logger.warning(
|
392
|
+
f"There are classes in the test dataset that are not specified in the `classes` "
|
393
|
+
f"attribute of model ({self.model_name}). This may lead to erroneous "
|
394
|
+
f"results."
|
395
|
+
)
|
396
|
+
|
397
|
+
self.metrics["precision_weighted"] = metrics.precision_score(
|
398
|
+
y_true, y_pred, labels=self.classes, average="weighted"
|
399
|
+
)
|
400
|
+
self.metrics["precision_micro"] = metrics.precision_score(
|
401
|
+
y_true, y_pred, labels=self.classes, average="micro"
|
402
|
+
)
|
403
|
+
self.metrics["precision_by_label"] = metrics.precision_score(
|
404
|
+
y_true, y_pred, labels=self.classes, average=None
|
405
|
+
).tolist()
|
406
|
+
self.metrics["recall_weighted"] = metrics.recall_score(
|
407
|
+
y_true, y_pred, labels=self.classes, average="weighted"
|
408
|
+
)
|
409
|
+
self.metrics["recall_micro"] = metrics.recall_score(
|
410
|
+
y_true, y_pred, labels=self.classes, average="micro"
|
411
|
+
)
|
412
|
+
self.metrics["recall_by_label"] = metrics.recall_score(
|
413
|
+
y_true, y_pred, labels=self.classes, average=None
|
414
|
+
).tolist()
|
415
|
+
self.metrics["f1_weighted"] = metrics.f1_score(
|
416
|
+
y_true, y_pred, labels=self.classes, average="weighted"
|
417
|
+
)
|
418
|
+
self.metrics["f1_micro"] = metrics.f1_score(
|
419
|
+
y_true, y_pred, labels=self.classes, average="micro"
|
420
|
+
)
|
421
|
+
self.metrics["f1_by_label"] = metrics.f1_score(
|
422
|
+
y_true, y_pred, labels=self.classes, average=None
|
423
|
+
).tolist()
|
424
|
+
self.metrics["jaccard_weighted"] = metrics.jaccard_score(
|
425
|
+
y_true, y_pred, labels=self.classes, average="weighted"
|
426
|
+
)
|
427
|
+
self.metrics["jaccard_micro"] = metrics.jaccard_score(
|
428
|
+
y_true, y_pred, labels=self.classes, average="micro"
|
429
|
+
)
|
430
|
+
self.metrics["jaccard_by_label"] = metrics.jaccard_score(
|
431
|
+
y_true, y_pred, labels=self.classes, average=None
|
432
|
+
).tolist()
|
433
|
+
|
434
|
+
if self.y_score is not None:
|
435
|
+
# Multiclass ROC
|
436
|
+
(
|
437
|
+
self.metrics["fpr_by_label"],
|
438
|
+
self.metrics["tpr_by_label"],
|
439
|
+
self.metrics["auc"],
|
440
|
+
self.metrics["roc_best_model_score"],
|
441
|
+
) = (dict(), dict(), dict(), dict())
|
442
|
+
for i, label in enumerate(self.classes):
|
443
|
+
(
|
444
|
+
self.metrics["fpr_by_label"][i],
|
445
|
+
self.metrics["tpr_by_label"][i],
|
446
|
+
_,
|
447
|
+
) = metrics.roc_curve(y_true, y_score[:, i], pos_label=self.classes[i])
|
448
|
+
self.metrics["auc"][i] = metrics.auc(
|
449
|
+
self.metrics["fpr_by_label"][i], self.metrics["tpr_by_label"][i]
|
450
|
+
)
|
451
|
+
youden_j = (
|
452
|
+
self.metrics["tpr_by_label"][i] - self.metrics["fpr_by_label"][i]
|
453
|
+
)
|
454
|
+
best_idx = np.argmax(youden_j)
|
455
|
+
self.metrics["roc_best_model_score"][i] = (
|
456
|
+
self.metrics["fpr_by_label"][i][best_idx],
|
457
|
+
self.metrics["tpr_by_label"][i][best_idx],
|
458
|
+
)
|
459
|
+
# Multiclass PR
|
460
|
+
(
|
461
|
+
self.metrics["recall_values"],
|
462
|
+
self.metrics["precision_values"],
|
463
|
+
self.metrics["pr_best_model_score"],
|
464
|
+
) = (dict(), dict(), dict())
|
465
|
+
for i, label in enumerate(self.classes):
|
466
|
+
(
|
467
|
+
self.metrics["precision_values"][i],
|
468
|
+
self.metrics["recall_values"][i],
|
469
|
+
_,
|
470
|
+
) = metrics.precision_recall_curve(
|
471
|
+
y_true, y_score[:, i], pos_label=self.classes[i]
|
472
|
+
)
|
473
|
+
pr_best_idx = np.argmax(
|
474
|
+
self.metrics["precision_values"][i]
|
475
|
+
+ self.metrics["recall_values"][i]
|
476
|
+
)
|
477
|
+
self.metrics["pr_best_model_score"][i] = (
|
478
|
+
self.metrics["recall_values"][i][pr_best_idx],
|
479
|
+
self.metrics["precision_values"][i][pr_best_idx],
|
480
|
+
)
|
481
|
+
|
482
|
+
@runtime_dependency(module="scipy", install_from=OptionalDependency.VIZ)
|
483
|
+
def _get_regression_metrics(self):
|
484
|
+
self.y_true = np.array(self.y_true)
|
485
|
+
self.y_pred = np.array(np.squeeze(self.y_pred))
|
486
|
+
args = [self.y_true, self.y_pred]
|
487
|
+
scoring_functions = {
|
488
|
+
"r2_score": ("r2_score", len(args)),
|
489
|
+
"explained_variance_score": ("explained_variance_score", len(args)),
|
490
|
+
"max_error": ("max_error", len(args)),
|
491
|
+
"mae": ("mean_absolute_error", len(args)),
|
492
|
+
"mse": ("mean_squared_error", len(args)),
|
493
|
+
"median_absolute_error": ("median_absolute_error", len(args)),
|
494
|
+
}
|
495
|
+
|
496
|
+
self.safe_metrics_call(scoring_functions, *args)
|
497
|
+
|
498
|
+
self.metrics["root_mean_squared_error"] = np.sqrt(self.metrics["mse"])
|
499
|
+
self.metrics["residuals"] = self.y_true - self.y_pred
|
500
|
+
self.metrics["mean_residuals"] = np.mean(self.metrics["residuals"])
|
501
|
+
|
502
|
+
# For QQ Plot:
|
503
|
+
portions = min(len(self.metrics["residuals"]), 100) + 1
|
504
|
+
norm_quantiles = (np.arange(portions) / portions)[1:]
|
505
|
+
self.metrics["norm_quantiles"] = scipy.stats.norm.ppf(norm_quantiles)
|
506
|
+
resid_quant = [
|
507
|
+
np.quantile(self.metrics["residuals"], p) for p in norm_quantiles
|
508
|
+
]
|
509
|
+
self.metrics["residual_quantiles"] = scipy.stats.zscore(resid_quant)
|
510
|
+
|
511
|
+
def safe_metrics_call(self, scoring_functions, *args, **kwargs):
|
512
|
+
"""Applies the sklearn function in `scoring_functions` to parameters in `args`.
|
513
|
+
|
514
|
+
Parameters
|
515
|
+
----------
|
516
|
+
scoring_functions: (dict)
|
517
|
+
Scoring functions dictionary
|
518
|
+
args: (keyword arguments)
|
519
|
+
Arguments passed to the sklearn function from metrics
|
520
|
+
|
521
|
+
Returns:
|
522
|
+
Nothing
|
523
|
+
|
524
|
+
Raises:
|
525
|
+
Exception: If an error is enountered applying the sklearn function fn to arguments.
|
526
|
+
"""
|
527
|
+
|
528
|
+
for name, (fn, n_params) in scoring_functions.items():
|
529
|
+
try:
|
530
|
+
if fn == "confusion_matrix":
|
531
|
+
self.metrics[name] = getattr(metrics, fn)(
|
532
|
+
**{
|
533
|
+
"y_true": args[0],
|
534
|
+
"y_pred": args[1],
|
535
|
+
"labels": kwargs["labels"],
|
536
|
+
}
|
537
|
+
)
|
538
|
+
else:
|
539
|
+
self.metrics[name] = getattr(metrics, fn)(
|
540
|
+
*(args[:n_params]), **kwargs
|
541
|
+
)
|
542
|
+
except Exception as e:
|
543
|
+
self.metrics[name] = f"Error unable to compute {fn}, due to: {e}"
|
@@ -0,0 +1,9 @@
|
|
1
|
+
#!/usr/bin/env python
|
2
|
+
# -*- coding: utf-8 -*--
|
3
|
+
|
4
|
+
# Copyright (c) 2020, 2022 Oracle and/or its affiliates.
|
5
|
+
# Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
|
6
|
+
|
7
|
+
import logging
|
8
|
+
|
9
|
+
logger = logging.getLogger(__name__)
|
File without changes
|
@@ -0,0 +1,21 @@
|
|
1
|
+
#!/usr/bin/env python
|
2
|
+
# -*- coding: utf-8 -*--
|
3
|
+
|
4
|
+
# Copyright (c) 2020, 2023 Oracle and/or its affiliates.
|
5
|
+
# Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
|
6
|
+
|
7
|
+
import logging
|
8
|
+
|
9
|
+
|
10
|
+
logger = logging.getLogger(__name__)
|
11
|
+
|
12
|
+
from .mlx_global_explainer import *
|
13
|
+
from .mlx_local_explainer import *
|
14
|
+
from .mlx_whatif_explainer import *
|
15
|
+
|
16
|
+
try:
|
17
|
+
import mlx
|
18
|
+
|
19
|
+
mlx.initjs()
|
20
|
+
except:
|
21
|
+
pass
|