diff --git a/src/py/flwr/server/strategy/fedxgb_nn_avg.py b/src/py/flwr/server/strategy/fedxgb_nn_avg.py index 020e0ef71267..f300633d0d9f 100644 --- a/src/py/flwr/server/strategy/fedxgb_nn_avg.py +++ b/src/py/flwr/server/strategy/fedxgb_nn_avg.py @@ -17,7 +17,7 @@ Strategy in the horizontal setting based on building Neural Network and averaging on prediction outcomes. -Paper: Coming +Paper: arxiv.org/abs/2304.07537 """ @@ -35,6 +35,13 @@ class FedXgbNnAvg(FedAvg): """Configurable FedXgbNnAvg strategy implementation.""" + def __init__(self, *args: Any, **kwargs: Any) -> None: + """Federated XGBoost [Ma et al., 2023] strategy. + + Implementation based on https://arxiv.org/abs/2304.07537. + """ + super().__init__(*args, **kwargs) + def __repr__(self) -> str: """Compute a string representation of the strategy.""" rep = f"FedXgbNnAvg(accept_failures={self.accept_failures})"