predict_gen {deepregression}

Generator function for deepregression objects

Description

Generator function for deepregression objects

Usage

predict_gen(
  object,
  newdata = NULL,
  batch_size = NULL,
  apply_fun = tfd_mean,
  convert_fun = as.matrix,
  ret_dist = FALSE
)

Arguments

object

deepregression model;

newdata

data.frame or list; for (optional) new data

batch_size

integer; NULL will use the default (20)

apply_fun

see ?predict.deepregression

convert_fun

see ?predict.deepregression

ret_dist

logical; whether to return the whole distribution or only the (mean) prediction

Value

matrix or list of distributions