add the default_trials to constants:

oracle · Sep 11, 2023 · 9f829a3 · 9f829a3
1 parent 8813e40
commit 9f829a3
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/ads/opctl/operator/lowcode/forecast/const.py b/ads/opctl/operator/lowcode/forecast/const.py
@@ -35,3 +35,4 @@ class SupportedMetrics(str, metaclass=ExtendedEnumMeta):
 }
 
 MAX_COLUMNS_AUTOMLX = 15
+DEFAULT_TRIALS = 10
diff --git a/ads/opctl/operator/lowcode/forecast/model/neuralprophet.py b/ads/opctl/operator/lowcode/forecast/model/neuralprophet.py
@@ -25,6 +25,7 @@
 )
 from ads.opctl import logger
 
+from ...forecast.const import DEFAULT_TRIALS
 from .. import utils
 from .base_model import ForecastOperatorBaseModel
 
@@ -167,7 +168,7 @@ def objective(trial):
                 )
                 study.optimize(
                     objective,
-                    n_trials=self.spec.tuning.n_trials if self.spec.tuning else 10,
+                    n_trials=self.spec.tuning.n_trials if self.spec.tuning else DEFAULT_TRIALS,
                     n_jobs=-1,
                 )
 

diff --git a/ads/opctl/operator/lowcode/forecast/model/prophet.py b/ads/opctl/operator/lowcode/forecast/model/prophet.py
@@ -15,6 +15,7 @@
 
 from ads.opctl import logger
 
+from ...forecast.const import DEFAULT_TRIALS
 from .. import utils
 from .base_model import ForecastOperatorBaseModel
 
@@ -144,7 +145,7 @@ def objective(trial):
                 )
                 study.optimize(
                     objective,
-                    n_trials=self.spec.tuning.n_trials if self.spec.tuning else 10,
+                    n_trials=self.spec.tuning.n_trials if self.spec.tuning else DEFAULT_TRIALS,
                     n_jobs=-1,
                 )
Original file line number	Diff line number	Diff line change
Expand Up		@@ -35,3 +35,4 @@ class SupportedMetrics(str, metaclass=ExtendedEnumMeta):
		}

		MAX_COLUMNS_AUTOMLX = 15
		DEFAULT_TRIALS = 10