tff.simulation.baselines.BaselineTask

Specification for a baseline learning simulation.

datasets A tff.simulation.baselines.BaselineTaskDatasets object specifying dataset-related aspects of the task, including training data and preprocessing functions.
model_fn A no-arg callable returning a tff.learning.models.VariableModel used for the task. Note that model_fn().input_spec must match datasets.element_type_structure.

Methods

__eq__

Method generated by attrs for class BaselineTask.

__ne__

Method generated by attrs for class BaselineTask.