tff.learning.models.BatchOutput

A structure that holds the output of a tff.learning.models.VariableModel.

tff.learning.models.BatchOutput(
    loss=None, predictions=None, num_examples=None
)

Used in the tutorials
Federated Learning for Image Classification

Attributes
`loss`	The scalar mean loss on the examples in the batch. If the model has multiple losses, it is the sum of all the individual losses.
`predictions`	Tensor of predictions on the examples. The first dimension must be the same size (the size of the batch).
`num_examples`	Number of examples seen in the batch.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2024-09-20 UTC.

Attributes