tff.simulation.baselines.BaselineTask

Specification for a baseline learning simulation.

tff.simulation.baselines.BaselineTask(
    datasets, model_fn
)

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__

__eq__(
    other
)

Method generated by attrs for class BaselineTask.

__ne__

__ne__(
    other
)

Method generated by attrs for class BaselineTask.