Source code for ignite.metrics.root_mean_squared_error

from __future__ import division
import math

from ignite.metrics.mean_squared_error import MeanSquaredError

[docs]class RootMeanSquaredError(MeanSquaredError): """ Calculates the root mean squared error. - `update` must receive output of the form (y_pred, y). """ def compute(self): mse = super(RootMeanSquaredError, self).compute() return math.sqrt(mse)

© Copyright 2022, PyTorch-Ignite Contributors. Last updated on 08/17/2022, 4:22:51 PM.

Built with Sphinx using a theme provided by Read the Docs.