File tree Expand file tree Collapse file tree 5 files changed +395
-0
lines changed
opctl/operator/lowcode/forecast Expand file tree Collapse file tree 5 files changed +395
-0
lines changed Original file line number Diff line number Diff line change @@ -165,6 +165,7 @@ class Framework(ExtendedEnum):
165165 PYOD = "pyod"
166166 SPACY = "spacy"
167167 PROPHET = "prophet"
168+ THETA = "theta"
168169 SKTIME = "sktime"
169170 STATSMODELS = "statsmodels"
170171 CUML = "cuml"
Original file line number Diff line number Diff line change @@ -16,6 +16,7 @@ class SupportedModels(ExtendedEnum):
1616 LGBForecast = "lgbforecast"
1717 AutoMLX = "automlx"
1818 AutoTS = "autots"
19+ Theta = "theta"
1920 # Auto = "auto"
2021
2122
Original file line number Diff line number Diff line change 44# Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
55
66from ads .opctl .operator .lowcode .common .transformations import Transformations
7+ from .theta import ThetaOperatorModel
78
89from ..const import (
910 AUTO_SELECT ,
@@ -46,6 +47,7 @@ class ForecastOperatorModelFactory:
4647 SupportedModels .LGBForecast : MLForecastOperatorModel ,
4748 SupportedModels .AutoMLX : AutoMLXOperatorModel ,
4849 SupportedModels .AutoTS : AutoTSOperatorModel ,
50+ SupportedModels .Theta : ThetaOperatorModel ,
4951 }
5052
5153 @classmethod
You can’t perform that action at this time.
0 commit comments