From a74468133524820ea02756dcb105bc2a5eb1a331 Mon Sep 17 00:00:00 2001 From: Chong Shen Ng Date: Thu, 18 Jul 2024 14:54:48 +0100 Subject: [PATCH] Add alternate distribution description --- .../flwr_datasets/partitioner/distribution_partitioner.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/datasets/flwr_datasets/partitioner/distribution_partitioner.py b/datasets/flwr_datasets/partitioner/distribution_partitioner.py index 701f6719682b..1054cdefd8f2 100644 --- a/datasets/flwr_datasets/partitioner/distribution_partitioner.py +++ b/datasets/flwr_datasets/partitioner/distribution_partitioner.py @@ -54,7 +54,9 @@ class DistributionPartitioner(Partitioner): # pylint: disable=R0902 and a configuration with 20 partitions and 2 unique labels per partition, the 0'th row of the distribution array (corresponding to class 0) will be assigned to partitions [0, 9, 10, 19], 1st row (class 1) to [0, 1, 10, 11], 2nd row (class 2) - to [1, 2, 11, 12], 3rd row (class 3) to [2, 3, 12, 13], etc ... + to [1, 2, 11, 12], 3rd row (class 3) to [2, 3, 12, 13], etc ... . Alternatively, the + distribution can be interpreted as partition 0 having classes 0 and 1, partition 1 + having classes 1 and 2, partition 2 having classes 2 and 3, etc ... The list representing the unique labels is sorted in ascending order. Parameters