tf.contrib.metrics.streaming_dynamic_auc( labels, predictions, curve='ROC', metrics_collections=(), updates_collections=(), name=None )
Defined in tensorflow/contrib/metrics/python/ops/metric_ops.py
.
Computes the apporixmate AUC by a Riemann sum with data-derived thresholds.
USAGE NOTE: this approach requires storing all of the predictions and labels for a single evaluation in memory, so it may not be usable when the evaluation batch size and/or the number of evaluation steps is very large.
Computes the area under the ROC or PR curve using each prediction as a threshold. This has the advantage of being resilient to the distribution of predictions by aggregating across batches, accumulating labels and predictions and performing the final calculation using all of the concatenated values.
labels
: A Tensor
of ground truth labels with the same shape as labels
and with values of 0 or 1 whose values are castable to int64
.predictions
: A Tensor
of predictions whose values are castable to float64
. Will be flattened into a 1-D Tensor
.curve
: The name of the curve for which to compute AUC, 'ROC' for the Receiving Operating Characteristic or 'PR' for the Precision-Recall curve.metrics_collections
: An optional iterable of collections that auc
should be added to.updates_collections
: An optional iterable of collections that update_op
should be added to.name
: An optional name for the variable_scope that contains the metric variables.auc
: A scalar Tensor
containing the current area-under-curve value.update_op
: An operation that concatenates the input labels and predictions to the accumulated values.ValueError
: If labels
and predictions
have mismatched shapes or if curve
isn't a recognized curve type.
© 2018 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/contrib/metrics/streaming_dynamic_auc