chronax.models.fedformer

`FEDformer`

chronax.models.fedformer.forecaster.FEDformer · inherits BaseForecaster

Public entry point for the Chronax FEDformer model, a JAX/Flax/Optax univariate time-series forecaster.

`init(self, ...)`

(Signature not available in source.)

`fit(self, y, X=None) -> Self`

(Signature details not available in source.)

Returns: Self (the fitted forecaster; sets self.model_).

`predict(self, h, X=None, level=None) -> dict`

(Signature details not available in source.)

Returns: dict (Prediction results).

`forecast(self, y, h, X=None, X_future=None, level=None, fitted=False) -> dict`

(Signature details not available in source.)

Returns: dict (Prediction results).

`FEDformerForecaster`

chronax.models.fedformer.forecaster.FEDformerForecaster · inherits BaseForecaster

Deprecated config-based alias for FEDformer.

`FEDformerConfig`

chronax.models.fedformer.model.FEDformerConfig

(Configuration class for the FEDformer model. Details not available in source.)

`FEDformerModel`

chronax.models.fedformer.model.FEDformerModel

(Flax Module implementing the core FEDformer architecture. Details not available in source.)

`RobustScaler`

chronax.models.fedformer.data.RobustScaler

(Data utility class for robust scaling. Details not available in source.)

`TrainState`

chronax.models.fedformer.train.TrainState

(Flax/Optax training state container. Details not available in source.)

`LossFn`

chronax.models.fedformer.loss.LossFn

(Type alias or class representing a loss function. Details not available in source.)

`LOSSES`

chronax.models.fedformer.loss.LOSSES

(Mapping or collection of available loss functions. Details not available in source.)

`FourierBlock`

chronax.models.fedformer.model.FourierBlock

(Flax Module component of the FEDformer model. Details not available in source.)

`FourierCrossAttention`

chronax.models.fedformer.model.FourierCrossAttention

(Flax Module component of the FEDformer model. Details not available in source.)

`MultiHeadProjection`

chronax.models.fedformer.model.MultiHeadProjection

(Flax Module component of the FEDformer model. Details not available in source.)

`SeasonalLayerNorm`

chronax.models.fedformer.model.SeasonalLayerNorm

(Flax Module component of the FEDformer model. Details not available in source.)

`TokenEmbedding`

chronax.models.fedformer.model.TokenEmbedding

(Flax Module component of the FEDformer model. Details not available in source.)

`EncoderLayer`

chronax.models.fedformer.model.EncoderLayer

(Flax Module component of the FEDformer model. Details not available in source.)

`Encoder`

chronax.models.fedformer.model.Encoder

(Flax Module component of the FEDformer model. Details not available in source.)

`DecoderLayer`

chronax.models.fedformer.model.DecoderLayer

(Flax Module component of the FEDformer model. Details not available in source.)

`Decoder`

chronax.models.fedformer.model.Decoder

(Flax Module component of the FEDformer model. Details not available in source.)

`moving_avg`

chronax.models.fedformer.model.moving_avg

(Math primitive function. Details not available in source.)

`series_decomp`

chronax.models.fedformer.model.series_decomp

(Math primitive function for time series decomposition. Details not available in source.)

`get_frequency_modes`

chronax.models.fedformer.model.get_frequency_modes

(Math primitive function. Details not available in source.)

`build_windows`

chronax.models.fedformer.data.build_windows

(Data utility function. Details not available in source.)

`split_train_val_windows`

chronax.models.fedformer.data.split_train_val_windows

(Data utility function. Details not available in source.)

`pad_sequence`

chronax.models.fedformer.data.pad_sequence

(Data utility function. Details not available in source.)

`create_batch`

chronax.models.fedformer.data.create_batch

(Data utility function. Details not available in source.)

`batch_generator`

chronax.models.fedformer.data.batch_generator

(Data utility function. Details not available in source.)

`mae`

chronax.models.fedformer.loss.mae

(Loss function: Mean Absolute Error. Details not available in source.)

`mse`

chronax.models.fedformer.loss.mse

(Loss function: Mean Squared Error. Details not available in source.)

`huber`

chronax.models.fedformer.loss.huber

(Loss function: Huber loss. Details not available in source.)

`masked_mae`

chronax.models.fedformer.loss.masked_mae

(Loss function: Masked Mean Absolute Error. Details not available in source.)

`masked_mse`

chronax.models.fedformer.loss.masked_mse

(Loss function: Masked Mean Squared Error. Details not available in source.)

`resolve`

chronax.models.fedformer.loss.resolve

(Function to resolve loss functions. Details not available in source.)