Add classification_signature_fn_with_probabilities and make it the default for LinearClassifier.

Change: 133795652

Author: tensorflower-gardener

tensorflow/contrib/learn/python/learn/estimators/linear.py

@@ -571,7 +571,8 @@ def default_input_fn(unused_estimator, examples):
         input_fn=input_fn or default_input_fn,
         input_feature_key=input_feature_key,
         use_deprecated_input_fn=use_deprecated_input_fn,
-        signature_fn=signature_fn or export.classification_signature_fn,
+        signature_fn=(
+            signature_fn or export.classification_signature_fn_with_prob),
         prediction_key=_PROBABILITIES,
         default_batch_size=default_batch_size,
         exports_to_keep=exports_to_keep)

tensorflow/contrib/learn/python/learn/utils/export.py

@@ -119,6 +119,23 @@ def classification_signature_fn(examples, unused_features, predictions):
   return default_signature, {}
 
 
+def classification_signature_fn_with_prob(
+    examples, unused_features, predictions):
+  """Classification signature from given examples and predicted probabilities.
+
+  Args:
+    examples: `Tensor`.
+    unused_features: `dict` of `Tensor`s.
+    predictions: `Tensor` of predicted probabilities.
+
+  Returns:
+    Tuple of default classification signature and empty named signatures.
+  """
+  default_signature = exporter.classification_signature(
+      examples, scores_tensor=predictions)
+  return default_signature, {}
+
+
 def regression_signature_fn(examples, unused_features, predictions):
   """Creates regression signature from given examples and predictions.