feat(datasets) Add semantic partitioner

datasets/README.md

@@ -42,6 +42,7 @@ Create **custom partitioning schemes** or choose from the **implemented [partiti
 * IID partitioning `IidPartitioner(num_partitions)`
 * Dirichlet partitioning `DirichletPartitioner(num_partitions, partition_by, alpha)`
 * Distribution partitioning `DistributionPartitioner(distribution_array, num_partitions, num_unique_labels_per_partition, partition_by, preassigned_num_samples_per_label, rescale)`
+* Image Semantic partitioning `ImageSemanticPartitioner(num_partitions)`
 * InnerDirichlet partitioning `InnerDirichletPartitioner(partition_sizes, partition_by, alpha)`
 * Pathological partitioning `PathologicalPartitioner(num_partitions, partition_by, num_classes_per_partition, class_assignment_mode)`
 * Natural ID partitioning `NaturalIdPartitioner(partition_by)`

datasets/doc/source/index.rst

@@ -97,6 +97,7 @@ Here are a few of the ``Partitioner`` s that are available: (for a full list see
 * IID partitioning ``IidPartitioner(num_partitions)``
 * Dirichlet partitioning ``DirichletPartitioner(num_partitions, partition_by, alpha)``
 * Distribution partitioning ``DistributionPartitioner(distribution_array, num_partitions, num_unique_labels_per_partition, partition_by, preassigned_num_samples_per_label, rescale)``
+* Image Semantic partitioning ``ImageSemanticPartitioner(num_partitions)``
 * InnerDirichlet partitioning ``InnerDirichletPartitioner(partition_sizes, partition_by, alpha)``
 * PathologicalPartitioner ``PathologicalPartitioner(num_partitions, partition_by, num_classes_per_partition, class_assignment_mode)``
 * Natural ID partitioner ``NaturalIdPartitioner(partition_by)``

datasets/flwr_datasets/partitioner/__init__.py

@@ -19,6 +19,7 @@
 from .distribution_partitioner import DistributionPartitioner
 from .exponential_partitioner import ExponentialPartitioner
 from .iid_partitioner import IidPartitioner
+from .image_semantic_partitioner import ImageSemanticPartitioner
 from .inner_dirichlet_partitioner import InnerDirichletPartitioner
 from .linear_partitioner import LinearPartitioner
 from .natural_id_partitioner import NaturalIdPartitioner
@@ -33,6 +34,7 @@
     "DistributionPartitioner",
     "ExponentialPartitioner",
     "IidPartitioner",
+    "ImageSemanticPartitioner",
     "InnerDirichletPartitioner",
     "LinearPartitioner",
     "NaturalIdPartitioner",