tf.compat.v1.metrics.mean_cosine_distance
Вычисляет косинусное расстояние между метками и предсказаниями.
tf.compat.v1.metrics.mean_cosine_distance(
labels,
predictions,
dim,
weights=None,
metrics_collections=None,
updates_collections=None,
name=None
)
Функция mean_cosine_distance создаёт две локальные переменные, total и count, которые используются для вычисления среднего косинусного расстояния между predictions и labels. Это среднее взвешено по weights, и в конечном итоге возвращается как mean_distance, что представляет собой идемпотентную операцию, которая просто делит total на count.
Для оценки метрики по потоку данных функция создаёт операцию update_op, которая обновляет эти переменные и возвращает mean_distance.
Если weights является None, веса по умолчанию равны 1. Используйте веса 0 для маскирования значений.
| Аргументы | |
|---|---|
labels | A Tensor произвольной формы. |
predictions | A Tensor той же формы, что и labels. |
dim | Размерность, по которой вычисляется косинусное расстояние. |
weights | Необязательные Tensor, ранг которых равен либо 0, либо такому же, как у labels, и которые должны быть совместимы по трансляции с labels (т.е., все измерения должны быть либо 1, либо такими же, как соответствующее измерение labels). Кроме того, измерение dim должно быть 1. |
metrics_collections | Необязательный список коллекций, к которому должна быть добавлена переменная значения метрики. |
updates_collections | Необязательный список коллекций, к которому должны быть добавлены операции обновления метрики. |
name | Необязательное имя области переменных. |
| Возвращаемые значения | |
|---|---|
mean_distance | A Tensor, представляющий текущее среднее значение, значение total, делённое на count. |
update_op | Операция, которая увеличивает переменные total и count соответствующим образом. |
| Исключения | |
|---|---|
ValueError | Если у predictions и labels несовпадающие формы, или если weights не None и его форма не соответствует predictions, или если metrics_collections или updates_collections не являются списком или кортежем. |
RuntimeError | Если включено выполнение в режиме eager. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/compat/v1/metrics/mean_cosine_distance