MEDfl 2.0.4__tar.gz → 2.0.4.dev1__tar.gz

{medfl-2.0.4 → MEDfl-2.0.4.dev1}/.gitignore

@@ -1,2 +1,5 @@
 __pycache__/
 MEDfl.egg-info/
+fl-env/
+dist/
+build/

MEDfl-2.0.4.dev1/MEDfl/rw/model.py

@@ -0,0 +1,19 @@
+import torch.nn as nn
+import torch.nn.functional as F
+
+class Net(nn.Module):
+    def __init__(self, input_dim):
+        super().__init__()
+        self.fc1 = nn.Linear(input_dim, 64)
+        self.fc2 = nn.Linear(64, 32)
+        self.fc3 = nn.Linear(32, 1)
+        self.dropout = nn.Dropout(0.3)
+        self.batchnorm1 = nn.BatchNorm1d(64)
+        self.batchnorm2 = nn.BatchNorm1d(32)
+
+    def forward(self, x):
+        x = F.relu(self.batchnorm1(self.fc1(x)))
+        x = self.dropout(x)
+        x = F.relu(self.batchnorm2(self.fc2(x)))
+        x = self.dropout(x)
+        return self.fc3(x)  # raw logits for BCEWithLogitsLoss

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

@@ -108,7 +108,7 @@ class Strategy:
             # Print individual client fit metrics
             print(f"\n[Server] 🔄 Round {rnd} - Client Training Metrics:")
             for i, (client_id, fit_res) in enumerate(results):
-                print(f"  Client {client_id.cid}: {fit_res.metrics}")
+                print(f" CTM Round {rnd} Client:{client_id.cid}: {fit_res.metrics}")
 
             # Call original aggregation function
             aggregated_params, metrics = original_agg_fit(rnd, results, failures)
@@ -125,7 +125,7 @@ class Strategy:
             # Print individual client evaluation metrics
             print(f"\n[Server] 📊 Round {rnd} - Client Evaluation Metrics:")
             for i, (client_id, eval_res) in enumerate(results):
-                print(f"  Client {client_id.cid}: {eval_res.metrics}")
+                print(f" CEM Round {rnd} Client:{client_id.cid}: {eval_res.metrics}")
 
             # Call original aggregation function
             loss, metrics = original_agg_eval(rnd, results, failures)

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

@@ -1,6 +1,6 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.1
 Name: MEDfl
-Version: 2.0.4
+Version: 2.0.4.dev1
 Summary: Python Open-source package for simulating federated learning and differential privacy
 Home-page: https://github.com/MEDomics-UdeS/MEDfl
 Author: MEDomics consortium
@@ -17,36 +17,6 @@ Classifier: Programming Language :: Python :: 3.9
 Requires-Python: >=3.8,<3.13
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: flwr~=1.18.0
-Requires-Dist: matplotlib~=3.6.3
-Requires-Dist: numpy~=1.26.4
-Requires-Dist: opacus~=1.5.3
-Requires-Dist: pandas~=1.5.2
-Requires-Dist: PyYAML~=6.0
-Requires-Dist: setuptools~=68.0.0
-Requires-Dist: Sphinx~=5.3.0
-Requires-Dist: SQLAlchemy~=1.4.47
-Requires-Dist: torch>=2.0.0
-Requires-Dist: datetime~=5.1
-Requires-Dist: scikit-learn~=1.6.1
-Requires-Dist: sphinx-jsonschema==1.19.1
-Requires-Dist: sphinx-rtd-dark-mode==1.2.4
-Requires-Dist: plotly==5.19.0
-Requires-Dist: optuna==3.5.0
-Requires-Dist: mysql-connector-python~=9.3.0
-Requires-Dist: seaborn~=0.13.2
-Dynamic: author
-Dynamic: author-email
-Dynamic: classifier
-Dynamic: description
-Dynamic: description-content-type
-Dynamic: home-page
-Dynamic: keywords
-Dynamic: license-file
-Dynamic: project-url
-Dynamic: requires-dist
-Dynamic: requires-python
-Dynamic: summary
 
 # MEDfl: Federated Learning and Differential Privacy Simulation Tool for Tabular Data
 ![Python Versions](https://img.shields.io/badge/python-3.9-blue)

{medfl-2.0.4 → MEDfl-2.0.4.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.1
 Name: MEDfl
-Version: 2.0.4
+Version: 2.0.4.dev1
 Summary: Python Open-source package for simulating federated learning and differential privacy
 Home-page: https://github.com/MEDomics-UdeS/MEDfl
 Author: MEDomics consortium
@@ -17,36 +17,6 @@ Classifier: Programming Language :: Python :: 3.9
 Requires-Python: >=3.8,<3.13
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: flwr~=1.18.0
-Requires-Dist: matplotlib~=3.6.3
-Requires-Dist: numpy~=1.26.4
-Requires-Dist: opacus~=1.5.3
-Requires-Dist: pandas~=1.5.2
-Requires-Dist: PyYAML~=6.0
-Requires-Dist: setuptools~=68.0.0
-Requires-Dist: Sphinx~=5.3.0
-Requires-Dist: SQLAlchemy~=1.4.47
-Requires-Dist: torch>=2.0.0
-Requires-Dist: datetime~=5.1
-Requires-Dist: scikit-learn~=1.6.1
-Requires-Dist: sphinx-jsonschema==1.19.1
-Requires-Dist: sphinx-rtd-dark-mode==1.2.4
-Requires-Dist: plotly==5.19.0
-Requires-Dist: optuna==3.5.0
-Requires-Dist: mysql-connector-python~=9.3.0
-Requires-Dist: seaborn~=0.13.2
-Dynamic: author
-Dynamic: author-email
-Dynamic: classifier
-Dynamic: description
-Dynamic: description-content-type
-Dynamic: home-page
-Dynamic: keywords
-Dynamic: license-file
-Dynamic: project-url
-Dynamic: requires-dist
-Dynamic: requires-python
-Dynamic: summary
 
 # MEDfl: Federated Learning and Differential Privacy Simulation Tool for Tabular Data
 ![Python Versions](https://img.shields.io/badge/python-3.9-blue)

{medfl-2.0.4 → MEDfl-2.0.4.dev1}/setup.py

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

medfl-2.0.4/MEDfl/rw/model.py

@@ -1,75 +0,0 @@
-# client.py
-import argparse
-import pandas as pd
-import flwr as fl
-import torch
-import torch.nn as nn
-import torch.optim as optim
-from MEDfl.rw.model import Net  # your model definition in model.py
-
-class FlowerClient(fl.client.NumPyClient):
-    def __init__(self, server_address: str, data_path: str = "data/data.csv"):
-        self.server_address = server_address
-
-        # 1. Load model
-        self.model = Net()
-        self.loss_fn = nn.MSELoss()
-        self.optimizer = optim.SGD(self.model.parameters(), lr=0.01)
-
-        # 2. Load data from CSV
-        df = pd.read_csv(data_path)
-        # Assume last column is label
-        X = df.iloc[:, :-1].values
-        y = df.iloc[:, -1].values
-
-        self.X_train = torch.tensor(X, dtype=torch.float32)
-        # If it's regression with single output; remove unsqueeze for multi-class
-        self.y_train = torch.tensor(y, dtype=torch.float32).unsqueeze(1)
-
-    def get_parameters(self, config):
-        return [val.cpu().numpy() for val in self.model.state_dict().values()]
-
-    def set_parameters(self, parameters):
-        params_dict = zip(self.model.state_dict().keys(), parameters)
-        state_dict = {k: torch.tensor(v) for k, v in params_dict}
-        self.model.load_state_dict(state_dict, strict=True)
-
-    def fit(self, parameters, config):
-        self.set_parameters(parameters)
-        self.model.train()
-        for _ in range(5):
-            self.optimizer.zero_grad()
-            preds = self.model(self.X_train)
-            loss = self.loss_fn(preds, self.y_train)
-            loss.backward()
-            self.optimizer.step()
-        # Return updated params, number of examples, and an empty metrics dict
-        return self.get_parameters(config), len(self.X_train), {}
-
-    def evaluate(self, parameters, config):
-        self.set_parameters(parameters)
-        self.model.eval()
-        with torch.no_grad():
-            preds = self.model(self.X_train)
-            loss = self.loss_fn(preds, self.y_train).item()
-        return float(loss), len(self.X_train), {}
-
-if __name__ == "__main__":
-    parser = argparse.ArgumentParser(description="Flower client")
-    parser.add_argument(
-        "--server_address",
-        type=str,
-        required=True,
-        help="Address of the Flower server (e.g., 127.0.0.1:8080)",
-    )
-    parser.add_argument(
-        "--data_path",
-        type=str,
-        default="data/data.csv",
-        help="Path to your CSV training data",
-    )
-    args = parser.parse_args()
-
-    # Instantiate and start the client
-    client = FlowerClient(server_address=args.server_address, data_path=args.data_path)
-    fl.client.start_numpy_client(server_address=client.server_address, client=client)