p2pfl.learning.frameworks.tensorflow.keras_dataset module

Keras dataset export strategy.

class p2pfl.learning.frameworks.tensorflow.keras_dataset.KerasExportStrategy[source]

Bases: DataExportStrategy

Export strategy for TensorFlow/Keras datasets.

static export(data, batch_size=None, **kwargs)[source]

Export the data as a TensorFlow Dataset.

Parameters:

  • data (Dataset) – The Hugging Face Dataset to export. Transforms should already be applied to the dataset via set_transform.

  • batch_size (int | None) – The batch size for the TensorFlow Dataset.

  • seed – The seed for the TensorFlow Dataset.

  • **kwargs – Additional keyword arguments.

Return type:

Dataset

Returns:

A TensorFlow Dataset.