fcube#

Module Contents#

FCUBE

FCUBE data set.
class FCUBE(root, train=True, generate=True, transform=None, target_transform=None, num_samples=4000)#

Bases: torch.utils.data.Dataset

FCUBE data set.

From paper Federated Learning on Non-IID Data Silos: An Experimental Study.

Parameters:

  • root (str) – Root for data file.

  • train (bool, optional) – Training set or test set. Default as True.

  • generate (bool, optional) – Whether to generate synthetic dataset. If True, then generate new synthetic FCUBE data even existed. Default as True.

  • transform (callable, optional) – A function/transform that takes in an numpy.ndarray and returns a transformed version.

  • target_transform (callable, optional) – A function/transform that takes in the target and transforms it.

  • num_samples (int, optional) – Total number of samples to generate. We suggest to use 4000 for training set, and 1000 for test set. Default is 4000 for trainset.

train_files#
test_files#
num_clients = 4#
_generate_train()#
_generate_test()#
_save_data()#
__len__()#
__getitem__(index)#
Parameters:

index (int) – Index

Returns:

(features, target) where target is index of the target class.

Return type:

tuple