File tree 1 file changed +8
-1
lines changed
ads/opctl/operator/lowcode/forecast/model
1 file changed +8
-1
lines changed Original file line number Diff line number Diff line change 3
3
# Copyright (c) 2023, 2024 Oracle and/or its affiliates.
4
4
# Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
5
5
6
- from ..const import AUTO_SELECT , SupportedModels
6
+ from ..const import AUTO_SELECT , SpeedAccuracyMode , SupportedModels
7
7
from ..model_evaluator import ModelEvaluator
8
8
from ..operator_config import ForecastOperatorConfig
9
9
from .arima import ArimaOperatorModel
@@ -66,6 +66,13 @@ def get_model(
66
66
if model_type == AUTO_SELECT :
67
67
model_type = cls .auto_select_model (datasets , operator_config )
68
68
operator_config .spec .model_kwargs = {}
69
+ # set the explanations accuracy mode to AUTOMLX if the selected model is automlx
70
+ if (
71
+ model_type == SupportedModels .AutoMLX
72
+ and operator_config .spec .explanations_accuracy_mode
73
+ == SpeedAccuracyMode .FAST_APPROXIMATE
74
+ ):
75
+ operator_config .spec .explanations_accuracy_mode = SpeedAccuracyMode .AUTOMLX
69
76
if model_type not in cls ._MAP :
70
77
raise UnSupportedModelError (model_type )
71
78
return cls ._MAP [model_type ](config = operator_config , datasets = datasets )
You can’t perform that action at this time.
0 commit comments