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