fl_sim.data_processing

fl_sim.data_processing#

本模块包含联邦数据集与数据处理工具。

基类#

FedDataset()

所有联邦数据集的基类。

FedVisionDataset([datadir, transform, seed])

所有联邦视觉（vision）数据集的基类。

FedNLPDataset([datadir, seed])

所有联邦自然语言（NLP）数据集的基类。

视觉数据集#

FedCIFAR([n_class, datadir, transform, seed])

联邦 CIFAR10/100 数据集。

FedCIFAR100([datadir, transform, seed])

FedEMNIST([datadir, transform, seed])

联邦 EMNIST 数据集。

FedMNIST([datadir, transform, seed])

MNIST 是用于研究手写数字 0-9 图像分类的数据集。

FedRotatedCIFAR10([datadir, num_rotations, ...])

带旋转增强的 CIFAR10 数据集。

FedRotatedMNIST([datadir, num_rotations, ...])

带旋转增强的 MNIST 数据集。

FedProxFEMNIST([datadir, transform, seed])

FedProx 论文中提出的联邦 EMNIST 子集。

FedProxMNIST([datadir, transform, seed])

FedProx 论文中提出的联邦 MNIST。

FedTinyImageNet([datadir, num_clients, ...])

Tiny ImageNet 数据集。

自然语言数据集#

FedShakespeare([datadir, seed])

联邦莎士比亚（Shakespeare）数据集。

FedProxSent140([datadir, seed])

FedProx 论文中使用的联邦 Sentiment140 数据集。

人造数据集#

FedSynthetic(alpha, beta, iid, num_clients)

联邦人造（synthetic）数据集。

LibSVM 数据集#

FedLibSVMDataset(dataset_name, num_clients)

数据集注册工具#

register_fed_dataset([name, override])

注册新联邦数据集的装饰器。

list_fed_dataset()

列出所有已注册的联邦数据集。

get_fed_dataset(name)

按名称获取已注册的联邦数据集。
目录