contrib.metrics.streaming_sparse_average_precision_at_top_k
tf.contrib.metrics.streaming_sparse_average_precision_at_top_k
tf.contrib.metrics.streaming_sparse_average_precision_at_top_k
streaming_sparse_average_precision_at_top_k( top_k_predictions, labels, weights=None, metrics_collections=None, updates_collections=None, name=None )
Defined in tensorflow/contrib/metrics/python/ops/metric_ops.py
.
Computes average precision@k of predictions with respect to sparse labels.
streaming_sparse_average_precision_at_top_k
creates two local variables, average_precision_at_<k>/total
and average_precision_at_<k>/max
, that are used to compute the frequency. This frequency is ultimately returned as average_precision_at_<k>
: an idempotent operation that simply divides average_precision_at_<k>/total
by average_precision_at_<k>/max
.
For estimation of the metric over a stream of data, the function creates an update_op
operation that updates these variables and returns the precision_at_<k>
. Set operations applied to top_k
and labels
calculate the true positives and false positives weighted by weights
. Then update_op
increments true_positive_at_<k>
and false_positive_at_<k>
using these values.
If weights
is None
, weights default to 1. Use weights of 0 to mask values.
Args:
-
top_k_predictions
: IntegerTensor
with shape [D1, ... DN, k] where N >= 1. Commonly, N=1 andpredictions_idx
has shape [batch size, k]. The final dimension must be set and contains the topk
predicted class indices. [D1, ... DN] must matchlabels
. Values should be in range [0, num_classes). -
labels
:int64
Tensor
orSparseTensor
with shape [D1, ... DN, num_labels] or [D1, ... DN], where the latter implies num_labels=1. N >= 1 and num_labels is the number of target classes for the associated prediction. Commonly, N=1 andlabels
has shape [batch_size, num_labels]. [D1, ... DN] must matchtop_k_predictions
. Values should be in range [0, num_classes). -
weights
:Tensor
whose rank is either 0, or n-1, where n is the rank oflabels
. If the latter, it must be broadcastable tolabels
(i.e., all dimensions must be either1
, or the same as the correspondinglabels
dimension). -
metrics_collections
: An optional list of collections that values should be added to. -
updates_collections
: An optional list of collections that updates should be added to. -
name
: Name of new update operation, and namespace for other dependent ops.
Returns:
-
mean_average_precision
: Scalarfloat64
Tensor
with the mean average precision values. -
update
:Operation
that increments variables appropriately, and whose value matchesmetric
.
Raises:
-
ValueError
: if the last dimension of top_k_predictions is not set.
© 2017 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_sparse_average_precision_at_top_k