ForecastingWorkflowConfig#

class openstef_models.presets.ForecastingWorkflowConfig(**data: Any) → None[source]#

Bases: BaseConfig

Configuration for forecasting workflows.

Defines all parameters needed to set up a forecasting model, including model type, hyperparameters, location information, data columns, and feature engineering settings.

Parameters:: data (Any)

kind: Literal['single']#: Discriminator tag for config type.

model_id: ModelIdentifier#: Unique identifier for the forecasting model.

run_name: str | None#: Optional name for this workflow run, can be used for versioning.

model: Literal['xgboost', 'gblinear', 'flatliner', 'median', 'constant_quantile', 'lgbm', 'lgbmlinear']#: Type of forecasting model to use.

quantiles: list[Quantile]#: List of quantiles to predict for probabilistic forecasting.

sample_interval: timedelta#: Time interval between consecutive data samples.

horizons: list[LeadTime]#: List of forecast horizons to predict.

xgboost_hyperparams: XGBoostHyperParams#: Hyperparameters for XGBoost forecaster.

gblinear_hyperparams: GBLinearHyperParams#: Hyperparameters for GBLinear forecaster.

lgbm_hyperparams: LGBMHyperParams#: Hyperparameters for LightGBM forecaster.

lgbmlinear_hyperparams: LGBMLinearHyperParams#: Hyperparameters for LightGBM forecaster.

location: LocationConfig#: Location information for the forecasting workflow.

target_column: str#: Name of the target variable column in datasets.

energy_price_column: str#: Name of the energy price column in datasets.

radiation_column: str#: Name of the radiation column in datasets.

wind_speed_column: str#: Name of the wind speed column in datasets.

pressure_column: str#: Name of the pressure column in datasets.

temperature_column: str#: Name of the temperature column in datasets.

relative_humidity_column: str#: Name of the relative humidity column in datasets.

selected_features: FeatureSelection#: Feature selection for which features to include/exclude.

predict_history: timedelta#: Amount of historical data available at prediction time.

cutoff_history: timedelta#: Amount of historical data to exclude from training and prediction due to incomplete features from lag-based preprocessing. When using lag transforms (e.g., lag-14), the first N days contain NaN values. Set this to match your maximum lag duration (e.g., timedelta(days=14)). Default of 0 assumes no invalid rows are created by preprocessing. Note: should be same as predict_history if you are using lags. We default to disabled to keep the same behaviour as openstef 3.0.

completeness_threshold: float#: Minimum fraction of data that should be available for making a regular forecast.

completeness_threshold_target_constant_quantile: float#: Minimum fraction of target data that should be available for making a constant quantile forecast.

flatliner_threshold: timedelta#: Number of minutes that the load has to be constant to detect a flatliner.

detect_non_zero_flatliner: bool#: If True, flatliners are also detected on non-zero values (median of the load).

predict_nonzero_flatliner: bool#: If True, predict the median of load measurements instead of zero (only for flatliner model).

shifters: list[Shifter]#: List of feature shifts to align aggregation intervals. Each Shifter can target different features with different aggregation periods.

rolling_aggregate_features: list[AggregationFunction]#: If not None, rolling aggregate(s) of load will be used as features in the model.

clip_features: FeatureSelection#: Feature selection for which features to clip to their learned range.

nan_on_outlier_features: FeatureSelection#: Feature selection for which features to replace out-of-range values with NaN. Defaults to no features (disabled).

max_day_lags: int#: Maximum number of days to look back for day-based lags. Default is 14 days (two weekly cycles). Set to 7 for a single weekly cycle.

sample_weight_config: SampleWeightConfig#: Sample weighting configuration. Controls how training samples are weighted. Defaults to weight_exponent=1.0 for gblinear, 0.0 (uniform) for other models.

data_splitter: DataSplitter#: Configuration for splitting data into training, validation, and test sets.

evaluation_metrics: list[MetricProvider]#: List of metric providers for evaluating model score.

mlflow_storage: MLFlowStorage | None#: Configuration for MLflow experiment tracking and model storage.

model_reuse_enable: bool#: Whether to enable reuse of previously trained models.

model_reuse_max_age: timedelta#: Maximum age of a model to be considered for reuse.

model_selection_enable: bool#: Whether to enable automatic model selection based on performance.

model_selection_metric: tuple[Quantile | Literal['global'], str, MetricDirection]#: Metric to monitor for model performance when retraining.

model_selection_old_model_penalty: float#: Penalty to apply to the old model’s metric to bias selection towards newer models.

model_performance_callback_enabled: bool#: Whether to enable the ModelPerformanceCallback that evaluates model performance at the end of fitting.

model_performance_callback_metric_threshold: tuple[Quantile | Literal['global'], str, MetricDirection, float]#: Metric to monitor for model performance threshold at the end of fitting.

verbosity: Literal[0, 1, 2, 3, True]#: Verbosity level. 0=silent, 1=warning, 2=info, 3=debug

tags: dict[str, str]#: Optional metadata tags for the model run.

experiment_tags: dict[str, str]#: Optional metadata tags for experiment tracking.

model_config: ClassVar[ConfigDict] = {'arbitrary_types_allowed': False, 'extra': 'ignore', 'protected_namespaces': ()}#: Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].