importance_sampler#

Module Contents#

MultiArmedBanditSampler

Refer to [Stochastic Optimization with Bandit Sampling](https://arxiv.org/abs/1708.02544).

OptimalSampler

Refer to [Optimal Client Sampling for Federated Learning](arxiv.org/abs/2010.13723).

class MultiArmedBanditSampler(n, T, L)#

Bases: fedlab.contrib.client_sampler.base_sampler.FedSampler

Refer to [Stochastic Optimization with Bandit Sampling](https://arxiv.org/abs/1708.02544).

sample(batch_size)#
update(loss)#
class OptimalSampler(n, k)#

Bases: fedlab.contrib.client_sampler.base_sampler.FedSampler

Refer to [Optimal Client Sampling for Federated Learning](arxiv.org/abs/2010.13723).

sample(size=None)#
update(loss)#
optim_solver(norms)#