baseline_constants module

Configuration file for common models/experiments

baseline_constants.MAIN_PARAMS = {'femnist': {'large': (400, 20, 2), 'medium': (100, 10, 2), 'small': (30, 10, 2)}, 'sent140': {'large': (24, 2, 2), 'medium': (16, 2, 2), 'small': (10, 2, 2)}, 'shakespeare': {'large': (20, 1, 2), 'medium': (8, 2, 2), 'small': (6, 2, 2)}}

Specifies execution parameters (tot_num_rounds, eval_every_num_rounds, clients_per_round)

Type:dict

baseline_constants.MODEL_PARAMS = {'femnist.cnn': (0.0003, 62), 'sent140.bag_dnn': (0.0003, 2), 'sent140.bag_log_reg': (0.0003, 2), 'sent140.stacked_lstm': (0.0003, 25, 2, 100), 'shakespeare.stacked_lstm': (0.0003, 80, 53, 256)}

Model specific parameter specification

Type:dict

baseline_constants.SIM_TIMES = ['small', 'medium', 'large']

Common sets of configuration for simulations

Type:list