Returns predictions for a single batch of samples.

predict_on_batch(object, x)

Arguments

object

Keras model object

x

Input data (vector, matrix, or array). You can also pass a tfdataset or a generator returning a list with (inputs, targets) or (inputs, targets, sample_weights).

Value

array of predictions.

See also