Module: tff.learning.metrics

Libraries for working with metrics in federated learning algorithms.

Classes

class NumBatchesCounter: A tf.keras.metrics.Metric that counts the number of batches seen.

class NumExamplesCounter: A tf.keras.metrics.Metric that counts the number of examples seen.

Functions

create_default_secure_sum_quantization_ranges(...): Create a nested structure of quantization ranges for secure sum encoding.

create_keras_metric_finalizer(...): Creates a finalizer function for the given Keras metric.

secure_sum_then_finalize(...): Creates a TFF computation that aggregates metrics using secure summation.

sum_then_finalize(...): Creates a TFF computation that aggregates metrics via sum_then_finalize.