diff --git a/stubs/tensorflow/tensorflow/keras/metrics.pyi b/stubs/tensorflow/tensorflow/keras/metrics.pyi index 30096af2cafa..33b8a3fb623d 100644 --- a/stubs/tensorflow/tensorflow/keras/metrics.pyi +++ b/stubs/tensorflow/tensorflow/keras/metrics.pyi @@ -109,6 +109,9 @@ class SparseTopKCategoricalAccuracy(MeanMetricWrapper): self, k: int = 5, name: str | None = "sparse_top_k_categorical_accuracy", dtype: DTypeLike | None = None ) -> None: ... +class MeanSquaredError(MeanMetricWrapper): + def __init__(self, name: str | None = "mean_squared_error", dtype: DTypeLike | None = None) -> None: ... + # TODO: Actually tensorflow.python.keras.utils.metrics_utils.Reduction, but that module # is currently missing from the stub. @type_check_only