Sequential Training Dataset

class darts.utils.data.sequential_dataset.DualCovariatesSequentialDataset(target_series, covariates=None, input_chunk_length=12, output_chunk_length=1, output_chunk_shift=0, max_samples_per_ts=None, use_static_covariates=True)[source]

Bases: DualCovariatesTrainingDataset

A time series dataset containing tuples of (past_target, historic_future_covariates, future_covariates, static_covariates, future_target). The “past” series (incl historic_future_covariates) have length input_chunk_length and the “future” series have length output_chunk_length. The “future” series are immediately consecutive to the “past” series. The slicing of past and future covariates matches that of past and future targets, respectively. The slicing itself relies on time indexes to align the series if they have unequal lengths.

Each series must be long enough to contain at least one (input, output) pair; i.e., each series must have length at least input_chunk_length + output_chunk_length. If these conditions are not satisfied, an error will be raised when trying to access some of the splits.

The sampling is uniform over the number of time series; i.e., the i-th sample of this dataset has a probability 1/N of coming from any of the N time series in the sequence. If the time series have different lengths, they will contain different numbers of slices. Therefore, some particular slices may be sampled more often than others if they belong to shorter time series.

Parameters
  • target_series (Union[TimeSeries, Sequence[TimeSeries]]) – One or a sequence of target TimeSeries.

  • covariates (Union[TimeSeries, Sequence[TimeSeries], None]) – Optionally, one or a sequence of TimeSeries containing future-known covariates. If this parameter is set, the provided sequence must have the same length as that of target_series. Moreover, all covariates in the sequence must have a time span large enough to contain all the required slices. The joint slicing of the target and covariates is relying on the time axes of both series.

  • input_chunk_length (int) – The length of the emitted past series.

  • output_chunk_length (int) – The length of the emitted future series.

  • output_chunk_shift (int) – Optionally, the number of steps to shift the start of the output chunk into the future.

  • max_samples_per_ts (Optional[int]) – This is an upper bound on the number of tuples that can be produced per time series. It can be used in order to have an upper bound on the total size of the dataset and ensure proper sampling. If None, it will read all of the individual time series in advance (at dataset creation) to know their sizes, which might be expensive on big datasets. If some series turn out to have a length that would allow more than max_samples_per_ts, only the most recent max_samples_per_ts samples will be considered.

  • use_static_covariates (bool) – Whether to use/include static covariate data from input series.

class darts.utils.data.sequential_dataset.FutureCovariatesSequentialDataset(target_series, covariates=None, input_chunk_length=12, output_chunk_length=1, output_chunk_shift=0, max_samples_per_ts=None, use_static_covariates=True)[source]

Bases: FutureCovariatesTrainingDataset

A time series dataset containing tuples of (past_target, future_covariates, static_covariates, future_target). The “past” series have length input_chunk_length and the “future” series have length output_chunk_length. The “future” series are immediately consecutive to the “past” series. The slicing of past and future covariates matches that of past and future targets, respectively. The slicing itself relies on time indexes to align the series if they have unequal lengths.

Each series must be long enough to contain at least one (input, output) pair; i.e., each series must have length at least input_chunk_length + output_chunk_length. If these conditions are not satisfied, an error will be raised when trying to access some of the splits.

The sampling is uniform over the number of time series; i.e., the i-th sample of this dataset has a probability 1/N of coming from any of the N time series in the sequence. If the time series have different lengths, they will contain different numbers of slices. Therefore, some particular slices may be sampled more often than others if they belong to shorter time series.

Parameters
  • target_series (Union[TimeSeries, Sequence[TimeSeries]]) – One or a sequence of target TimeSeries.

  • covariates (Union[TimeSeries, Sequence[TimeSeries], None]) – Optionally, one or a sequence of TimeSeries containing future-known covariates. If this parameter is set, the provided sequence must have the same length as that of target_series. Moreover, all covariates in the sequence must have a time span large enough to contain all the required slices. The joint slicing of the target and covariates is relying on the time axes of both series.

  • input_chunk_length (int) – The length of the emitted past series.

  • output_chunk_length (int) – The length of the emitted future series.

  • output_chunk_shift (int) – Optionally, the number of steps to shift the start of the output chunk into the future.

  • max_samples_per_ts (Optional[int]) – This is an upper bound on the number of tuples that can be produced per time series. It can be used in order to have an upper bound on the total size of the dataset and ensure proper sampling. If None, it will read all of the individual time series in advance (at dataset creation) to know their sizes, which might be expensive on big datasets. If some series turn out to have a length that would allow more than max_samples_per_ts, only the most recent max_samples_per_ts samples will be considered.

  • use_static_covariates (bool) – Whether to use/include static covariate data from input series.

class darts.utils.data.sequential_dataset.MixedCovariatesSequentialDataset(target_series, past_covariates=None, future_covariates=None, input_chunk_length=12, output_chunk_length=1, output_chunk_shift=0, max_samples_per_ts=None, use_static_covariates=True)[source]

Bases: MixedCovariatesTrainingDataset

A time series dataset containing tuples of (past_target, past_covariates, historic_future_covariates, future_covariates, static_covariates, future_target). The “past” series (incl historic_future_covariates) have length input_chunk_length and the “future” series have length output_chunk_length. The “future” series are immediately consecutive to the “past” series. The slicing of past and future covariates matches that of past and future targets, respectively. The slicing itself relies on time indexes to align the series if they have unequal lengths.

Each series must be long enough to contain at least one (input, output) pair; i.e., each series must have length at least input_chunk_length + output_chunk_length. If these conditions are not satisfied, an error will be raised when trying to access some of the splits.

The sampling is uniform over the number of time series; i.e., the i-th sample of this dataset has a probability 1/N of coming from any of the N time series in the sequence. If the time series have different lengths, they will contain different numbers of slices. Therefore, some particular slices may be sampled more often than others if they belong to shorter time series.

Parameters
  • target_series (Union[TimeSeries, Sequence[TimeSeries]]) – One or a sequence of target TimeSeries.

  • past_covariates (Union[TimeSeries, Sequence[TimeSeries], None]) – Optionally, one or a sequence of TimeSeries containing past-observed covariates. If this parameter is set, the provided sequence must have the same length as that of target_series. Moreover, all covariates in the sequence must have a time span large enough to contain all the required slices. The joint slicing of the target and covariates is relying on the time axes of both series.

  • future_covariates (Union[TimeSeries, Sequence[TimeSeries], None]) – Optionally, one or a sequence of TimeSeries containing future-known covariates. This has to follow the same constraints as past_covariates.

  • input_chunk_length (int) – The length of the emitted past series.

  • output_chunk_length (int) – The length of the emitted future series.

  • output_chunk_shift (int) – Optionally, the number of steps to shift the start of the output chunk into the future.

  • max_samples_per_ts (Optional[int]) – This is an upper bound on the number of tuples that can be produced per time series. It can be used in order to have an upper bound on the total size of the dataset and ensure proper sampling. If None, it will read all of the individual time series in advance (at dataset creation) to know their sizes, which might be expensive on big datasets. If some series turn out to have a length that would allow more than max_samples_per_ts, only the most recent max_samples_per_ts samples will be considered.

  • use_static_covariates (bool) – Whether to use/include static covariate data from input series.

class darts.utils.data.sequential_dataset.PastCovariatesSequentialDataset(target_series, covariates=None, input_chunk_length=12, output_chunk_length=1, output_chunk_shift=0, max_samples_per_ts=None, use_static_covariates=True)[source]

Bases: PastCovariatesTrainingDataset

A time series dataset containing tuples of (past_target, past_covariates, static_covariates, future_target). The “past” series have length input_chunk_length and the “future” series have length output_chunk_length. The “future” series are immediately consecutive to the “past” series. The slicing of past and future covariates matches that of past and future targets, respectively. The slicing itself relies on time indexes to align the series if they have unequal lengths.

Each series must be long enough to contain at least one (input, output) pair; i.e., each series must have length at least input_chunk_length + output_chunk_length. If these conditions are not satisfied, an error will be raised when trying to access some of the splits.

The sampling is uniform over the number of time series; i.e., the i-th sample of this dataset has a probability 1/N of coming from any of the N time series in the sequence. If the time series have different lengths, they will contain different numbers of slices. Therefore, some particular slices may be sampled more often than others if they belong to shorter time series.

Parameters
  • target_series (Union[TimeSeries, Sequence[TimeSeries]]) – One or a sequence of target TimeSeries.

  • covariates (Union[TimeSeries, Sequence[TimeSeries], None]) – Optionally, one or a sequence of TimeSeries containing past-observed covariates. If this parameter is set, the provided sequence must have the same length as that of target_series. Moreover, all covariates in the sequence must have a time span large enough to contain all the required slices. The joint slicing of the target and covariates is relying on the time axes of both series.

  • input_chunk_length (int) – The length of the emitted past series.

  • output_chunk_length (int) – The length of the emitted future series.

  • output_chunk_shift (int) – Optionally, the number of steps to shift the start of the output chunk into the future.

  • max_samples_per_ts (Optional[int]) – This is an upper bound on the number of tuples that can be produced per time series. It can be used in order to have an upper bound on the total size of the dataset and ensure proper sampling. If None, it will read all of the individual time series in advance (at dataset creation) to know their sizes, which might be expensive on big datasets. If some series turn out to have a length that would allow more than max_samples_per_ts, only the most recent max_samples_per_ts samples will be considered.

  • use_static_covariates (bool) – Whether to use/include static covariate data from input series.

class darts.utils.data.sequential_dataset.SplitCovariatesSequentialDataset(target_series, past_covariates=None, future_covariates=None, input_chunk_length=12, output_chunk_length=1, output_chunk_shift=0, max_samples_per_ts=None, use_static_covariates=True)[source]

Bases: SplitCovariatesTrainingDataset

A time series dataset containing tuples of (past_target, past_covariates, future_covariates, static_covariates, future_target). The “past” series have length input_chunk_length and the “future” series have length output_chunk_length. The “future” series are immediately consecutive to the “past” series. The slicing of past and future covariates matches that of past and future targets, respectively. The slicing itself relies on time indexes to align the series if they have unequal lengths.

Each series must be long enough to contain at least one (input, output) pair; i.e., each series must have length at least input_chunk_length + output_chunk_length. If these conditions are not satisfied, an error will be raised when trying to access some of the splits.

The sampling is uniform over the number of time series; i.e., the i-th sample of this dataset has a probability 1/N of coming from any of the N time series in the sequence. If the time series have different lengths, they will contain different numbers of slices. Therefore, some particular slices may be sampled more often than others if they belong to shorter time series.

Parameters
  • target_series (Union[TimeSeries, Sequence[TimeSeries]]) – One or a sequence of target TimeSeries.

  • past_covariates (Union[TimeSeries, Sequence[TimeSeries], None]) – Optionally, one or a sequence of TimeSeries containing past-observed covariates. If this parameter is set, the provided sequence must have the same length as that of target_series. Moreover, all covariates in the sequence must have a time span large enough to contain all the required slices. The joint slicing of the target and covariates is relying on the time axes of both series.

  • future_covariates (Union[TimeSeries, Sequence[TimeSeries], None]) – Optionally, one or a sequence of TimeSeries containing future-known covariates. This has to follow the same constraints as past_covariates.

  • input_chunk_length (int) – The length of the emitted past series.

  • output_chunk_length (int) – The length of the emitted future series.

  • output_chunk_shift (int) – Optionally, the number of steps to shift the start of the output chunk into the future.

  • max_samples_per_ts (Optional[int]) – This is an upper bound on the number of tuples that can be produced per time series. It can be used in order to have an upper bound on the total size of the dataset and ensure proper sampling. If None, it will read all of the individual time series in advance (at dataset creation) to know their sizes, which might be expensive on big datasets. If some series turn out to have a length that would allow more than max_samples_per_ts, only the most recent max_samples_per_ts samples will be considered.

  • use_static_covariates (bool) – Whether to use/include static covariate data from input series.