openfl.interface.aggregation_functions.fedcurv_weighted_average.FedCurvWeightedAverage
- class openfl.interface.aggregation_functions.fedcurv_weighted_average.FedCurvWeightedAverage(*args, **kwargs)
Bases:
WeightedAverageAggregation function of FedCurv algorithm.
Applies weighted average aggregation to all tensors except Fisher matrices variables (u_t, v_t). These variables are summed without weights.
FedCurv paper: https://arxiv.org/pdf/1910.07796.pdf
Methods
Apply aggregation.
- __call__(local_tensors, db_iterator, tensor_name, fl_round, tags)
Use magic function for ease.
- call(local_tensors, tensor_db, tensor_name, fl_round, tags)
Apply aggregation.