fit-VaeModel-tf_dataset-method: fit

fit,VaeModel,tf_dataset-methodR Documentation

fit

Description

Fit a VaeModel

Usage

## S4 method for signature 'VaeModel,tf_dataset'
fit(
  model,
  x,
  batch_size = 128L,
  epochs = 100L,
  learning_rate = 0.01,
  compile = TRUE,
  beta = 1
)

Arguments

model

a VaeModel object, initialized by ‘new(’VaeModel', model = VaeModel(...))'

x

a tf_dataset object

batch_size

Batch size (default: 128L)

epochs

Number of training epochs (default: 100L)

learning_rate

Learning rate (default: 1e-2)

compile

Whether or not compile the tensorflow model (default: TRUE)

beta

Beta sequences (default: 1)

Value

a VaeModel


gongx030/seatac documentation built on April 15, 2023, 5:53 a.m.