View source on GitHub |
Specification for a baseline learning simulation.
tff.simulation.baselines.BaselineTask(
datasets, model_fn
)
Attributes | |
---|---|
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.