Description Usage Arguments Value See Also
The generator is run in parallel to the model, for efficiency. For instance, this allows you to do real-time data augmentation on images on CPU in parallel to training your model on GPU.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | fit_generator(
object,
generator,
steps_per_epoch,
epochs = 1,
verbose = getOption("keras.fit_verbose", default = 1),
callbacks = NULL,
view_metrics = getOption("keras.view_metrics", default = "auto"),
validation_data = NULL,
validation_steps = NULL,
class_weight = NULL,
max_queue_size = 10,
workers = 1,
initial_epoch = 0
)
|
object |
Keras model object |
generator |
A generator (e.g. like the one provided by
The output of the generator must be a list of one of these forms: - (inputs, targets) - (inputs, targets, sample_weights) This list (a single output of the generator) makes a single batch.
Therefore, all arrays in this list must have the same length (equal to
the size of this batch). Different batches may have different sizes.
For example, the last batch of the epoch is commonly smaller than the
others, if the size of the dataset is not divisible by the batch size.
The generator is expected to loop over its data indefinitely. An epoch
finishes when |
steps_per_epoch |
Total number of steps (batches of samples) to yield
from |
epochs |
Integer. Number of epochs to train the model.
An epoch is an iteration over the entire data provided, as defined by
|
verbose |
Verbosity mode (0 = silent, 1 = progress bar, 2 = one line per epoch). |
callbacks |
List of callbacks to apply during training. |
view_metrics |
View realtime plot of training metrics (by epoch). The
default ( |
validation_data |
this can be either:
|
validation_steps |
Only relevant if |
class_weight |
Optional named list mapping class indices (integer) to a weight (float) value, used for weighting the loss function (during training only). This can be useful to tell the model to "pay more attention" to samples from an under-represented class. |
max_queue_size |
Maximum size for the generator queue. If unspecified,
|
workers |
Maximum number of threads to use for parallel processing. Note that
parallel processing will only be performed for native Keras generators (e.g.
|
initial_epoch |
epoch at which to start training (useful for resuming a previous training run) |
Training history object (invisibly)
Other model functions:
compile.keras.engine.training.Model()
,
evaluate.keras.engine.training.Model()
,
evaluate_generator()
,
fit.keras.engine.training.Model()
,
get_config()
,
get_layer()
,
keras_model_sequential()
,
keras_model()
,
multi_gpu_model()
,
pop_layer()
,
predict.keras.engine.training.Model()
,
predict_generator()
,
predict_on_batch()
,
predict_proba()
,
summary.keras.engine.training.Model()
,
train_on_batch()
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.