Skip to content

Commit c068dd8

Browse files
committed
added theta forecasting model
1 parent 5ef6e9c commit c068dd8

File tree

5 files changed

+395
-0
lines changed

5 files changed

+395
-0
lines changed

ads/model/model_metadata.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff 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"

ads/opctl/operator/lowcode/forecast/const.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff 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

ads/opctl/operator/lowcode/forecast/model/factory.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
# Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
55

66
from ads.opctl.operator.lowcode.common.transformations import Transformations
7+
from .theta import ThetaOperatorModel
78

89
from ..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

0 commit comments

Comments
 (0)