Skip to content

Commit f624b4c

Browse files
committed
feat: add tf.keras.metrics.MeanSquaredError
1 parent cd8b26b commit f624b4c

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

stubs/tensorflow/tensorflow/keras/metrics.pyi

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -109,6 +109,9 @@ class SparseTopKCategoricalAccuracy(MeanMetricWrapper):
109109
self, k: int = 5, name: str | None = "sparse_top_k_categorical_accuracy", dtype: DTypeLike | None = None
110110
) -> None: ...
111111

112+
class MeanSquaredError(MeanMetricWrapper):
113+
def __init__(self, name: str | None = "mean_squared_error", dtype: DTypeLike | None = None) -> None: ...
114+
112115
# TODO: Actually tensorflow.python.keras.utils.metrics_utils.Reduction, but that module
113116
# is currently missing from the stub.
114117
@type_check_only

0 commit comments

Comments
 (0)