View source: R/custom_train_on_batch.R
Fits model using data generator.
1 2 3 4 5 6 7 8 9 10 | custom_fit_generator(
metric_names,
model,
generator,
epochs,
steps_per_epoch,
validation_generator = NULL,
validation_steps_per_epoch = NULL,
model_filepath = NULL
)
|
metric_names |
Metric names. |
model |
Model. |
generator |
Data generator. |
epochs |
Number of epochs. |
steps_per_epoch |
Steps per epoch. |
validation_generator |
Validation data generator. |
validation_steps_per_epoch |
Validation steps per epoch. |
model_filepath |
Path to save the model. |
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.