提交 95138da6 编辑于 作者: Yu Xiao's avatar Yu Xiao 提交者: tensorflow-copybara
浏览文件

Change the docstring of `federated_aggregate_keras_metric` as it can take...

Change the docstring of `federated_aggregate_keras_metric` as it can take either metrics or metric constructors now.

PiperOrigin-RevId: 413026905
上级 f5ef56e0
...@@ -201,8 +201,10 @@ def federated_aggregate_keras_metric( ...@@ -201,8 +201,10 @@ def federated_aggregate_keras_metric(
"""Aggregates variables a keras metric placed at CLIENTS to SERVER. """Aggregates variables a keras metric placed at CLIENTS to SERVER.
Args: Args:
metrics: A single `tf.keras.metrics.Metric` or a `Sequence` of metrics . The metrics: A single or a `Sequence` of `tf.keras.metrics.Metric` objects, or
order must match the order of variables in `federated_values`. a single or a `Sequence` of no-arg callables that each constructs a
`tf.keras.metrics.Metric`. The order must match the order of variables in
`federated_values`.
federated_values: A single federated value, or a `Sequence` of federated federated_values: A single federated value, or a `Sequence` of federated
values. The values must all have `tff.CLIENTS` placement. If value is a values. The values must all have `tff.CLIENTS` placement. If value is a
`Sequence` type, it must match the order of the sequence in `metrics. `Sequence` type, it must match the order of the sequence in `metrics.
......
支持 Markdown
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册