The Sequential model API
To get started, read this guide to the Keras Sequential model.
Useful attributes of Model
model.layers
is a list of the layers added to the model.
Sequential model methods
compile
compile(self, optimizer, loss, metrics=None, sample_weight_mode=None, weighted_metrics=None)
Configures the learning process.
Arguments
- optimizer: str (name of optimizer) or optimizer object. See optimizers.
- loss: str (name of objective function) or objective function. See losses.
- metrics: list of metrics to be evaluated by the model
during training and testing.
Typically you will use
metrics=['accuracy']
. See metrics. - sample_weight_mode: if you need to do timestep-wise sample weighting (2D weights), set this to "temporal". "None" defaults to sample-wise weights (1D).
- weighted_metrics: list of metrics to be evaluated and weighted by sample_weight or class_weight during training and testing
- **kwargs: for Theano/CNTK backends, these are passed into
K.function. When using the TensorFlow backend, these are
passed into
tf.Session.run
.
Example
model = Sequential()
model.add(Dense(32, input_shape=(500,)))
model.add(Dense(10, activation='softmax'))
model.compile(optimizer='rmsprop',
loss='categorical_crossentropy',
metrics=['accuracy'])
fit
fit(self, x, y, batch_size=32, epochs=10, verbose=1, callbacks=None, validation_split=0.0, validation_data=None, shuffle=True, class_weight=None, sample_weight=None, initial_epoch=0)
Trains the model for a fixed number of epochs.
Arguments
- x: input data, as a Numpy array or list of Numpy arrays (if the model has multiple inputs).
- y: labels, as a Numpy array.
- batch_size: integer. Number of samples per gradient update.
- epochs: integer, the number of epochs to train the model.
- verbose: 0 for no logging to stdout, 1 for progress bar logging, 2 for one log line per epoch.
- callbacks: list of
keras.callbacks.Callback
instances. List of callbacks to apply during training. See callbacks. - validation_split: float (0. < x < 1). Fraction of the data to use as held-out validation data.
- validation_data: tuple (x_val, y_val) or tuple (x_val, y_val, val_sample_weights) to be used as held-out validation data. Will override validation_split.
- shuffle: boolean or str (for 'batch'). Whether to shuffle the samples at each epoch. 'batch' is a special option for dealing with the limitations of HDF5 data; it shuffles in batch-sized chunks.
- class_weight: dictionary mapping classes to a weight value, used for scaling the loss function (during training only).
- sample_weight: Numpy array of weights for the training samples, used for scaling the loss function (during training only). You can either pass a flat (1D) Numpy array with the same length as the input samples (1:1 mapping between weights and samples), or in the case of temporal data, you can pass a 2D array with shape (samples, sequence_length), to apply a different weight to every timestep of every sample. In this case you should make sure to specify sample_weight_mode="temporal" in compile().
- initial_epoch: epoch at which to start training (useful for resuming a previous training run)
Returns
A History
object. Its History.history
attribute is
a record of training loss values and metrics values
at successive epochs, as well as validation loss values
and validation metrics values (if applicable).
Raises
- RuntimeError: if the model was never compiled.
evaluate
evaluate(self, x, y, batch_size=32, verbose=1, sample_weight=None)
Computes the loss on some input data, batch by batch.
Arguments
- x: input data, as a Numpy array or list of Numpy arrays (if the model has multiple inputs).
- y: labels, as a Numpy array.
- batch_size: integer. Number of samples per gradient update.
- verbose: verbosity mode, 0 or 1.
- sample_weight: sample weights, as a Numpy array.
Returns
Scalar test loss (if the model has no metrics)
or list of scalars (if the model computes other metrics).
The attribute model.metrics_names
will give you
the display labels for the scalar outputs.
Raises
- RuntimeError: if the model was never compiled.
predict
predict(self, x, batch_size=32, verbose=0)
Generates output predictions for the input samples.
The input samples are processed batch by batch.
Arguments
- x: the input data, as a Numpy array.
- batch_size: integer.
- verbose: verbosity mode, 0 or 1.
Returns
A Numpy array of predictions.
train_on_batch
train_on_batch(self, x, y, class_weight=None, sample_weight=None)
Single gradient update over one batch of samples.
Arguments
- x: input data, as a Numpy array or list of Numpy arrays (if the model has multiple inputs).
- y: labels, as a Numpy array.
- class_weight: dictionary mapping classes to a weight value, used for scaling the loss function (during training only).
- sample_weight: sample weights, as a Numpy array.
Returns
Scalar training loss (if the model has no metrics)
or list of scalars (if the model computes other metrics).
The attribute model.metrics_names
will give you
the display labels for the scalar outputs.
Raises
- RuntimeError: if the model was never compiled.
test_on_batch
test_on_batch(self, x, y, sample_weight=None)
Evaluates the model over a single batch of samples.
Arguments
- x: input data, as a Numpy array or list of Numpy arrays (if the model has multiple inputs).
- y: labels, as a Numpy array.
- sample_weight: sample weights, as a Numpy array.
Returns
Scalar test loss (if the model has no metrics)
or list of scalars (if the model computes other metrics).
The attribute model.metrics_names
will give you
the display labels for the scalar outputs.
Raises
- RuntimeError: if the model was never compiled.
predict_on_batch
predict_on_batch(self, x)
Returns predictions for a single batch of samples.
Arguments
- x: input data, as a Numpy array or list of Numpy arrays (if the model has multiple inputs).
Returns
A Numpy array of predictions.
fit_generator
fit_generator(self, generator, steps_per_epoch, epochs=1, verbose=1, callbacks=None, validation_data=None, validation_steps=None, class_weight=None, max_queue_size=10, workers=1, use_multiprocessing=False, initial_epoch=0)
Fits the model on data generated batch-by-batch by a Python generator.
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.
Arguments
- generator: A generator.
The output of the generator must be either
- a tuple (inputs, targets)
- a tuple (inputs, targets, sample_weights).
All arrays should contain the same number of samples.
The generator is expected to loop over its data
indefinitely. An epoch finishes when
steps_per_epoch
batches have been seen by the model.
- steps_per_epoch: Total number of steps (batches of samples)
to yield from
generator
before declaring one epoch finished and starting the next epoch. It should typically be equal to the number of unique samples of your dataset divided by the batch size. - epochs: Integer, total number of iterations on the data.
- verbose: Verbosity mode, 0, 1, or 2.
- callbacks: List of callbacks to be called during training.
- validation_data: This can be either
- A generator for the validation data
- A tuple (inputs, targets)
- A tuple (inputs, targets, sample_weights).
- validation_steps: Only relevant if
validation_data
is a generator. Number of steps to yield from validation generator at the end of every epoch. It should typically be equal to the number of unique samples of your validation dataset divided by the batch size. - class_weight: Dictionary mapping class indices to a weight for the class.
- max_queue_size: Maximum size for the generator queue
- workers: Maximum number of processes to spin up
- use_multiprocessing: if True, use process based threading. Note that because this implementation relies on multiprocessing, you should not pass non picklable arguments to the generator as they can't be passed easily to children processes.
- initial_epoch: Epoch at which to start training (useful for resuming a previous training run)
Returns
A History
object.
Raises
- RuntimeError: if the model was never compiled.
Example
def generate_arrays_from_file(path):
while 1:
f = open(path)
for line in f:
# create Numpy arrays of input data
# and labels, from each line in the file
x, y = process_line(line)
yield (x, y)
f.close()
model.fit_generator(generate_arrays_from_file('/my_file.txt'),
steps_per_epoch=1000, epochs=10)
evaluate_generator
evaluate_generator(self, generator, steps, max_queue_size=10, workers=1, use_multiprocessing=False)
Evaluates the model on a data generator.
The generator should return the same kind of data
as accepted by test_on_batch
.
Arguments
- generator: Generator yielding tuples (inputs, targets) or (inputs, targets, sample_weights)
- steps: Total number of steps (batches of samples)
to yield from
generator
before stopping. - max_queue_size: maximum size for the generator queue
- workers: maximum number of processes to spin up
- use_multiprocessing: if True, use process based threading. Note that because this implementation relies on multiprocessing, you should not pass non picklable arguments to the generator as they can't be passed easily to children processes.
Returns
Scalar test loss (if the model has no metrics)
or list of scalars (if the model computes other metrics).
The attribute model.metrics_names
will give you
the display labels for the scalar outputs.
Raises
- RuntimeError: if the model was never compiled.
predict_generator
predict_generator(self, generator, steps, max_queue_size=10, workers=1, use_multiprocessing=False, verbose=0)
Generates predictions for the input samples from a data generator.
The generator should return the same kind of data as accepted by
predict_on_batch
.
Arguments
- generator: generator yielding batches of input samples.
- steps: Total number of steps (batches of samples)
to yield from
generator
before stopping. - max_queue_size: maximum size for the generator queue
- workers: maximum number of processes to spin up
- use_multiprocessing: if True, use process based threading. Note that because this implementation relies on multiprocessing, you should not pass non picklable arguments to the generator as they can't be passed easily to children processes.
- verbose: verbosity mode, 0 or 1.
Returns
A Numpy array of predictions.
get_layer
get_layer(self, name=None, index=None)
Retrieve a layer that is part of the model.
Returns a layer based on either its name (unique) or its index in the graph. Indices are based on order of horizontal graph traversal (bottom-up).
Arguments
- name: string, name of layer.
- index: integer, index of layer.
Returns
A layer instance.