aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/estimator/canned/metric_keys.py
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/python/estimator/canned/metric_keys.py')
-rw-r--r--tensorflow/python/estimator/canned/metric_keys.py5
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/python/estimator/canned/metric_keys.py b/tensorflow/python/estimator/canned/metric_keys.py
index 4f7c849ba4..9d49240fea 100644
--- a/tensorflow/python/estimator/canned/metric_keys.py
+++ b/tensorflow/python/estimator/canned/metric_keys.py
@@ -47,3 +47,8 @@ class MetricKeys(object):
PROBABILITY_MEAN_AT_CLASS = 'probability_mean/class%d'
AUC_AT_CLASS = 'auc/class%d'
AUC_PR_AT_CLASS = 'auc_precision_recall/class%d'
+
+ # The following require a class name applied.
+ PROBABILITY_MEAN_AT_NAME = 'probability_mean/%s'
+ AUC_AT_NAME = 'auc/%s'
+ AUC_PR_AT_NAME = 'auc_precision_recall/%s'