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 | Tensor произвольной формы. |
predictions | Tensor той же формы, что и labels. |
dim | Измерение, вдоль которого вычисляется косинусное расстояние. |
weights | Необязательный Tensor, ранг которого равен 0 или совпадает с рангом labels и который должен поддерживать широковещательное приведение к labels (то есть каждое измерение должно быть либо 1, либо совпадать с соответствующим измерением labels). Кроме того, измерение dim должно быть 1. |
metrics_collections | Необязательный список коллекций, в которые следует добавить переменную значения метрики. |
updates_collections | Необязательный список коллекций, в которые следует добавить операции обновления метрики. |
name | Необязательное имя variable_scope. |
| Возвращает | |
|---|---|
mean_distance | Tensor, представляющий текущее среднее значение: значение total, деленное на count. |
update_op | Операция, которая соответствующим образом увеличивает значения переменных total и count. |
| Вызывает исключения | |
|---|---|
ValueError | Если формы predictions и labels не совпадают, или если weights не является None и его форма не совпадает с predictions, или если metrics_collections либо updates_collections не является списком или кортежем. |
RuntimeError | Если включено энергичное выполнение. |
© 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/api_docs/python/tf/compat/v1/metrics/mean_cosine_distance