FedPDMLP

class fl_sim.models.FedPDMLP(dim_in: int, dim_hidden: int, dim_out: int, ndim: int = 2)

Bases: Sequential, CLFMixin, SizeMixin, DiffMixin

Multi-layer perceptron modified from FedPD/models.py.

Parameters:

• dim_in (int) – Number of input features.

• dim_hidden (int) – Number of hidden features.

• dim_out (int) – Number of output features.

• ndim (int, default 2) – Number of dimensions of input data. 2 for image data, 1 for sequence data, 0 for vectorized data.