From f624b4c802c488f26ad1aa177daa32af6f947573 Mon Sep 17 00:00:00 2001 From: Hoel Bagard Date: Thu, 15 Jan 2026 10:17:19 +0900 Subject: [PATCH] feat: add tf.keras.metrics.MeanSquaredError --- stubs/tensorflow/tensorflow/keras/metrics.pyi | 3 +++ 1 file changed, 3 insertions(+) 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