make_generator: creates a generator for training

View source: R/generator.R

make_generatorR Documentation

creates a generator for training

Description

creates a generator for training

Usage

make_generator(
  data_image,
  data_tab,
  batch_size,
  target_size,
  color_mode,
  is_trafo = FALSE,
  x_col,
  shuffle = TRUE,
  seed = 42L
)

Arguments

data_image

data.frame with image location as character

data_tab

tabular data as given by prepare_data

batch_size

integer

target_size

size of the image without sample and colour channel

color_mode

character; 'rgb' or 'grayscale'

is_trafo

logical; indicating whether model is a trafo model

x_col

name of image column in data_image

shuffle

logical for shuffling data

seed

seed for shuffling in generators


davidruegamer/deepregression documentation built on May 30, 2022, 6:21 p.m.