Source code for bayesflow.metrics.root_mean_squard_error

from functools import partial
import keras

from bayesflow.utils.serialization import serializable
from .functional import root_mean_squared_error




@serializable("bayesflow.metrics")
class RootMeanSquaredError(keras.metrics.MeanMetricWrapper):
    def __init__(self, name="root_mean_squared_error", dtype=None, **kwargs):
        fn = partial(root_mean_squared_error, **kwargs)
        super().__init__(fn, name=name, dtype=dtype)