| from models.toto_model import TotoModel |
| from models.chronos2_model import Chronos2Model |
| from models.statistical_baselines import ( |
| AutoARIMAModel, |
| AutoETSModel, |
| DriftModel, |
| HoltWintersModel, |
| MeanModel, |
| MovingAverageModel, |
| NaiveModel, |
| SeasonalNaiveModel, |
| ThetaModel, |
| ) |
| from models.emerging_models import ( |
| Moirai2Model, |
| SundialBaseModel, |
| TiRexModel, |
| ) |
|
|
| FOUNDATION_MODELS = [ |
| "Toto", |
| "Chronos-2", |
| "TiRex", |
| "Moirai-2.0", |
| "Sundial-Base", |
| ] |
|
|
| STATISTICAL_MODELS = [ |
| "Naive", |
| "Seasonal Naive", |
| "Mean", |
| "Drift", |
| "Moving Average", |
| "AutoARIMA", |
| "AutoETS", |
| "Theta", |
| "Holt-Winters", |
| ] |
|
|
| MODEL_CATEGORIES = { |
| "Foundation Models": FOUNDATION_MODELS, |
| "Statistical Models": STATISTICAL_MODELS, |
| } |
|
|
| MODEL_REGISTRY = { |
| "Toto": TotoModel, |
| "Chronos-2": Chronos2Model, |
| "TiRex": TiRexModel, |
| "Moirai-2.0": Moirai2Model, |
| "Sundial-Base": SundialBaseModel, |
| "Naive": NaiveModel, |
| "Seasonal Naive": SeasonalNaiveModel, |
| "Mean": MeanModel, |
| "Drift": DriftModel, |
| "Moving Average": MovingAverageModel, |
| "AutoARIMA": AutoARIMAModel, |
| "AutoETS": AutoETSModel, |
| "Theta": ThetaModel, |
| "Holt-Winters": HoltWintersModel, |
| } |
|
|
| __all__ = [ |
| "MODEL_REGISTRY", |
| "MODEL_CATEGORIES", |
| "FOUNDATION_MODELS", |
| "STATISTICAL_MODELS", |
| "TotoModel", |
| "Chronos2Model", |
| "NaiveModel", |
| "SeasonalNaiveModel", |
| "MeanModel", |
| "DriftModel", |
| "MovingAverageModel", |
| "AutoARIMAModel", |
| "AutoETSModel", |
| "ThetaModel", |
| "HoltWintersModel", |
| "TiRexModel", |
| "Moirai2Model", |
| "SundialBaseModel", |
| ] |
|
|
