tf.compat.v1.metrics.root_mean_squared_error
Вычисляет среднеквадратическую ошибку (RMSE) между метками и прогнозами.
tf.compat.v1.metrics.root_mean_squared_error(
labels, predictions, weights=None, metrics_collections=None,
updates_collections=None, name=None
)
Функция root_mean_squared_error создаёт две локальные переменные, total и count, используемые для вычисления среднеквадратической ошибки. Это среднее взвешено по weights, и в итоге возвращается как root_mean_squared_error: идемпотентная операция, которая извлекает квадратный корень из отношения total к count.
Для оценки метрики по потоку данных функция создаёт операцию update_op, которая обновляет эти переменные и возвращает root_mean_squared_error. Внутренне операция squared_error вычисляет поэлементный квадрат разницы между predictions и labels. Затем update_op увеличивает total на сумму произведений weights и squared_error, и увеличивает count на сумму weights.
Если weights равно None, веса по умолчанию равны 1. Используйте веса 0 для маскирования значений.
| Аргументы | |
|---|---|
labels | A Tensor той же формы, что и predictions. |
predictions | A Tensor произвольной формы. |
weights | Необязательные Tensor с рангом 0 или таким же, как у labels, и должны быть совместимы с labels (т.е., все измерения должны быть либо 1, либо такими же, как соответствующее измерение labels). |
metrics_collections | Необязательный список коллекций, к которым root_mean_squared_error должен быть добавлен. |
updates_collections | Необязательный список коллекций, к которым update_op должен быть добавлен. |
name | Необязательное имя области переменных. |
| Возвращаемые значения | |
|---|---|
root_mean_squared_error | A Tensor, представляющий текущее среднее значение, значение total, делённое на count. |
update_op | Операция, которая соответствующим образом увеличивает переменные total и count и значение которой соответствует root_mean_squared_error. |
| Исключения | |
|---|---|
ValueError | Если predictions и labels имеют несовпадающие формы, или если weights не None и его форма не совпадает с predictions, или если metrics_collections или updates_collections не являются списком или кортежем. |
RuntimeError | Если включена немедленная обработка. |
© 2020 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/versions/r2.3/api_docs/python/tf/compat/v1/metrics/root_mean_squared_error
