Skip to content

taps.apps.configs.fedlearn

taps/apps/configs/fedlearn.py

FedlearnConfig

Bases: AppConfig

Federated learning application configuration.

get_app() 

get_app() -> App

Create an application instance from the config.

Source code in taps/apps/configs/fedlearn.py 
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
def get_app(self) -> App:
    """Create an application instance from the config."""
    from taps.apps.fedlearn.app import FedlearnApp
    from taps.apps.fedlearn.types import DataChoices

    return FedlearnApp(
        clients=self.clients,
        rounds=self.rounds,
        device=self.device,
        epochs=self.batch_size,
        batch_size=self.batch_size,
        lr=self.lr,
        dataset=DataChoices(self.dataset),
        data_dir=self.data_dir,
        train=self.train,
        test=self.test,
        participation=self.participation,
        seed=self.seed,
    )