Type RegressionHead
Namespace tensorflow_estimator.python.estimator.api.estimator
Parent Head
Interfaces IRegressionHead
Methods
Properties
Public instance methods
IDictionary<string, Tensor> predictions(IEnumerable<IGraphNodeBase> logits)
IDictionary<string, Tensor> predictions(ValueTuple<PythonClassContainer, PythonClassContainer> logits)
IDictionary<string, Tensor> predictions(IndexedSlices logits)
IDictionary<string, Tensor> predictions(IGraphNodeBase logits)
IDictionary<string, Tensor> predictions(object logits)
IDictionary<string, Tensor> predictions(string logits)
object predictions_dyn(object logits)
Add operations to compute predictions by the model. If logistic_loss is being used, predicted probabilities are returned.
If poisson_loss is being used, predictions are exponentiated.
Otherwise, (raw) linear predictions (w*x) are returned.
Returns
-
object
- An Operation that computes the predictions for examples.