Datasets

Partitioners

DatasetPartitioner

An abstract base class for dataset splitting strategies that considers random states from both NumPy and PyTorch.

IIDPartitioner

Partitioner that splits a dataset into IID subsets.

ShardPartitioner

Partitioner that splits a dataset into shards and assigns shards to clients.

DirichletPartitioner

Partitioner that uses Dirichlet distribution to allocate samples to clients.