partitionerΒΆ

Flower Datasets Partitioner package.

Classes

DirichletPartitioner(num_partitions, ...[, ...])

Partitioner based on Dirichlet distribution.

DistributionPartitioner(distribution_array, ...)

Partitioner based on a distribution.

ExponentialPartitioner(num_partitions)

Partitioner creates partitions of size that are correlated with exp(id).

GroupedNaturalIdPartitioner(partition_by, ...)

Partition dataset by creating groups of natural ids.

IdToSizeFncPartitioner(num_partitions, ...)

Base class for the deterministic size partitioning based on the partition_id.

IidPartitioner(num_partitions)

Partitioner creates each partition sampled randomly from the dataset.

InnerDirichletPartitioner(partition_sizes, ...)

Partitioner based on Dirichlet distribution.

LinearPartitioner(num_partitions)

Partitioner creates partitions of size that are linearly correlated with id.

NaturalIdPartitioner(partition_by)

Partitioner for a dataset that can be divided by a column with partition ids.

Partitioner()

The base partitioner class that enables obtaining federated partitions.

PathologicalPartitioner(num_partitions, ...)

Partition dataset such that each partition has a chosen number of classes.

ShardPartitioner(num_partitions, partition_by)

Partitioner based on shard of (typically) unique classes.

SizePartitioner(partition_sizes)

Partitioner that creates each partition with the size specified by a user.

SquarePartitioner(num_partitions)

Partitioner creates partitions of size that are correlated with squared id.