CNNFEMnist#
- class fl_sim.models.CNNFEMnist(num_classes: int = 62)[source]#
Bases:
Sequential
,CLFMixin
,SizeMixin
,DiffMixin
Convolutional neural network using FEMnist type input.
Modified from FedPD/models.py.
Input shape:
(batch_size, 1, 28, 28)
.- Parameters:
num_classes (int, default 62) – Number of output classes.