tf.contrib.learn.TensorFlowEstimator.predict()

tf.contrib.learn.TensorFlowEstimator.predict(x, axis=1, batch_size=None)

Predict class or regression for x.

For a classification model, the predicted class for each sample in x is returned. For a regression model, the predicted value based on x is returned.

Args:

• x: array-like matrix, [n_samples, n_features...] or iterator.
• axis: Which axis to argmax for classification. By default axis 1 (next after batch) is used. Use 2 for sequence predictions.
• batch_size: If test set is too big, use batch size to split it into mini batches. By default the batch_size member variable is used.

Returns:

• y: array of shape [n_samples]. The predicted classes or predicted value.