训练数据集基类¶
- class darts.utils.data.training_dataset.DualCovariatesTrainingDataset[source]¶
- 基础类: - TrainingDataset,- ABC- DualCovariatesTorchModel 训练数据集的抽象类。它包含由 (past_target, historic_future_covariates, future_covariates, static_covariates, future_target) 组成的 np.ndarray。协变量是可选的,可以为 None。 
- class darts.utils.data.training_dataset.FutureCovariatesTrainingDataset[source]¶
- 基础类: - TrainingDataset,- ABC- FutureCovariatesTorchModel 训练数据集的抽象类。它包含由 (past_target, future_covariate, static_covariates, future_target) 组成的 np.ndarray。协变量是可选的,可以为 None。 
- class darts.utils.data.training_dataset.MixedCovariatesTrainingDataset[source]¶
- 基础类: - TrainingDataset,- ABC- MixedCovariatesTorchModel 训练数据集的抽象类。它包含由 (past_target, past_covariates, historic_future_covariates, future_covariates, static_covariates, future_target) 组成的 np.ndarray。协变量是可选的,可以为 None。 
- class darts.utils.data.training_dataset.PastCovariatesTrainingDataset[source]¶
- 基础类: - TrainingDataset,- ABC- PastCovariatesTorchModel 训练数据集的抽象类。它包含由 (past_target, past_covariate, static_covariates, future_target) 组成的 np.ndarray。协变量是可选的,可以为 None。 
- class darts.utils.data.training_dataset.SplitCovariatesTrainingDataset[source]¶
- 基础类: - TrainingDataset,- ABC- SplitCovariatesTorchModel 训练数据集的抽象类。它包含由 (past_target, past_covariates, future_covariates, static_covariates, future_target) 组成的 np.ndarray。协变量是可选的,可以为 None。 
- class darts.utils.data.training_dataset.TrainingDataset[source]¶
- 基础类: - ABC,- Dataset- Darts 中所有 torch 模型训练数据集的超类。这些包括 - “PastCovariates” 数据集(用于 PastCovariatesTorchModel):包含 (past_target,
- past_covariates, static_covariates, future_target) 
 
- “FutureCovariates” 数据集(用于 FutureCovariatesTorchModel):包含 (past_target,
- future_covariates, static_covariates, future_target) 
 
- “DualCovariates” 数据集(用于 DualCovariatesTorchModel):包含 (past_target,
- historic_future_covariates, future_covariates, static_covariates, future_target) 
 
- “MixedCovariates” 数据集(用于 MixedCovariatesTorchModel):包含 (past_target,
- past_covariates, historic_future_covariates, future_covariates, static_covariates, future_target) 
 
- “SplitCovariates” 数据集(用于 SplitCovariatesTorchModel):包含 (past_target,
- past_covariates, future_covariates, static_covariates, future_target) 
 
 - 协变量是可选的,可以为 None。 - 这旨在用于训练(或验证),除了 future_target 之外的所有数据都表示模型输入(future_target 是模型训练预测的输出)。 - Darts 的 TorchForecastingModel 可以使用 fit_from_dataset() 方法从正确类型的 TrainingDataset 实例中拟合。 - TrainingDataset 继承了 torch Dataset;这意味着实现必须提供 __getitem__() 方法。 - 它包含 np.ndarray(而不是 TimeSeries),因为训练只需要值,因此在切片时,通过仅返回 TimeSeries 底层数据的 numpy 视图,我们可以获得显著的性能提升。