Skip to contents

Fit VAEReg

Usage

vae_fit(
  model,
  X,
  y,
  X_val = NULL,
  y_val = NULL,
  epochs = 80L,
  batch_size = 64L,
  patience = 10L,
  verbose = 0L
)

Arguments

model

Python VAEReg object from vae_build()

X

matrix (n x p)

y

numeric vector (n)

X_val

optional matrix

y_val

optional numeric vector

epochs

integer

batch_size

integer

patience

integer for early stopping (only if validation provided)

verbose

0/1/2

Value

The fitted model, invisibly.

Examples

if (FALSE) { # \dontrun{
  vae_configure()
  X <- matrix(rnorm(200), nrow = 20, ncol = 10)
  y <- rnorm(20)
  model <- vae_build(input_dim = ncol(X))
  vae_fit(model, X, y, epochs = 5L)
} # }