predict_model
- Estimator.predict_model(data, batch_size=None)[source]
Predict all elements in
data.- Parameters
data (tf.data.Dataset | np.ndarray) – Data to predict, with first dimension the number of elements.
batch_size (int | None) – Batch size. If None, the training batch size is used.
- Returns
prediction
- Return type
Iterable