fedbox.datasets

This submodule defines the federated version of the following datasets.

  1. FEMNIST (written digits subsets) (see paper) that can be loaded with femnist(...).
  2. CIFAR10 and CIFAR100 (see paper) that can be loaded with cifar10(...) and cifar100(...), respectively.

In addition, here we define the routine partition(...) that partitions a centralized dataset in multiple subsets where each corresponds to a client. This function is responsible injecting the specified level of heterogeneity in the generation of the federated subsets.

 1'''
 2This submodule defines the federated version of the following datasets.
 3<ol>
 4  <li>FEMNIST (written digits subsets) (see <a href="https://arxiv.org/pdf/1812.01097">paper</a>) that can be loaded with <b>femnist(...)</b>.</li>
 5  <li>CIFAR10 and CIFAR100 (see <a href="https://www.cs.toronto.edu/~kriz/learning-features-2009-TR.pdf">paper</a>) that can be loaded with <b>cifar10(...)</b> and <b>cifar100(...)</b>, respectively.</li>
 6</ol>
 7In addition, here we define the routine **partition(...)** that partitions a centralized dataset in multiple subsets where each corresponds to a client. This function
 8is responsible injecting the specified level of heterogeneity in the generation of the federated subsets.
 9'''
10
11from .cifar10 import cifar10
12from .cifar100 import cifar100
13from .femnist import femnist, FEMNIST
14from .utils import FederatedSubset, partition