mirror of
https://github.com/davidhalter/typeshed.git
synced 2026-06-25 18:24:03 +08:00
tensorflow: add tf.keras.metrics.MeanSquaredError (#15282)
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user