MEDfl 0.1.30__py3-none-any.whl → 0.1.32__py3-none-any.whl

{MEDfl-0.1.30.dist-info → MEDfl-0.1.32.dist-info}/METADATA

@@ -1,127 +1,127 @@
-Metadata-Version: 2.1
-Name: MEDfl
-Version: 0.1.30
-Summary: Python Open-source package for simulating federated learning and differential privacy
-Home-page: https://github.com/HaithemLamri/MEDfl
-Author: MEDomics consortium
-Author-email: medomics.info@gmail.com
-Project-URL: Documentation, https://
-Project-URL: Github, https://github.com/HaithemLamri/MEDfl
-Keywords: federated learning differential privacy medical research
-Classifier: Development Status :: 3 - Alpha
-Classifier: Intended Audience :: Developers
-Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Programming Language :: Python :: 3.8
-Classifier: Programming Language :: Python :: 3.9
-Requires-Python: >=3.8,<3.11
-Description-Content-Type: text/markdown
-Requires-Dist: flwr (~=1.7.0)
-Requires-Dist: matplotlib (~=3.6.3)
-Requires-Dist: numpy (~=1.24.2)
-Requires-Dist: opacus (~=1.3.0)
-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 (~=1.13.1)
-Requires-Dist: datetime (~=5.1)
-Requires-Dist: scikit-learn (~=1.1.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
-Requires-Dist: mysql-connector-python
-
-# MEDfl : Federated Learning and Differential Privacy Simulation Tool for Tabular Data
-![Python Versions](https://img.shields.io/badge/python-3.9-blue)
-![Build Status](https://travis-ci.org/MEDomics-UdeS/MEDfl.svg?branch=main)
-
-![GitHub contributors](https://img.shields.io/github/contributors/scottydocs/README-template.md)
-![License: MIT](https://img.shields.io/badge/license-MIT-green)
-
-
-
-
-## Table of Contents
-  * [1. Introduction](#1-introduction)
-  * [2. Installation](#2-installation)
-  * [3. Documentation](#3-documentation)
-  * [4. Getting started](#4-Getting-started)
-  * [5. Acknowledgement](#5-acknowledgement)
-  * [6. Authors](#6-authors)
-  * [7. Statement](#7-statement)
-
-## 1. Introduction
-This Python package is an open-source tool designed for simulating federated learning and incorporating differential privacy. It empowers researchers and developers to effortlessly create, execute, and assess federated learning pipelines while seamlessly working with various tabular datasets.
-
-
-
-
-
-## 2. Installation
-
-### Python installation
-The MEDfl package requires *python 3.9* or more to be run. If you don't have it installed  on your machine, check out the following link  [Python ](https://www.python.org/downloads/).
-It also requires MySQL database
-
-### Package installation
-For now, you can  install the ``MEDfl``package as:
-```
-git clone https://github.com/MEDomics-UdeS/MEDfl.git
-cd MEDfl
-pip install -e .
-```
-### MySQL DB configuration
-MEDfl requires a MySQL DB connection, and this is in order to allow users to work with their own tabular datasets,  we have created a bash script to install and configure A MySQL DB with phpmyadmin monitoring system, run the following command then change your credential on the MEDfl/scripts/base.py and MEDfl/scripts/db_config.ini files
-```
-sudo bash MEDfl/scripts/setup_mysql.sh
-```
-
-### Project Base URL Update
-Please ensure to modify the `base_url` parameter in the `MEDfl/global_params.yaml` file. The `base_url` represents the path to the MEDfl project on your local machine. Update this value accordingly.
-
-## 3. Documentation
-We used sphinx to create the documentation for this project.  you can generate and host it locally by compiling the documentation source code using:
-```
-cd docs
-make clean
-make html
-```
-
-Then open it locally using:
-
-```
-cd _build/html
-python -m http.server
-```
-
-## 4. Getting started
-
-We have created a complete tutorial for the different functionalities of the package. it can be found here [tutorial](https://github.com/MEDomics-UdeS/MEDfl/notebooks/First_tuto.ipynb)
-
-
-
-## 5. Acknowledgment
-MEDfl is an open-source package that welcomes any contribution and feedback. We wish that this package could serve the growing private AI research community.
-
-## 6. Authors
-* [MEDomics](https://github.com/medomics/): MEDomics consortium.
-
-## 7. Statement
-
-This package is part of https://github.com/medomics, a package providing research utility tools for developing precision medicine applications.
-
-```
-MIT License
-
-Copyright (C) 2022 MEDomics consortium
-
-Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
-```
+Metadata-Version: 2.1
+Name: MEDfl
+Version: 0.1.32
+Summary: Python Open-source package for simulating federated learning and differential privacy
+Home-page: https://github.com/MEDomics-UdeS/MEDfl
+Author: MEDomics consortium
+Author-email: medomics.info@gmail.com
+Project-URL: Documentation, https://
+Project-URL: Github, https://github.com/MEDomics-UdeS/MEDfl
+Keywords: federated learning differential privacy medical research
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Requires-Python: >=3.8,<3.11
+Description-Content-Type: text/markdown
+Requires-Dist: flwr ~=1.7.0
+Requires-Dist: matplotlib ~=3.6.3
+Requires-Dist: numpy ~=1.24.2
+Requires-Dist: opacus ~=1.3.0
+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 ~=1.13.1
+Requires-Dist: datetime ~=5.1
+Requires-Dist: scikit-learn ~=1.1.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
+Requires-Dist: mysql-connector-python
+
+# MEDfl : Federated Learning and Differential Privacy Simulation Tool for Tabular Data
+![Python Versions](https://img.shields.io/badge/python-3.9-blue)
+![Build Status](https://travis-ci.org/MEDomics-UdeS/MEDfl.svg?branch=main)
+
+![GitHub contributors](https://img.shields.io/github/contributors/scottydocs/README-template.md)
+![License: MIT](https://img.shields.io/badge/license-MIT-green)
+
+
+
+
+## Table of Contents
+  * [1. Introduction](#1-introduction)
+  * [2. Installation](#2-installation)
+  * [3. Documentation](#3-documentation)
+  * [4. Getting started](#4-Getting-started)
+  * [5. Acknowledgement](#5-acknowledgement)
+  * [6. Authors](#6-authors)
+  * [7. Statement](#7-statement)
+
+## 1. Introduction
+This Python package is an open-source tool designed for simulating federated learning and incorporating differential privacy. It empowers researchers and developers to effortlessly create, execute, and assess federated learning pipelines while seamlessly working with various tabular datasets.
+
+
+
+
+
+## 2. Installation
+
+### Python installation
+The MEDfl package requires *python 3.9* or more to be run. If you don't have it installed  on your machine, check out the following link  [Python ](https://www.python.org/downloads/).
+It also requires MySQL database
+
+### Package installation
+For now, you can  install the ``MEDfl``package as:
+```
+git clone https://github.com/MEDomics-UdeS/MEDfl.git
+cd MEDfl
+pip install -e .
+```
+### MySQL DB configuration
+MEDfl requires a MySQL DB connection, and this is in order to allow users to work with their own tabular datasets,  we have created a bash script to install and configure A MySQL DB with phpmyadmin monitoring system, run the following command then change your credential on the MEDfl/scripts/base.py and MEDfl/scripts/db_config.ini files
+```
+sudo bash MEDfl/scripts/setup_mysql.sh
+```
+
+### Project Base URL Update
+Please ensure to modify the `base_url` parameter in the `MEDfl/global_params.yaml` file. The `base_url` represents the path to the MEDfl project on your local machine. Update this value accordingly.
+
+## 3. Documentation
+We used sphinx to create the documentation for this project.  you can generate and host it locally by compiling the documentation source code using:
+```
+cd docs
+make clean
+make html
+```
+
+Then open it locally using:
+
+```
+cd _build/html
+python -m http.server
+```
+
+## 4. Getting started
+
+We have created a complete tutorial for the different functionalities of the package. it can be found here [tutorial](https://github.com/MEDomics-UdeS/MEDfl/notebooks/First_tuto.ipynb)
+
+
+
+## 5. Acknowledgment
+MEDfl is an open-source package that welcomes any contribution and feedback. We wish that this package could serve the growing private AI research community.
+
+## 6. Authors
+* [MEDomics](https://github.com/medomics/): MEDomics consortium.
+
+## 7. Statement
+
+This package is part of https://github.com/medomics, a package providing research utility tools for developing precision medicine applications.
+
+```
+MIT License
+
+Copyright (C) 2022 MEDomics consortium
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+```

MEDfl-0.1.32.dist-info/RECORD

@@ -0,0 +1,34 @@
+Medfl/__init__.py,sha256=pMyTkws4slDkLQgPpPdKoWdQf1rZGHXAlNxfPFFZM-I,81
+Medfl/LearningManager/__init__.py,sha256=mvlAmHEHljXGaB6Ij0EPN0Txw21qX25ELK3X7QkoVwA,358
+Medfl/LearningManager/client.py,sha256=9WyLYCsI9JuHjneLbbzDf7HtzjYINuLfqwkbxOsrBrE,6083
+Medfl/LearningManager/dynamicModal.py,sha256=0mTvDJlss0uSJ3_EXOuL_d-zRmFyXaKB4W4ql-uEX8Y,10821
+Medfl/LearningManager/federated_dataset.py,sha256=fQqIbhO6LSk16Ob9z6RohaZ8X71Ff-yueynjulrl4M0,2141
+Medfl/LearningManager/flpipeline.py,sha256=M4-OL4nlogv08J_YsyDsGHXR6xe8BWx4HIsuL1QyUvY,7303
+Medfl/LearningManager/model.py,sha256=DA7HP34Eq1Ra65OlkBmjH9d2MD7OEbsOhfxD48l4QOk,7679
+Medfl/LearningManager/params.yaml,sha256=5I-NljhnSaqzjkWVNzrOtrB8z7tnHBKiBmY-mKGhBQM,450
+Medfl/LearningManager/params_optimiser.py,sha256=pjhDskhSPuca-jnarYoJcFVBvRkdD9tD3992q_eMPSE,18060
+Medfl/LearningManager/plot.py,sha256=iPqMV9rVd7hquoFixDL20OzXI5bMpBW41bkVmTKIWtE,7927
+Medfl/LearningManager/server.py,sha256=7edxPkZ9Ju3Mep_BSHQpUNgW9HKfCui3_l996buJVlU,7258
+Medfl/LearningManager/strategy.py,sha256=n0asQajkHfGLDX3QbbV5qntQA-xuJZU8Z92XccZENsA,3517
+Medfl/LearningManager/utils.py,sha256=vEhkpyC7iLsn4wp1wDh7GzAn5MCJ7T69jkS9lfmKA1Y,9936
+Medfl/NetManager/__init__.py,sha256=RhO9Ya6wXOdM6qO58wjTD-lNL7-q8KvPDvSccYP9wUY,246
+Medfl/NetManager/database_connector.py,sha256=JKfFLom7I4zuykb8m7aY4cUYZy4j0-i9w3R_jkPzjXY,1516
+Medfl/NetManager/dataset.py,sha256=eEuVzCp5dGD4tvDVKq6jlSReecge7T20ByG4d7_cnXU,2869
+Medfl/NetManager/flsetup.py,sha256=CS7531I08eLm6txMIDWFMCIrPP-dNpOLBTaR2BR6X0c,11754
+Medfl/NetManager/net_helper.py,sha256=T5Y-03SVskK8oXXIpiXzASyEDPQJbcuGVpIS9FnmzI8,7066
+Medfl/NetManager/net_manager_queries.py,sha256=bJ7V6NZAjLM0sEPrczN7vUeB7sMZZzYZjO-rnUCpQPc,4228
+Medfl/NetManager/network.py,sha256=D7liCx-1PGx9Z3pctI1_NOGG--gGtvzfr_-BCOdwsOg,7023
+Medfl/NetManager/node.py,sha256=Azye74P4ZUDQxPrQOId4BK8hX-9RzFtN2Ji8NXbDCEM,6726
+Medfl/scripts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+Medfl/scripts/base.py,sha256=pR7StIt3PpX30aoh53gMkpeNJMHytAPhdc7N09tCITA,781
+Medfl/scripts/create_db.py,sha256=PgA6N68iTSfnrt6zy7FDZX2lLjQQ7Ual1Y0efve8gf4,3943
+alembic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+alembic/env.py,sha256=a4zJAzPNLHnIrUlXCqf_8vuAlFu0pceFJJKM1PQaOI4,1649
+scripts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+scripts/base.py,sha256=pR7StIt3PpX30aoh53gMkpeNJMHytAPhdc7N09tCITA,781
+scripts/config.ini,sha256=tDQQlpwZbCFQuSS017yjEiLglLihp6wETbWtNrSWeAA,82
+scripts/create_db.py,sha256=iWE1z33rU_KbIeqoVzdWLhDRLk00TcRf2iYuqpDzOjw,4494
+MEDfl-0.1.32.dist-info/METADATA,sha256=CRkLasq6gCSG73VSFARBVssUEGtbp819nnXnr9hwuZI,5582
+MEDfl-0.1.32.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+MEDfl-0.1.32.dist-info/top_level.txt,sha256=dIL9X8HOFuaVSzpg40DVveDPrymWRoShHtspH7kkjdI,14
+MEDfl-0.1.32.dist-info/RECORD,,

{MEDfl-0.1.30.dist-info → MEDfl-0.1.32.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.37.1)
+Generator: bdist_wheel (0.43.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

{MEDfl-0.1.30.dist-info → MEDfl-0.1.32.dist-info}/top_level.txt

@@ -1,3 +1,2 @@
 MEDfl
 alembic
-scripts

Medfl/LearningManager/__init__.py

@@ -1,13 +1,13 @@
-# # MEDfl/LearningManager/__init__.py
-
-# # Import modules from this package
-# from .client import *
-# from .dynamicModal import *
-# from .flpipeline import *
-# from .federated_dataset import *
-# from .model import *
-# from .params_optimiser import *
-# from .plot import *
-# from .server import *
-# from .strategy import *
-# from .utils import *
+# # MEDfl/LearningManager/__init__.py
+
+# # Import modules from this package
+# from .client import *
+# from .dynamicModal import *
+# from .flpipeline import *
+# from .federated_dataset import *
+# from .model import *
+# from .params_optimiser import *
+# from .plot import *
+# from .server import *
+# from .strategy import *
+# from .utils import *

Medfl/LearningManager/client.py

@@ -1,150 +1,150 @@
-#!/usr/bin/env python3
-import flwr as fl
-from opacus import PrivacyEngine
-from torch.utils.data import DataLoader
-
-from .model import Model
-from .utils import params
-import torch
-
-class FlowerClient(fl.client.NumPyClient):
-    """
-    FlowerClient class for creating MEDfl clients.
-
-    Attributes:
-        cid (str): Client ID.
-        local_model (Model): Local model of the federated learning network.
-        trainloader (DataLoader): DataLoader for training data.
-        valloader (DataLoader): DataLoader for validation data.
-        diff_priv (bool): Flag indicating whether to use differential privacy.
-    """
-    def __init__(self, cid: str, local_model: Model, trainloader: DataLoader, valloader: DataLoader, diff_priv: bool = params["diff_privacy"]):
-        """
-        Initializes the FlowerClient instance.
-
-        Args:
-            cid (str): Client ID.
-            local_model (Model): Local model of the federated learning network.
-            trainloader (DataLoader): DataLoader for training data.
-            valloader (DataLoader): DataLoader for validation data.
-            diff_priv (bool): Flag indicating whether to use differential privacy.
-        """
-        self.cid = cid
-        self.local_model = local_model
-        self.trainloader = trainloader
-        self.valloader = valloader
-        if torch.cuda.is_available():
-            num_cuda_devices = torch.cuda.device_count()
-            if num_cuda_devices > 0:
-                device_idx = int(self.cid) % num_cuda_devices
-                self.device = torch.device(f"cuda:{device_idx}")
-                self.local_model.model.to(self.device)
-            else:
-                # Handle case where CUDA is available but no CUDA devices are found
-                raise RuntimeError("CUDA is available, but no CUDA devices are found.")
-        else:
-            # Handle case where CUDA is not available
-            self.device = torch.device("cpu")
-            self.local_model.model.to(self.device)
-
-        self.privacy_engine = PrivacyEngine(secure_mode=False)
-        self.diff_priv = diff_priv
-        self.epsilons = []
-        self.accuracies = []
-        self.losses = []
-        if self.diff_priv:
-            model, optimizer, self.trainloader = self.privacy_engine.make_private_with_epsilon(
-                module=self.local_model.model.train(),
-                optimizer=self.local_model.optimizer,
-                data_loader=self.trainloader,
-                epochs=params["train_epochs"],
-                target_epsilon=float(params["EPSILON"]),
-                target_delta= float(params["DELTA"]),
-                max_grad_norm=params["MAX_GRAD_NORM"],
-            )
-            setattr(self.local_model, "model", model)
-            setattr(self.local_model, "optimizer", optimizer)
-        self.validate()
-
-    def validate(self):
-        """Validates cid, local_model, trainloader, valloader."""
-        if not isinstance(self.cid, str):
-            raise TypeError("cid argument must be a string")
-
-        if not isinstance(self.local_model, Model):
-            raise TypeError("local_model argument must be a MEDfl.LearningManager.model.Model")
-
-        if not isinstance(self.trainloader, DataLoader):
-            raise TypeError("trainloader argument must be a torch.utils.data.dataloader")
-
-        if not isinstance(self.valloader, DataLoader):
-            raise TypeError("valloader argument must be a torch.utils.data.dataloader")
-
-        if not isinstance(self.diff_priv, bool):
-            raise TypeError("diff_priv argument must be a bool")
-
-    def get_parameters(self, config):
-        """
-        Returns the current parameters of the local model.
-
-        Args:
-            config: Configuration information.
-
-        Returns:
-            Numpy array: Parameters of the local model.
-        """
-        print(f"[Client {self.cid}] get_parameters")
-        return self.local_model.get_parameters()
-
-    def fit(self, parameters, config):
-        """
-        Fits the local model to the received parameters using federated learning.
-
-        Args:
-            parameters: Parameters received from the server.
-            config: Configuration information.
-
-        Returns:
-            Tuple: Parameters of the local model, number of training examples, and privacy information.
-        """
-        print('\n -------------------------------- \n  this is the config of the client')
-        print(f"[Client {self.cid}] fit, config: {config}")
-        # print(config['epochs'])
-        print('\n -------------------------------- \n  ')
-        self.local_model.set_parameters(parameters)
-        for _ in range(params["train_epochs"]):
-            epsilon = self.local_model.train(
-                self.trainloader,
-                epoch=_,
-                device=self.device,
-                privacy_engine=self.privacy_engine,
-                diff_priv=self.diff_priv,
-            )
-            self.epsilons.append(epsilon)
-        print(f"epsilon of client {self.cid} : eps = {epsilon}")
-        return (
-            self.local_model.get_parameters(),
-            len(self.trainloader),
-            {"epsilon": epsilon},
-        )
-
-    def evaluate(self, parameters, config):
-        """
-        Evaluates the local model on the validation data and returns the loss and accuracy.
-
-        Args:
-            parameters: Parameters received from the server.
-            config: Configuration information.
-
-        Returns:
-            Tuple: Loss, number of validation examples, and accuracy information.
-        """
-        print(f"[Client {self.cid}] evaluate, config: {config}")
-        self.local_model.set_parameters(parameters)
-        loss, accuracy , auc = self.local_model.evaluate(
-            self.valloader, device=self.device
-        )
-        self.losses.append(loss)
-        self.accuracies.append(accuracy)
-        
-        return float(loss), len(self.valloader), {"accuracy": float(accuracy)}
+#!/usr/bin/env python3
+import flwr as fl
+from opacus import PrivacyEngine
+from torch.utils.data import DataLoader
+
+from .model import Model
+from .utils import params
+import torch
+
+class FlowerClient(fl.client.NumPyClient):
+    """
+    FlowerClient class for creating MEDfl clients.
+
+    Attributes:
+        cid (str): Client ID.
+        local_model (Model): Local model of the federated learning network.
+        trainloader (DataLoader): DataLoader for training data.
+        valloader (DataLoader): DataLoader for validation data.
+        diff_priv (bool): Flag indicating whether to use differential privacy.
+    """
+    def __init__(self, cid: str, local_model: Model, trainloader: DataLoader, valloader: DataLoader, diff_priv: bool = params["diff_privacy"]):
+        """
+        Initializes the FlowerClient instance.
+
+        Args:
+            cid (str): Client ID.
+            local_model (Model): Local model of the federated learning network.
+            trainloader (DataLoader): DataLoader for training data.
+            valloader (DataLoader): DataLoader for validation data.
+            diff_priv (bool): Flag indicating whether to use differential privacy.
+        """
+        self.cid = cid
+        self.local_model = local_model
+        self.trainloader = trainloader
+        self.valloader = valloader
+        if torch.cuda.is_available():
+            num_cuda_devices = torch.cuda.device_count()
+            if num_cuda_devices > 0:
+                device_idx = int(self.cid) % num_cuda_devices
+                self.device = torch.device(f"cuda:{device_idx}")
+                self.local_model.model.to(self.device)
+            else:
+                # Handle case where CUDA is available but no CUDA devices are found
+                raise RuntimeError("CUDA is available, but no CUDA devices are found.")
+        else:
+            # Handle case where CUDA is not available
+            self.device = torch.device("cpu")
+            self.local_model.model.to(self.device)
+
+        self.privacy_engine = PrivacyEngine(secure_mode=False)
+        self.diff_priv = diff_priv
+        self.epsilons = []
+        self.accuracies = []
+        self.losses = []
+        if self.diff_priv:
+            model, optimizer, self.trainloader = self.privacy_engine.make_private_with_epsilon(
+                module=self.local_model.model.train(),
+                optimizer=self.local_model.optimizer,
+                data_loader=self.trainloader,
+                epochs=params["train_epochs"],
+                target_epsilon=float(params["EPSILON"]),
+                target_delta= float(params["DELTA"]),
+                max_grad_norm=params["MAX_GRAD_NORM"],
+            )
+            setattr(self.local_model, "model", model)
+            setattr(self.local_model, "optimizer", optimizer)
+        self.validate()
+
+    def validate(self):
+        """Validates cid, local_model, trainloader, valloader."""
+        if not isinstance(self.cid, str):
+            raise TypeError("cid argument must be a string")
+
+        if not isinstance(self.local_model, Model):
+            raise TypeError("local_model argument must be a MEDfl.LearningManager.model.Model")
+
+        if not isinstance(self.trainloader, DataLoader):
+            raise TypeError("trainloader argument must be a torch.utils.data.dataloader")
+
+        if not isinstance(self.valloader, DataLoader):
+            raise TypeError("valloader argument must be a torch.utils.data.dataloader")
+
+        if not isinstance(self.diff_priv, bool):
+            raise TypeError("diff_priv argument must be a bool")
+
+    def get_parameters(self, config):
+        """
+        Returns the current parameters of the local model.
+
+        Args:
+            config: Configuration information.
+
+        Returns:
+            Numpy array: Parameters of the local model.
+        """
+        print(f"[Client {self.cid}] get_parameters")
+        return self.local_model.get_parameters()
+
+    def fit(self, parameters, config):
+        """
+        Fits the local model to the received parameters using federated learning.
+
+        Args:
+            parameters: Parameters received from the server.
+            config: Configuration information.
+
+        Returns:
+            Tuple: Parameters of the local model, number of training examples, and privacy information.
+        """
+        print('\n -------------------------------- \n  this is the config of the client')
+        print(f"[Client {self.cid}] fit, config: {config}")
+        # print(config['epochs'])
+        print('\n -------------------------------- \n  ')
+        self.local_model.set_parameters(parameters)
+        for _ in range(params["train_epochs"]):
+            epsilon = self.local_model.train(
+                self.trainloader,
+                epoch=_,
+                device=self.device,
+                privacy_engine=self.privacy_engine,
+                diff_priv=self.diff_priv,
+            )
+            self.epsilons.append(epsilon)
+        print(f"epsilon of client {self.cid} : eps = {epsilon}")
+        return (
+            self.local_model.get_parameters(),
+            len(self.trainloader),
+            {"epsilon": epsilon},
+        )
+
+    def evaluate(self, parameters, config):
+        """
+        Evaluates the local model on the validation data and returns the loss and accuracy.
+
+        Args:
+            parameters: Parameters received from the server.
+            config: Configuration information.
+
+        Returns:
+            Tuple: Loss, number of validation examples, and accuracy information.
+        """
+        print(f"[Client {self.cid}] evaluate, config: {config}")
+        self.local_model.set_parameters(parameters)
+        loss, accuracy , auc = self.local_model.evaluate(
+            self.valloader, device=self.device
+        )
+        self.losses.append(loss)
+        self.accuracies.append(accuracy)
+        
+        return float(loss), len(self.valloader), {"accuracy": float(accuracy)}