Use standard names for inference in TensorForest.

Change: 150094096

tensorflow/contrib/tensor_forest/BUILD

@@ -150,6 +150,7 @@ py_library(
     srcs = ["client/eval_metrics.py"],
     srcs_version = "PY2AND3",
     deps = [
+        "//tensorflow/contrib/learn:estimator_constants_py",
         "//tensorflow/contrib/losses:losses_py",
         "//tensorflow/contrib/metrics:metrics_py",
         "//tensorflow/python:array_ops",

tensorflow/contrib/tensor_forest/client/eval_metrics.py

@@ -20,14 +20,15 @@ from __future__ import print_function
 import numpy as np
 
 from tensorflow.contrib import losses
+from tensorflow.contrib.learn.python.learn.estimators import prediction_key
 from tensorflow.contrib.metrics.python.ops import metric_ops
 
 from tensorflow.python.ops import array_ops
 from tensorflow.python.ops import math_ops
 from tensorflow.python.ops import nn
 
-INFERENCE_PROB_NAME = 'inference'
-INFERENCE_PRED_NAME = 'predictions'
+INFERENCE_PROB_NAME = prediction_key.PredictionKey.CLASSES
+INFERENCE_PRED_NAME = prediction_key.PredictionKey.PROBABILITIES
 
 
 def _top_k_generator(k):