tf.metrics.mean_cosine_distance(
labels,
predictions,
dim,
weights=None,
metrics_collections=None,
updates_collections=None,
name=None
)
Defined in tensorflow/python/ops/metrics_impl.py
.
Computes the cosine distance between the labels and predictions.
The mean_cosine_distance
function creates two local variables,
total
and count
that are used to compute the average cosine distance
between predictions
and labels
. This average is weighted by weights
,
and it is ultimately returned as mean_distance
, which is an idempotent
operation that simply divides total
by count
.
For estimation of the metric over a stream of data, the function creates an
update_op
operation that updates these variables and returns the
mean_distance
.
If weights
is None
, weights default to 1. Use weights of 0 to mask values.
Args:
labels
: ATensor
of arbitrary shape.predictions
: ATensor
of the same shape aslabels
.dim
: The dimension along which the cosine distance is computed.weights
: OptionalTensor
whose rank is either 0, or the same rank aslabels
, and must be broadcastable tolabels
(i.e., all dimensions must be either1
, or the same as the correspondinglabels
dimension). Also, dimensiondim
must be1
.metrics_collections
: An optional list of collections that the metric value variable should be added to.updates_collections
: An optional list of collections that the metric update ops should be added to.name
: An optional variable_scope name.
Returns:
mean_distance
: ATensor
representing the current mean, the value oftotal
divided bycount
.update_op
: An operation that increments thetotal
andcount
variables appropriately.
Raises:
ValueError
: Ifpredictions
andlabels
have mismatched shapes, or ifweights
is notNone
and its shape doesn't matchpredictions
, or if eithermetrics_collections
orupdates_collections
are not a list or tuple.RuntimeError
: If eager execution is enabled.