MEDfl 2.0.4.dev10__tar.gz → 2.0.4.dev11__tar.gz

{medfl-2.0.4.dev10 → medfl-2.0.4.dev11}/MEDfl/rw/strategy.py

@@ -1,5 +1,7 @@
 # File: MEDfl/rw/strategy.py
 
+import os
+import numpy as np
 import flwr as fl
 from typing import Callable, Optional, Dict, Any, List, Tuple
 from flwr.common import GetPropertiesIns
@@ -38,6 +40,7 @@ class Strategy:
       - Custom metric aggregation
       - Per-client & aggregated metric logging
       - Synchronous get_properties() inspection in configure_fit()
+      - Saving global parameters every saveOnRounds to savingPath
     """
 
     def __init__(
@@ -56,6 +59,9 @@ class Strategy:
         threshold: float = 0.5,
         learning_rate: float = 0.01,
         optimizer_name: str = "SGD",
+        savingPath: str = "",
+        saveOnRounds: int = 3,
+        total_rounds: int = 3,
     ) -> None:
         self.name = name
         self.fraction_fit = fraction_fit
@@ -74,6 +80,9 @@ class Strategy:
         self.threshold = threshold
         self.learning_rate = learning_rate
         self.optimizer_name = optimizer_name
+        self.savingPath = savingPath
+        self.saveOnRounds = saveOnRounds
+        self.total_rounds = total_rounds
 
         self.strategy_object: Optional[fl.server.strategy.Strategy] = None
 
@@ -116,6 +125,13 @@ class Strategy:
                 print(f" CTM Round {server_round} Client:{client_id.cid}: {fit_res.metrics}")
             agg_params, metrics = original_agg_fit(server_round, results, failures)
             print(f"[Server] ✅ Round {server_round} - Aggregated Training Metrics: {metrics}\n")
+            # save the model parameters if savingPath is set on each saveOnRounds 
+            if self.savingPath and ((server_round % self.saveOnRounds == 0) or (self.total_rounds and server_round == self.total_rounds)):
+                arrays = fl.common.parameters_to_ndarrays(agg_params)
+                # Determine filename: final_model on last round else round_{n}
+                filename = f"round_{server_round}_final_model.npz" if server_round == self.total_rounds else f"round_{server_round}_model.npz"
+                filepath = os.path.join(self.savingPath, filename)
+                np.savez(filepath, *arrays)
             return agg_params, metrics
         strat.aggregate_fit = logged_agg_fit
 

{medfl-2.0.4.dev10 → medfl-2.0.4.dev11}/MEDfl.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: MEDfl
-Version: 2.0.4.dev10
+Version: 2.0.4.dev11
 Summary: Python Open-source package for simulating federated learning and differential privacy
 Home-page: https://github.com/MEDomics-UdeS/MEDfl
 Author: MEDomics consortium

{medfl-2.0.4.dev10 → medfl-2.0.4.dev11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: MEDfl
-Version: 2.0.4.dev10
+Version: 2.0.4.dev11
 Summary: Python Open-source package for simulating federated learning and differential privacy
 Home-page: https://github.com/MEDomics-UdeS/MEDfl
 Author: MEDomics consortium

{medfl-2.0.4.dev10 → medfl-2.0.4.dev11}/setup.py

@@ -14,7 +14,7 @@ with open('requirements.txt') as f:
 
 setup(
     name="MEDfl",
-    version="2.0.4.dev10",
+    version="2.0.4.dev11",
     author="MEDomics consortium",
     author_email="medomics.info@gmail.com",
     description="Python Open-source package for simulating federated learning and differential privacy",