From 04e59396199d3fde4af20a88ad44e74d387f86b5 Mon Sep 17 00:00:00 2001 From: Vikas Pandey Date: Mon, 24 Mar 2025 18:09:55 +0000 Subject: [PATCH 1/2] auto-select fallback to automlx for fast approximate mode --- ads/opctl/operator/lowcode/forecast/model/factory.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/ads/opctl/operator/lowcode/forecast/model/factory.py b/ads/opctl/operator/lowcode/forecast/model/factory.py index 446709a0d..1ffb25ece 100644 --- a/ads/opctl/operator/lowcode/forecast/model/factory.py +++ b/ads/opctl/operator/lowcode/forecast/model/factory.py @@ -3,7 +3,7 @@ # Copyright (c) 2023, 2024 Oracle and/or its affiliates. # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/ -from ..const import AUTO_SELECT, SupportedModels +from ..const import AUTO_SELECT, SpeedAccuracyMode, SupportedModels from ..model_evaluator import ModelEvaluator from ..operator_config import ForecastOperatorConfig from .arima import ArimaOperatorModel @@ -66,6 +66,13 @@ def get_model( if model_type == AUTO_SELECT: model_type = cls.auto_select_model(datasets, operator_config) operator_config.spec.model_kwargs = {} + # set the explanations accuracy mode to AUTOMLX if the selected model is automlx + if ( + model_type == SupportedModels.AutoMLX + and operator_config.spec.explanations_accuracy_mode + == SpeedAccuracyMode.FAST_APPROXIMATE + ): + operator_config.spec.explanations_accuracy_mode = "AUTOMLX" if model_type not in cls._MAP: raise UnSupportedModelError(model_type) return cls._MAP[model_type](config=operator_config, datasets=datasets) From d1bda3e3035adcbe62a77f21084e19c827e4bdca Mon Sep 17 00:00:00 2001 From: Vikas Pandey Date: Mon, 24 Mar 2025 18:36:05 +0000 Subject: [PATCH 2/2] update --- ads/opctl/operator/lowcode/forecast/model/factory.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ads/opctl/operator/lowcode/forecast/model/factory.py b/ads/opctl/operator/lowcode/forecast/model/factory.py index 1ffb25ece..09583c58b 100644 --- a/ads/opctl/operator/lowcode/forecast/model/factory.py +++ b/ads/opctl/operator/lowcode/forecast/model/factory.py @@ -72,7 +72,7 @@ def get_model( and operator_config.spec.explanations_accuracy_mode == SpeedAccuracyMode.FAST_APPROXIMATE ): - operator_config.spec.explanations_accuracy_mode = "AUTOMLX" + operator_config.spec.explanations_accuracy_mode = SpeedAccuracyMode.AUTOMLX if model_type not in cls._MAP: raise UnSupportedModelError(model_type) return cls._MAP[model_type](config=operator_config, datasets=datasets)