tensorflow: add tf.keras.metrics.MeanSquaredError (#15282)

This commit is contained in:
Hoël Bagard
2026-01-15 16:34:40 +09:00
committed by GitHub
parent cd8b26b0ce
commit 3789c36681
@@ -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