prepare_dl: Prepare dataloders

View source: R/prepare-data.R

prepare_dlR Documentation

Prepare dataloders

Description

Prepare dataloders

Usage

prepare_dl(
  data,
  formula,
  index,
  timesteps,
  horizon,
  categorical = NULL,
  validation = NULL,
  sample_frac = 1,
  batch_size,
  shuffle,
  jump,
  parsed_formula = NULL,
  flatten = FALSE,
  ...
)

Arguments

data

(data.frame) An input data.frame object with. For now only single data frames are handled with no categorical features.

formula

(formula) A formula describing, how to use the data

index

(character) The index column name.

timesteps

(integer) The time series chunk length.

horizon

(integer) Forecast horizon.

categorical

(character) Categorical features.

validation

(data.frame or numeric) Validation dataset or percent of TODO.

sample_frac

(numeric) Sample a fraction of rows (default: 1, i.e.: all the rows).

batch_size

(integer) Batch size.

shuffle

(logical) A dataloader argument - shuffle rows or not?

jump

(integer) Input window shift.


krzjoa/torchts documentation built on June 24, 2022, 5:30 a.m.