tf.compat.v1.metrics.mean_cosine_distance

Stay organized with collections Save and categorize content based on your preferences.

Computes the cosine distance between the labels and predictions.

tf.compat.v1.metrics.mean_cosine_distance(
    labels,
    predictions,
    dim,
    weights=None,
    metrics_collections=None,
    updates_collections=None,
    name=None
)

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 anupdate_op operation that updates these variables and returns themean_distance.

If weights is None, weights default to 1. Use weights of 0 to mask values.

Args
labels	A Tensor of arbitrary shape.
predictions	A Tensor of the same shape as labels.
dim	The dimension along which the cosine distance is computed.
weights	Optional Tensor whose rank is either 0, or the same rank aslabels, and must be broadcastable to labels (i.e., all dimensions must be either 1, or the same as the corresponding labels dimension). Also, dimension dim must be 1.
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	A Tensor representing the current mean, the value oftotal divided by count.
update_op	An operation that increments the total and count variables appropriately.

Raises
ValueError	If predictions and labels have mismatched shapes, or ifweights is not None and its shape doesn't match predictions, or if either metrics_collections or updates_collections are not a list or tuple.
RuntimeError	If eager execution is enabled.

tf.compat.v1.metrics.mean_cosine_distance | TensorFlow v2.16.1 (original) (raw)