fedavg algorithm is implemented

Author: gaborvecsei

fed_learn/fed_server.py

@@ -3,10 +3,11 @@
 from keras import datasets
 
 import fed_learn
+from fed_learn.weight_summarizer import WeightSummarizer
 
 
 class Server:
-    def __init__(self, model_fn: Callable, nb_clients: int, weight_summarizer: fed_learn.WeightSummarizer):
+    def __init__(self, model_fn: Callable, nb_clients: int, weight_summarizer: WeightSummarizer):
         self.nb_clients = nb_clients
         self.weight_summarizer = weight_summarizer
 

fed_learn/weight_summarizer.py

@@ -16,5 +16,11 @@ def __init__(self):
         super().__init__()
 
     def process(self, client_weight_list: List[List[np.ndarray]]) -> List[np.ndarray]:
-        # TODO: implement simple averaging
-        return client_weight_list[0]
+        weights_average = [np.zeros_like(w) for w in client_weight_list[0]]
+
+        for i in range(len(weights_average)):
+            w = weights_average[i]
+            for k in range(len(client_weight_list)):
+                w += client_weight_list[k][i]
+            w /= len(client_weight_list)
+        return weights_average