make_train_valid_generators: Create train / validation data generators from a task and...

View source: R/DataGenerators.R

make_train_valid_generatorsR Documentation

Create train / validation data generators from a task and params

Description

Create train / validation data generators from a task and params

Usage

make_train_valid_generators(
  task,
  x_transform,
  y_transform,
  validation_split = 1/3,
  batch_size = 128L
)

Arguments

task

Task
An mlr3 Task.

x_transform

function
Function used to transform data to a keras input format for features.

y_transform

function
Function used to transform data to a keras input format for the response.

validation_split

numeric
Fraction of data to use for validation.

batch_size

integer
Batch_size for the generators.


mlr-org/mlr3keras documentation built on April 12, 2022, 11:35 a.m.