From 3789c366811fc9073df253ddf1b26d5b6e9be1a6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ho=C3=ABl=20Bagard?= <34478245+hoel-bagard@users.noreply.github.com> Date: Thu, 15 Jan 2026 16:34:40 +0900 Subject: [PATCH] `tensorflow`: add `tf.keras.metrics.MeanSquaredError` (#15282) --- 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 30096af2c..33b8a3fb6 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