Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/unit8co/darts
/ functions
Functions
3,734 in github.com/unit8co/darts
⨍
Functions
3,734
◇
Types & classes
510
↳
Endpoints
59
↓ 1 callers
Method
fit_detect
Trains the detector and detects anomalies on the same series. Parameters ---------- series Time series to be used
darts/ad/detectors/detectors.py:154
↓ 1 callers
Method
fit_transform
( self, x: torch.Tensor, mask: torch.Tensor | None = None )
darts/models/components/patchtst_fm_submodels.py:43
↓ 1 callers
Method
forward
(self, x: torch.Tensor)
darts/models/forecasting/tsmixer_model.py:211
↓ 1 callers
Method
forward
(self, *args, **kwargs)
darts/models/forecasting/sf_model.py:38
↓ 1 callers
Method
from_pickle
Read a pickled ``TimeSeries``. Parameters ---------- path : string path pointing to a pickle file that will be lo
darts/timeseries.py:1537
↓ 1 callers
Method
generate_predict_encodings
( self, n: int, series: TimeSeriesLike, past_covariates: TimeSeriesLike | None
darts/models/forecasting/forecasting_model.py:3589
↓ 1 callers
Method
get_attention_mask_future
Returns causal mask to apply for self-attention layer that acts on future input only. The model will attend to all `False` values.
darts/models/forecasting/tft_model.py:416
↓ 1 callers
Function
get_component_names
Extract and return the components of target series, static covariate, past and future covariates series. Parameters ---------- model
darts/explainability/utils.py:264
↓ 1 callers
Method
get_feature_importances
Returns the feature importances for the encoder, decoder and static covariates as pd.DataFrames. If multiple series were used in :fun
darts/explainability/explainability_result.py:631
↓ 1 callers
Method
get_feature_times_future
Helper function called by `get_feature_times` that extracts all of the times within `future_covariates` that can be used to create fe
darts/tests/utils/tabularization/test_create_lagged_prediction_data.py:150
↓ 1 callers
Method
get_feature_times_future
Helper function called by `get_feature_times` that extracts all times within `future_covariates` that can be used to create features.
darts/tests/utils/tabularization/test_create_lagged_training_data.py:177
↓ 1 callers
Method
get_feature_times_past
Helper function called by `get_feature_times` that extracts all times within `past_covariates` that can be used to create features. M
darts/tests/utils/tabularization/test_create_lagged_training_data.py:144
↓ 1 callers
Method
get_feature_times_target
Helper function called by `get_feature_times` that extracts all times within a `target_series` that can be used to create a feature a
darts/tests/utils/tabularization/test_create_lagged_training_data.py:115
↓ 1 callers
Method
get_feature_times_target_or_past
Helper function called by `get_feature_times` that extracts all of the times within `target_series` *or* `past_covariates` that can b
darts/tests/utils/tabularization/test_create_lagged_prediction_data.py:114
↓ 1 callers
Method
get_feature_times_target_prediction
Helper function that returns all the times within `target_series` that can be used to create features for prediction. Since
darts/tests/utils/tabularization/test_get_feature_times.py:98
↓ 1 callers
Method
get_feature_times_target_training
Helper function that returns all the times within `target_series` that can be used to create features and labels for training.
darts/tests/utils/tabularization/test_get_feature_times.py:38
↓ 1 callers
Method
get_matrices
Returns the G matrix given a specified reconciliation method.
darts/dataprocessing/transformers/reconciliation.py:232
↓ 1 callers
Method
get_relative_index
Returns scaled time index relative to prediction point.
darts/models/forecasting/tft_model.py:386
↓ 1 callers
Method
has_value
(cls, value)
darts/metrics/utils.py:61
↓ 1 callers
Method
helper_check_type_window
(scorer, **kwargs)
darts/tests/ad/test_scorers.py:746
↓ 1 callers
Method
helper_check_val_set
(self, model_cls, model_kwargs, fit_patch)
darts/tests/models/forecasting/test_torch_forecasting_model.py:1841
↓ 1 callers
Method
helper_check_val_set
( self, model_cls, model_kwargs, fit_patch, use_weights, stride )
darts/tests/models/forecasting/test_sklearn_models.py:2434
↓ 1 callers
Method
helper_evaluate_nll_scorer
( NLLscorer_to_test, distribution_arrays, deterministic_values, real_NLL_value
darts/tests/ad/test_scorers.py:1509
↓ 1 callers
Method
helper_get_model_params
( self, model_cls, series: dict, output_chunk_length: int )
darts/tests/utils/historical_forecasts/test_historical_forecasts.py:2971
↓ 1 callers
Method
helper_predict_from_ds_raise_on_missing_input
Helper function to test that the model raises an error when calling `predict_from_dataset()` after `fit_from_dataset()` with missing inputs.
darts/tests/models/forecasting/test_torch_forecasting_model.py:2776
↓ 1 callers
Function
helper_test_append_values
(test_series: TimeSeries)
darts/tests/test_timeseries.py:2511
↓ 1 callers
Method
helper_test_freq_coversion
(self, test_cases)
darts/tests/models/forecasting/test_prophet.py:175
↓ 1 callers
Method
helper_test_pred_length
(self, pytorch_model, series)
darts/tests/models/forecasting/test_block_RNN.py:184
↓ 1 callers
Method
helper_test_pred_length
(self, pytorch_model, series)
darts/tests/models/forecasting/test_transformer_model.py:93
↓ 1 callers
Method
helper_test_pred_length
(self, pytorch_model, series)
darts/tests/models/forecasting/test_TCN.py:186
↓ 1 callers
Method
helper_test_pred_length
(self, pytorch_model, series)
darts/tests/models/forecasting/test_RNN.py:162
↓ 1 callers
Method
helper_test_prediction_accuracy
prediction should be almost equal to y_true. Absolute tolarance is set to 0.2 to give some flexibility
darts/tests/models/forecasting/test_TFT.py:327
↓ 1 callers
Function
helper_test_prepend_values
(test_series: TimeSeries)
darts/tests/test_timeseries.py:2547
↓ 1 callers
Function
ic
Interval Coverage (IC). IC gives a binary outcome with `1` if the observation is within the interval, and `0` otherwise. For the true series
darts/metrics/metrics.py:3764
↓ 1 callers
Function
increment
Recursive function filling S for a given base component and all its ancestors
darts/dataprocessing/transformers/reconciliation.py:55
↓ 1 callers
Method
init_size
(self, n: int, m: int)
darts/dataprocessing/dtw/window.py:302
↓ 1 callers
Method
init_weights
(self)
darts/models/forecasting/tft_submodels.py:195
↓ 1 callers
Method
init_weights
(self)
darts/models/forecasting/tft_submodels.py:372
↓ 1 callers
Method
init_weights
(self)
darts/models/forecasting/tft_submodels.py:581
↓ 1 callers
Method
inverse_transform
Inverse transforms a (sequence of) series by calling the user-implemented `ts_inverse_transform` method. In case a sequence or list of lists
darts/dataprocessing/transformers/invertible_data_transformer.py:244
↓ 1 callers
Method
inverse_transform
(self, data, *args, **kwargs)
darts/tests/utils/historical_forecasts/test_historical_forecasts.py:4655
↓ 1 callers
Method
inverse_transform
(self, x: torch.Tensor)
darts/models/components/patchtst_fm_submodels.py:57
↓ 1 callers
Method
load
Loads a model from a given path or file handle. Parameters ---------- path Path or file handle from whic
darts/models/forecasting/ensemble_model.py:602
↓ 1 callers
Method
load_from_checkpoint
Load the model from automatically saved checkpoints under '{work_dir}/darts_logs/{model_name}/checkpoints/'. This method is used for
darts/models/forecasting/torch_forecasting_model.py:2198
↓ 1 callers
Method
load_model_weights
Load the model weights from a safetensors file. Parameters ---------- module The PyTorch module to load the weigh
darts/models/components/huggingface_connector.py:92
↓ 1 callers
Function
load_validation_inputs
Load validation inputs for TimesFM2p5Model fidelity tests. The data imports here are adapted from the `20-SKLearnModel-examples` notebook.
darts/tests/models/forecasting/test_timesfm2p5.py:40
↓ 1 callers
Function
load_validation_inputs
Load validation inputs for TiRexModel fidelity tests.
darts/tests/models/forecasting/test_tirex.py:42
↓ 1 callers
Function
load_validation_inputs
Load validation inputs for PatchTSTFMModel fidelity tests. The data imports here are adapted from the `20-SKLearnModel-examples` notebook.
darts/tests/models/forecasting/test_patchtst_fm.py:27
↓ 1 callers
Function
load_validation_inputs
Load validation inputs for Chronos2Model fidelity tests. The data imports here are adapted from the `20-SKLearnModel-examples` notebook.
darts/tests/models/forecasting/test_chronos2.py:52
↓ 1 callers
Method
load_weights_from_checkpoint
Load only the weights from automatically saved checkpoints under '{work_dir}/darts_logs/{model_name}/ checkpoints/'. This method is u
darts/models/forecasting/torch_forecasting_model.py:2317
↓ 1 callers
Method
make_splitter
( cls, data: TimeSeriesLike, test_size: float | int | None = 0.25, axis: int |
darts/utils/model_selection.py:160
↓ 1 callers
Function
mic
Mean Interval Coverage (MIC). MIC gives the time-aggregated Interval Coverage :func:`~darts.metrics.metrics.ic` - the ratio of observations b
darts/metrics/metrics.py:3874
↓ 1 callers
Method
on_validation_end
(self, trainer, pl_module)
darts/utils/callbacks.py:188
↓ 1 callers
Method
option_context
Context manager to temporarily set options.
darts/config.py:327
↓ 1 callers
Method
plot_attention
Plots the attention heads of the `TFTModel`. Parameters ---------- expl_result A `TFTExplainabilityResult` object
darts/explainability/tft_explainer.py:318
↓ 1 callers
Function
plot_ccf
Plots the Cross Correlation Function (CCF) between `ts` and `ts_other`, highlighting it at lag `m`, with corresponding significance interval.
darts/utils/statistics.py:820
↓ 1 callers
Function
plot_hist
This function plots the histogram of values in a TimeSeries instance or an array-like. All types of TimeSeries are supported (uni-, multivariate,
darts/utils/statistics.py:940
↓ 1 callers
Function
plot_pacf
Plots the Partial Autocorrelation Function (PACF) of `ts`, highlighting it at lag `m`, with corresponding significance interval. Uses :func:`
darts/utils/statistics.py:722
↓ 1 callers
Method
plot_variable_selection
Plots the variable selection / feature importances of the `TFTModel` based on the input. The figure includes three subplots: - encode
darts/explainability/tft_explainer.py:243
↓ 1 callers
Method
predict_likelihood_parameters
Returns the distribution parameters as a array, extracted from the raw model outputs.
darts/utils/likelihood_models/sklearn.py:92
↓ 1 callers
Method
predict_proba
(*args)
darts/tests/models/forecasting/test_classifier_model.py:1157
↓ 1 callers
Method
predict_raw
Returns the output of the base Facebook Prophet model in form of a pandas DataFrame. Note however, that the output of this method is not suppo
darts/models/forecasting/prophet_model.py:462
↓ 1 callers
Method
predict_series
Filters the given sequence of target time series with the filtering model. Parameters ---------- series The seque
darts/ad/anomaly_model/filtering_am.py:138
↓ 1 callers
Method
predict_series
Abstract method to implement the generation of predictions for the input `series`.
darts/ad/anomaly_model/anomaly_model.py:119
↓ 1 callers
Method
predict_series
Computes the historical forecasts that would have been obtained by the underlying forecasting model on `series`. `retrain` is set to
darts/ad/anomaly_model/forecasting_am.py:243
↓ 1 callers
Method
predict_step
performs the prediction step batch output of Darts' :class:`TorchInferenceDataset` - tuple of ``(past target, past cov,
darts/models/forecasting/pl_forecasting_module.py:335
↓ 1 callers
Function
prepare_onnx_inputs
Helper function to slice and concatenate the input features. In order to remove the dependency on the `model` argument, it can be decomposed into
darts/utils/onnx_utils.py:11
↓ 1 callers
Function
process_horizons_and_targets
Processes the input horizons and target component names. horizons Optionally, an integer or sequence of integers representing the future
darts/explainability/utils.py:206
↓ 1 callers
Function
r2_score
Coefficient of Determination :math:`R^2` (see [1]_ for more details). For the true series :math:`y` and predicted series :math:`\\hat{y}` of leng
darts/metrics/metrics.py:2332
↓ 1 callers
Function
random_walk_timeseries
Creates a random walk univariate TimeSeries, where each step is obtained by sampling a gaussian distribution with mean `mean` and standard de
darts/utils/timeseries_generation.py:305
↓ 1 callers
Method
reset_model
Resets the model object and removes all stored data - model, checkpoints, loggers and training history.
darts/models/forecasting/torch_forecasting_model.py:441
↓ 1 callers
Method
reset_option
Reset option(s) to default value(s).
darts/config.py:307
↓ 1 callers
Function
retain_period_common_to_all
Trims all series in the provided list, if necessary, so that the returned time series have a common span (corresponding to largest time sub-i
darts/utils/ts_utils.py:236
↓ 1 callers
Method
sample
Samples a prediction from the likelihood distribution and the predicted parameters.
darts/utils/likelihood_models/sklearn.py:86
↓ 1 callers
Method
sample
Samples a prediction from the likelihood distribution and the predicted parameters.
darts/utils/likelihood_models/statsforecast.py:80
↓ 1 callers
Method
save
Saves the model under a given path. Creates two files under ``path`` (model object) and ``path``.ckpt (checkpoint). Note: P
darts/models/forecasting/torch_forecasting_model.py:2053
↓ 1 callers
Method
select_best_model
Performs a grid search over all hyper parameters to select the best model, using the fitted values on the training series `ts`.
darts/models/forecasting/theta.py:442
↓ 1 callers
Method
set
Set the option value with validation.
darts/config.py:83
↓ 1 callers
Method
set_n_jobs
Set the number of processors to be used by the transformer while processing multiple ``TimeSeries``. Parameters ---------- va
darts/dataprocessing/transformers/base_data_transformer.py:207
↓ 1 callers
Method
set_verbose
Set the verbosity status. `True` for enabling the detailed report about scaler's operation progress, `False` for no additional inform
darts/dataprocessing/transformers/base_data_transformer.py:191
↓ 1 callers
Function
setup_test_case
()
darts/tests/test_timeseries_static_covariates.py:32
↓ 1 callers
Method
shap_explanations_single
Similar to :func:`shap_explanations()`, but computes SHAP explanations for only one forecasted timestamp, which corresponds to the la
darts/explainability/shap_adapters/shap_adapter.py:291
↓ 1 callers
Function
suppress_lightning_warnings
(suppress_all: bool = False)
darts/logging.py:216
↓ 1 callers
Method
to_cpu
Updates the PyTorch Lightning Trainer parameters to move the model to CPU the next time :func:`fit()` or :func:`predict()` is called.
darts/models/forecasting/torch_forecasting_model.py:2512
↓ 1 callers
Method
to_dense
Returns ------- Dense n x m numpy array, where empty cells are set to np.inf
darts/dataprocessing/dtw/cost_matrix.py:45
↓ 1 callers
Method
to_pickle
Save the series in pickle format. Parameters ---------- path : string path to a file where current object will be
darts/timeseries.py:4408
↓ 1 callers
Method
transform
( self, series: TimeSeriesLike, *args, component_mask: np.ndarray | None = Non
darts/dataprocessing/transformers/fittable_data_transformer.py:295
↓ 1 callers
Method
transform
This method applies transformation to the non-transformed encoded covariates output of `SequentialEncoder._encode_sequence()` after being merg
darts/dataprocessing/encoders/encoder_base.py:860
↓ 1 callers
Method
transform
(self, data, *args, **kwargs)
darts/tests/utils/historical_forecasts/test_historical_forecasts.py:4651
↓ 1 callers
Method
transform
(self, x: torch.Tensor)
darts/models/components/patchtst_fm_submodels.py:50
↓ 1 callers
Method
transformers
Returns a tuple of (past transformer, future transformer).
darts/dataprocessing/encoders/encoders.py:1370
↓ 1 callers
Method
ts_fit
The function that will be applied to each series when :func:`fit` is called. If the `global_fit` attribute is set to `False`, then `ts_fit` s
darts/dataprocessing/transformers/fittable_data_transformer.py:179
↓ 1 callers
Method
ts_inverse_transform
The function that will be applied to each series when :func:`inverse_transform` is called. The function must take as first argument a ``TimeS
darts/dataprocessing/transformers/invertible_data_transformer.py:165
↓ 1 callers
Method
ts_transform
The function that will be applied to each series when :func:`transform()` is called. This method is not implemented in the base class and mus
darts/dataprocessing/transformers/base_data_transformer.py:227
↓ 1 callers
Method
untrained_model
Returns a new (untrained) model instance created with the same parameters.
darts/models/forecasting/forecasting_model.py:2648
Method
__add__
(self, other)
darts/timeseries.py:5372
Method
__add__
(self, other: int)
darts/utils/ts_utils.py:44
Method
__call__
(self, *args, **kwargs)
darts/utils/utils.py:72
Method
__call__
(self, X)
darts/tests/explainability/test_shap_explainer.py:69
Method
__call__
(cls, *args, **kwargs)
darts/models/forecasting/forecasting_model.py:104
← previous
next →
901–1,000 of 3,734, ranked by callers