tf.contrib.learn.TensorFlowRNNRegressor.predict()

tf.contrib.learn.TensorFlowRNNRegressor.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.
doc_TensorFlow
2016-10-14 13:07:06
Comments
Leave a Comment

Please login to continue.