From 4881e7a6a98dc1d4cf3b5fb247c2e5a822c928a4 Mon Sep 17 00:00:00 2001 From: Adam Narozniak Date: Wed, 24 Jul 2024 17:04:50 +0200 Subject: [PATCH] Clarify the partitioner parameter of FederatedDataset --- datasets/flwr_datasets/federated_dataset.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/datasets/flwr_datasets/federated_dataset.py b/datasets/flwr_datasets/federated_dataset.py index ecdaa1653e12..e913b9095d17 100644 --- a/datasets/flwr_datasets/federated_dataset.py +++ b/datasets/flwr_datasets/federated_dataset.py @@ -54,9 +54,11 @@ class FederatedDataset: no operation is applied. partitioners : Dict[str, Union[Partitioner, int]] A dictionary mapping the Dataset split (a `str`) to a `Partitioner` or an `int` - (representing the number of IID partitions that this split should be partitioned - into). One or multiple `Partitioner` objects can be specified in that manner, - but at most, one per split. + (representing the number of IID partitions that this split should be + partitioned into, i.e., using the default partitioner + `IidPartitioner `_). One or multiple `Partitioner` + objects can be specified in that manner, but at most, one per split. shuffle : bool Whether to randomize the order of samples. Applied prior to preprocessing operations, speratelly to each of the present splits in the dataset. It uses