MCPcopy Create free account

hub / github.com/unit8co/darts / functions

Functions3,734 in github.com/unit8co/darts

Method__init__
TBATS based on the `Statsforecasts package <https://github.com/Nixtla/statsforecast>`__. Trigonometric Box-Cox transform, ARMA errors, Trend
darts/models/forecasting/sf_tbats.py:12
Method__init__
Auto-ARIMA based on the `Statsforecasts package <https://github.com/Nixtla/statsforecast>`__. Automatically selects the best AutoRegressive I
darts/models/forecasting/sf_auto_arima.py:12
Method__init__
(self)
darts/models/forecasting/sklearn_model.py:1620
Method__init__
Extension of `SKLearnModel` for regression models that support categorical features. Parameters ---------- model
darts/models/forecasting/sklearn_model.py:1625
Method__init__
Regression Model Can be used to fit any scikit-learn-like regressor class to predict the target time series from lagged values. Note:
darts/models/forecasting/sklearn_model.py:1947
Method__init__
SKLearn Classifier Model Can be used to fit any scikit-learn-like classifier class to predict the target time series with categorical
darts/models/forecasting/sklearn_model.py:2203
Method__init__
XGBoost Model for classification forecasting Parameters ---------- lags Lagged target `series` values used to pre
darts/models/forecasting/xgboost.py:389
Method__init__
CatBoost Model for classification forecasting Parameters ---------- lags Lagged target `series` values used to pr
darts/models/forecasting/catboost_model.py:484
Method__init__
This class allows to create custom RNN modules that can later be used with Darts' :class:`RNNModel`. It adds the backbone that is required to
darts/models/forecasting/rnn_model.py:36
Method__init__
PyTorch module implementing an RNN to be used in `RNNModel`. PyTorch module implementing a simple RNN with the specified `name` type.
darts/models/forecasting/rnn_model.py:221
Method__init__
Kalman filter Forecaster This model uses a Kalman filter to produce forecasts. It uses a :class:`darts.models.filtering.kalman_filter
darts/models/forecasting/kalman_forecaster.py:29
Method__init__
Use a regression model for ensembling individual models' predictions using the stacking technique [1]_. The provided regression mode
darts/models/forecasting/regression_ensemble_model.py:29
Method__init__
(self, expansion_coefficient_dim, target_length)
darts/models/forecasting/nbeats.py:49
Method__init__
(self, target_length)
darts/models/forecasting/nbeats.py:68
Method__init__
PyTorch module implementing one stack of the N-BEATS architecture that comprises multiple basic blocks. Parameters ----------
darts/models/forecasting/nbeats.py:237
Method__init__
PyTorch module implementing the N-BEATS architecture. Parameters ---------- output_dim Number of output component
darts/models/forecasting/nbeats.py:363
Method__init__
Neural Basis Expansion Analysis Time Series Forecasting (N-BEATS). This is an implementation of the N-BEATS architecture, as outlined in [1]_
darts/models/forecasting/nbeats.py:539
Method__init__
Random Forest Model Parameters ---------- lags Lagged target `series` values used to predict the next time step/s
darts/models/forecasting/random_forest.py:31
Method__init__
Auto-CES based on the `Statsforecasts package <https://github.com/Nixtla/statsforecast>`__. Automatically selects the best Complex Exponentia
darts/models/forecasting/sf_auto_ces.py:12
Method__init__
Fast Fourier Transform Model This model performs forecasting on a TimeSeries instance using FFT, subsequent frequency filtering (cont
darts/models/forecasting/fft.py:214
Method__init__
PyTorch module implementing a residual block module used in `_TCNModule`. Parameters ---------- num_filters The n
darts/models/forecasting/tcn_model.py:28
Method__init__
PyTorch module implementing a dilated TCN module used in `TCNModel`. Parameters ---------- input_size The dimens
darts/models/forecasting/tcn_model.py:135
Method__init__
(self, *args, batch_first: bool = False, **kwargs)
darts/models/forecasting/tft_submodels.py:54
Method__init__
Embedding layer for categorical variables including groups of categorical variables. Enabled for static and dynamic categories (i.e. 3 dimensi
darts/models/forecasting/tft_submodels.py:80
Method__init__
( self, output_size: int, batch_first: bool = False, trainable: bool = False )
darts/models/forecasting/tft_submodels.py:134
Method__init__
( self, input_size: int, hidden_size: int | None = None, dropout: float | None
darts/models/forecasting/tft_submodels.py:178
Method__init__
( self, input_size: int, output_size: int | None = None, trainable_add: bool =
darts/models/forecasting/tft_submodels.py:211
Method__init__
( self, input_size: int, hidden_size: int | None = None, skip_size: int | None
darts/models/forecasting/tft_submodels.py:289
Method__init__
( self, input_size: int, hidden_size: int, output_size: int, dropout:
darts/models/forecasting/tft_submodels.py:324
Method__init__
Calculate weights for ``num_inputs`` variables which are each of size ``input_size``
darts/models/forecasting/tft_submodels.py:401
Method__init__
(self, dropout: float | None = None, scale: bool = True)
darts/models/forecasting/tft_submodels.py:532
Method__init__
(self, n_head: int, d_model: int, dropout: float = 0.0)
darts/models/forecasting/tft_submodels.py:561
Method__init__
This class allows to create custom block RNN modules that can later be used with Darts' :class:`BlockRNNModel`. It adds the backbone that is r
darts/models/forecasting/block_rnn_model.py:31
Method__init__
PyTorch module implementing a block RNN to be used in `BlockRNNModel`. PyTorch module implementing a simple block RNN with the specified `nam
darts/models/forecasting/block_rnn_model.py:118
Method__init__
PyTorch module implementing one stack of the N-BEATS architecture that comprises multiple basic blocks. Parameters ----------
darts/models/forecasting/nhits.py:210
Method__init__
PyTorch module implementing the N-HiTS architecture. Parameters ---------- input_dim The number of input componen
darts/models/forecasting/nhits.py:321
Method__init__
An implementation of the N-HiTS model, as presented in [1]_. N-HiTS is similar to N-BEATS (implemented in :class:`NBEATSModel`), but
darts/models/forecasting/nhits.py:464
Method__init__
(self, add_encoders: dict | None = None)
darts/models/forecasting/forecasting_model.py:2911
Method__init__
(self, add_encoders: dict | None = None)
darts/models/forecasting/forecasting_model.py:2992
Method__init__
A batch normalization layer that normalizes over the last two dimensions of a Tensor.
darts/models/forecasting/tsmixer_model.py:68
Method__init__
A module for feature mixing with flexibility in normalization and activation based on the `PyTorch implementation of TSMixer <https://github.c
darts/models/forecasting/tsmixer_model.py:88
Method__init__
Conditional mix layer combining time and feature mixing with static context based on the `PyTorch implementation of TSMixer <https://github.co
darts/models/forecasting/tsmixer_model.py:224
Method__init__
Initializes the TSMixer module for use within a Darts forecasting model. Parameters ---------- input_dim
darts/models/forecasting/tsmixer_model.py:311
Method__init__
Time-Series Mixer (TSMixer): An All-MLP Architecture for Time Series. This is an implementation of the TSMixer architecture, as outlined in [
darts/models/forecasting/tsmixer_model.py:519
Method__init__
ARIMA ARIMA-type models extensible with exogenous variables (future covariates) and seasonal components. Parameters -
darts/models/forecasting/arima.py:34
Method__init__
StatsForecast Model. Can be used to fit any `StatsForecast` base model. For more information on available models, see the `StatsForec
darts/models/forecasting/sf_model.py:43
Method__init__
Naive combination model Naive implementation of `EnsembleModel` Returns the average of all predictions of the constituent models
darts/models/forecasting/naive_ensemble_model.py:20
Method__init__
Croston method as presented `in this paper <https://otexts.com/fpp3/counts.html>`__ and based on the `Statsforecasts package <https://github.c
darts/models/forecasting/sf_croston.py:18
Method__init__
PyTorch module implementing the TFT architecture from `this paper <https://arxiv.org/pdf/1912.09363.pdf>`__ The implementation is built upon `
darts/models/forecasting/tft_model.py:39
Method__init__
Pytorch Lightning (PL)-based Forecasting Model. This class is meant to be inherited to create a new PL-based forecasting model. It go
darts/models/forecasting/torch_forecasting_model.py:136
Method__init__
( self, forecasting_models: list[ForecastingModel], ensemble_model: SKLearnModel | Non
darts/models/forecasting/ensemble_model.py:78
Method__init__
( self, tirex_kwargs: dict[str, Any], all_quantiles: tuple[float, ...], enable
darts/models/forecasting/tirex_model.py:44
Method__init__
An implementation of the 4Theta method with configurable `theta` parameter. See M4 competition `solution <https://github.com/Mcompet
darts/models/forecasting/theta.py:205
Method__init__
Pytorch module implementing the Residual Block from the TiDE paper.
darts/models/forecasting/tide_model.py:22
Method__init__
Pytorch module implementing the TiDE architecture. Parameters ---------- input_dim The number of input components
darts/models/forecasting/tide_model.py:62
Method__init__
VARIMA Parameters ---------- p : int Order (number of time lags) of the autoregressive model (AR) d : int
darts/models/forecasting/varima.py:30
Method__init__
Auto-TBATS based on the `Statsforecasts package <https://github.com/Nixtla/statsforecast>`__. Automatically selects the best TBATS model from
darts/models/forecasting/sf_auto_tbats.py:12
Method__init__
Auto-ETS based on the `Statsforecasts package <https://github.com/Nixtla/statsforecast>`__. Automatically selects the best Exponential Smooth
darts/models/forecasting/sf_auto_ets.py:12
Method__init__
PyTorch module implementing the TimesFM 2.5 model, ported from `google-research/timesfm <https://github.com/google-research/timesfm/>`_ and
darts/models/forecasting/timesfm2p5_model.py:98
Method__init__
Auto-MFLES based on the `Statsforecasts package <https://github.com/Nixtla/statsforecast>`__. Automatically selects the best MFLES model from
darts/models/forecasting/sf_auto_mfles.py:15
Method__init__
Auto-Theta based on the `Statsforecasts package <https://github.com/Nixtla/statsforecast>`__. Automatically selects the best Theta model usin
darts/models/forecasting/sf_auto_theta.py:12
Method__init__
PyTorch Lightning-based Forecasting Module. This class is meant to be inherited to create a new PyTorch Lightning-based forecasting
darts/models/forecasting/pl_forecasting_module.py:80
Method__init__
PyTorch module implementing the Chronos-2 model, ported from `amazon-science/chronos-forecasting <https://github.com/amazon-science/chronos-fo
darts/models/forecasting/chronos2_model.py:54
Method__init__
An implementation of positional encoding as described in 'Attention is All you Need' by Vaswani et al. (2017) Parameters ----------
darts/models/forecasting/transformer_model.py:78
Method__init__
PyTorch module implementing a Transformer to be used in `TransformerModel`. PyTorch module implementing a simple encoder-decoder transformer
darts/models/forecasting/transformer_model.py:121
Method__init__
( self, context_length: int = 8192, d_patch: int = 16, d_model: int = 1024,
darts/models/forecasting/patchtst_fm_model.py:51
Method__init__
PyTorch module implementing PatchTST-FM, ported from `ibm-granite/granite-tsfm <https://github.com/ibm-granite/granite-tsfm>`_ and ada
darts/models/forecasting/patchtst_fm_model.py:153
Method__init__
PyTorch Lightning module that wraps around the NeuralForecast model and implements the :func:`forward()` API for Darts' ``PLForecastingModule`
darts/models/forecasting/nf_model.py:131
Method__init__
NeuralForecast Model. Can be used to fit any `NeuralForecast` univariate or multivariate base model. For a list of available base mod
darts/models/forecasting/nf_model.py:337
Method__init__
PyTorch module implementing the N-HiTS architecture. Parameters ---------- input_dim The number of input componen
darts/models/forecasting/nlinear.py:23
Method__init__
LGBM Model for classification forecasting Parameters ---------- lags Lagged target `series` values used to predic
darts/models/forecasting/lgbm.py:376
Method__init__
Foundation Forecasting Model with PyTorch Lightning backend. This class is meant to be inherited to create a new foundation forecasting model
darts/models/forecasting/foundation_model.py:20
Method__init__
Pytorch module for implementing naive models. Implement your own naive module by subclassing from `_GlobalNaiveModule`, and implement the
darts/models/forecasting/global_baseline_models.py:66
Method__init__
Base class for global naive models. The naive models inherit from `MixedCovariatesTorchModel` giving access to past, future, and static covari
darts/models/forecasting/global_baseline_models.py:100
Method__init__
( self, agg_fn: Callable[[torch.Tensor, int], torch.Tensor], *args, **kwargs )
darts/models/forecasting/global_baseline_models.py:280
Method__init__
Global Naive Aggregate Model. The model generates forecasts for each `series` as described below: - take an aggregate (computed with
darts/models/forecasting/global_baseline_models.py:293
Method__init__
Global Naive Seasonal Model. The model generates forecasts for each `series` as described below: - take the value from each target c
darts/models/forecasting/global_baseline_models.py:453
Method__init__
Linear regression model. Parameters ---------- lags Lagged target `series` values used to predict the next time s
darts/models/forecasting/linear_regression_model.py:30
Method__init__
Facebook Prophet This class provides a basic wrapper around `Facebook Prophet <https://github.com/facebook/prophet>`__. It supports a
darts/models/forecasting/prophet_model.py:29
Method__init__
Exponential Smoothing This is a wrapper around `statsmodels Holt-Winters' Exponential Smoothing <https://www.statsmodels.org
darts/models/forecasting/exponential_smoothing.py:22
Method__init__
Naive Conformal Prediction Model. A probabilistic model that adds calibrated intervals around the median forecast from a pre-trained
darts/models/forecasting/conformal_models.py:1589
Method__init__
Conformalized Quantile Regression Model. A probabilistic model that calibrates the quantile predictions from a pre-trained probabilistic glob
darts/models/forecasting/conformal_models.py:1722
Method__init__
Naive Seasonal Model This model always predicts the value of `K` time steps ago. When `K=1`, this model predicts the last value of th
darts/models/forecasting/baselines.py:67
Method__init__
Naive Drift Model This model fits a line between the first and last point of the training series, and extends it in the future. For a
darts/models/forecasting/baselines.py:127
Method__init__
Naive Moving Average Model This model forecasts using an autoregressive moving average (ARMA). Parameters ----------
darts/models/forecasting/baselines.py:181
Method__init__
(self, kernel_size)
darts/models/forecasting/dlinear.py:49
Method__init__
PyTorch module implementing the DLinear architecture. Parameters ---------- input_dim The number of input compone
darts/models/forecasting/dlinear.py:64
Method__init__
An implementation of the DLinear model, as presented in [1]_. This implementation is improved by allowing the optional use of past covariates
darts/models/forecasting/dlinear.py:225
Method__init__
This model uses the ``GaussianProcessRegressor`` of scikit-learn to fit a Gaussian Process to the supplied TimeSeries. This can then
darts/models/filtering/gaussian_process_filter.py:15
Method__init__
This model implements a Kalman filter over a time series. The key method is `KalmanFilter.filter()`. It considers the provid
darts/models/filtering/kalman_filter.py:20
Method__init__
Parameters ---------- window The length of the window over which to average values centered S
darts/models/filtering/moving_average_filter.py:15
Method__init__
(self)
darts/models/filtering/filtering_model.py:23
Method__init__
The base class for forecasting model explainers. It defines the *minimal* behavior that all forecasting model explainers support.
darts/explainability/explainability.py:24
Method__init__
SHAP Explainer for SKLearn and Torch Models. **Definitions**: - A background series is a ``TimeSeries`` used to train the SHAP expla
darts/explainability/shap_explainer.py:83
Method__init__
( self, explained_components: dict[str, Any] | list[dict[str, Any]], )
darts/explainability/explainability_result.py:62
Method__init__
( self, explained_forecasts: dict[int, dict[str, TimeSeries]] | list[dict[int, dict[st
darts/explainability/explainability_result.py:219
Method__init__
( self, explained_forecasts: dict[int, dict[str, TimeSeries]] | list[dict[int, dict[st
darts/explainability/explainability_result.py:408
Method__init__
( self, explained_components: dict[str, TimeSeries], feature_values: dict[str, TimeSer
darts/explainability/explainability_result.py:502
Method__init__
Explainer class for the `TFTModel`. **Definitions** - A background series is a `TimeSeries` that is used as a default for g
darts/explainability/tft_explainer.py:50
Method__init__
(self, *args, time_index: pd.Index, **kwargs)
darts/explainability/shap_adapters/shap_adapter.py:37
← previousnext →1,201–1,300 of 3,734, ranked by callers