pryvx 2.4.1__py3-none-any.whl → 2.4.3__py3-none-any.whl

pryvx/__init__.py

@@ -1,4 +1,5 @@
 from .psi import OPRF
 from .phe import PHE
 from .smpc import SMPC
-from .gdp import GDP
+from .gdp import GDP
+from .ldp import LDP

pryvx/fl_client.py

@@ -2,26 +2,35 @@ import grpc
 from pryvx import pryvx_pb2
 from pryvx import pryvx_pb2_grpc
 from sklearn.linear_model import LogisticRegression
+from sklearn.ensemble import RandomForestClassifier
 import pickle
 
 
-def train(features, labels):
+def train_logistic_classifier(features, labels):
     model = LogisticRegression()
     model.fit(features, labels)
-
     serialized_model = pickle.dumps(model)
+    return serialized_model, model
 
-    return serialized_model
 
+def train_random_forest_classifier(features, labels):
+    model = RandomForestClassifier(n_estimators=100, random_state=42)
+    model.fit(features, labels)
+    serialized_model = pickle.dumps(model)
+    return serialized_model, model
 
-def send_params(serialized_model, connection_url):
+
+def send_params(serialized_model, connection_url, client_id):
 
     with grpc.insecure_channel(connection_url) as channel:
         stub = pryvx_pb2_grpc.ModelServiceStub(channel)
 
+        # Attach metadata (client ID) in the request
+        metadata = (("client_id", client_id),)
+
         model_params = pryvx_pb2.ModelParams(params=serialized_model)
 
-        response = stub.SendModelParams(model_params)
+        response = stub.SendModelParams(model_params, metadata=metadata)
 
-        return "Model Params sent to server"
+        return f"✅ Model Params sent to server from {client_id}"
 

pryvx/fl_ensemble.py

@@ -0,0 +1,20 @@
+import numpy as np
+
+def federated_ensemble(estimators, X_test):
+    """
+    Perform soft voting ensemble on a list of trained estimators.
+    
+    :param estimators: List of trained models that support `predict_proba`
+    :param X_test: Test dataset (features only)
+    :return: Final predictions after soft voting
+    """
+    if not estimators:
+        raise ValueError("No models provided for ensemble.")
+
+    # Sum all probability predictions
+    avg_probs = sum(model.predict_proba(X_test) for model in estimators) / len(estimators)
+
+    # Convert probabilities to final predictions (highest probability wins)
+    y_pred_ensemble = np.argmax(avg_probs, axis=1)
+    
+    return y_pred_ensemble

pryvx/fl_server.py

@@ -3,21 +3,41 @@ from pryvx import pryvx_pb2
 from pryvx import pryvx_pb2_grpc
 import pickle
 from concurrent import futures
+import os
+import datetime
+
+# Directory to store received models
+MODEL_SAVE_PATH = "received_models/"
+os.makedirs(MODEL_SAVE_PATH, exist_ok=True)
 
-# Server
 class ModelServicer(pryvx_pb2_grpc.ModelServiceServicer):
     def __init__(self):
-        self.client_params = {}
+        self.client_models = {}  # Store models from clients
 
     def SendModelParams(self, request, context):
-        # Deserialize the model
-        loaded_model = pickle.loads(request.params)
-
-        # save model to gcp storage bucket
-
-        print("Received model params from client")
-
-        return pryvx_pb2.ModelParams(params=request.params)
+        try:
+            # Extract metadata (client ID)
+            client_id = dict(context.invocation_metadata()).get("client_id", "unknown_client")
+            
+            # Deserialize model received from client
+            model = pickle.loads(request.params)
+
+            # Assign a unique client ID
+            client_id = f"client_{len(self.client_models) + 1}"
+            self.client_models[client_id] = model
+            
+            # Save model to disk
+            model_filename = os.path.join(MODEL_SAVE_PATH, f"{client_id}_model.pkl")
+            with open(model_filename, "wb") as f:
+                
+                pickle.dump(model, f)
+
+            print(f"✅ Received and saved model from {client_id}")
+
+            return pryvx_pb2.ModelResponse(message=f"Model from {client_id} received and saved.")
+        except Exception as e:
+            print(f"❌ Error processing model: {e}")
+            return pryvx_pb2.ModelResponse(message="Failed to process model")
 
 
 def start_server():

pryvx/gdp.py

@@ -1,15 +1,27 @@
 import numpy as np
 
-def laplace_mechanism(true_value, sensitivity, epsilon):
+# Laplace Mechanism for continuous data.
+# Gaussian Mechanism for continuous data (commonly used in GDP).
+    
+def laplace_mechanism(query_result, sensitivity, epsilon):
     """Applies the Laplace mechanism for GDP."""
     scale = sensitivity / epsilon
     noise = np.random.laplace(0, scale, 1)
-    return true_value + noise[0]
+    return query_result + noise[0]
 
+def gaussian_mechanism(query_result, epsilon=0.5, delta=1e-5, sensitivity=1.0):
+    """Adds Gaussian noise to a global query result."""
+    sigma = np.sqrt(2 * np.log(1.25 / delta)) * sensitivity / epsilon
+    noise = np.random.normal(0, sigma)
+    return query_result + noise
 
 class GDP:
     @staticmethod
     def add_noise(query_result, sensitivity, epsilon):
         return laplace_mechanism(query_result, sensitivity, epsilon)
+    
+    @staticmethod
+    def add_gaussian_noise(query_result):
+        return gaussian_mechanism(query_result)
 
 

pryvx/ldp.py

@@ -0,0 +1,26 @@
+import numpy as np
+import random
+
+# Laplace Mechanism (common for numeric data)
+# Randomized Response (common for categorical/binary data)
+
+def laplace_mechanism(value, epsilon, sensitivity=1.0):
+    """Adds Laplace noise to a numeric value for LDP."""
+    scale = sensitivity / epsilon
+    noise = np.random.laplace(0, scale)
+    return value + noise
+
+def randomized_response(value, epsilon):
+    """Implements randomized response for binary data."""
+    p = np.exp(epsilon) / (1 + np.exp(epsilon))
+    return value if random.random() < p else 1 - value
+
+class LDP:
+    @staticmethod
+    def add_numerical_noise(value, epsilon):
+        return laplace_mechanism(value, epsilon)
+    
+    def add_categorical_noise(value, epsilon):
+        return randomized_response(value, epsilon)
+
+

{pryvx-2.4.1.dist-info → pryvx-2.4.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pryvx
-Version: 2.4.1
+Version: 2.4.3
 Summary: A comprehensive package for privacy-enhancing technologies
 Home-page: UNKNOWN
 Author: PryvX (Jayesh Kenaudekar)

pryvx-2.4.3.dist-info/RECORD

@@ -0,0 +1,16 @@
+pryvx/__init__.py,sha256=QvpYISJ8MH7XnbxC5iTMTycq7Svsld0NIbMozcndYjo,111
+pryvx/fl_client.py,sha256=gQNjv99tDDjaGj0JTdAVomefqR2F3_xNLcnHJ6aGZZo,1152
+pryvx/fl_ensemble.py,sha256=lvgvd-zJDSBh-RyMAjTyJZwu0eFiUNXfCU7J2Vk7g0g,715
+pryvx/fl_server.py,sha256=JqYtlLtLTcKxrNB8oY-hdZmG8VajS7mIqAY3OjOUxmU,1822
+pryvx/gdp.py,sha256=ToLtWNvBKtSs-3Na1pGe_J6dWvBxXhV2raUJkNtKMg4,925
+pryvx/ldp.py,sha256=WxBQLnZ_b2wNV1rlN7rtKXaQ5waUPPvnRus-4tUnCB8,810
+pryvx/phe.py,sha256=HN70aUNTq0w9vRPMaNGfTnNi00gMQ1B7R7Cm6r6rheE,2974
+pryvx/pryvx_pb2.py,sha256=-ce40VMW0nQcjyhAJy1BvI32wtWNm425LSIExspK_qg,1200
+pryvx/pryvx_pb2_grpc.py,sha256=pRQKblTNIVEwHsMQmhC-HGJFzsLh0sdOzQ-ksFYVK-A,2437
+pryvx/psi.py,sha256=V0BUJwYfiFWhKtEL27q45yxY7U-I7B4olKQhaGbgN50,1346
+pryvx/smpc.py,sha256=FZcE3gZo8WUVPmcBHSgP8iOkWt3AKcBLSLCtKczy8hs,2247
+pryvx-2.4.3.dist-info/LICENSE,sha256=pbIMXbaorAIVW-fDch2tvtZRkVA3mz-UnXQqeCf4LDg,1086
+pryvx-2.4.3.dist-info/METADATA,sha256=Ytztu3-NrHGnG2XkIBalwkZSMQiP5d7UzU0V5C8wpdQ,1066
+pryvx-2.4.3.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
+pryvx-2.4.3.dist-info/top_level.txt,sha256=1iHoeevu_FoFjdPg8HyMlgvHNUm1--9QEpFMaJEa4hw,6
+pryvx-2.4.3.dist-info/RECORD,,

pryvx-2.4.1.dist-info/RECORD

@@ -1,14 +0,0 @@
-pryvx/__init__.py,sha256=A6GcpkYgwgf_HOrneIhGs3iTsiEyvd9vYEMsmrMUqWQ,89
-pryvx/fl_client.py,sha256=p8QPlv5-CjTCwfzN0QkPr7bjKJEbU3c9Aa3HhpvZNJE,677
-pryvx/fl_server.py,sha256=9fUqZfA9RZOe0fM7FYnk_8qLNw6VMj-x8tzeGXZpgKA,890
-pryvx/gdp.py,sha256=-HK3Xml-huOdfz2SSmBvpgca0iU9-RwCiaD6ZyMq7bY,408
-pryvx/phe.py,sha256=HN70aUNTq0w9vRPMaNGfTnNi00gMQ1B7R7Cm6r6rheE,2974
-pryvx/pryvx_pb2.py,sha256=-ce40VMW0nQcjyhAJy1BvI32wtWNm425LSIExspK_qg,1200
-pryvx/pryvx_pb2_grpc.py,sha256=pRQKblTNIVEwHsMQmhC-HGJFzsLh0sdOzQ-ksFYVK-A,2437
-pryvx/psi.py,sha256=V0BUJwYfiFWhKtEL27q45yxY7U-I7B4olKQhaGbgN50,1346
-pryvx/smpc.py,sha256=FZcE3gZo8WUVPmcBHSgP8iOkWt3AKcBLSLCtKczy8hs,2247
-pryvx-2.4.1.dist-info/LICENSE,sha256=pbIMXbaorAIVW-fDch2tvtZRkVA3mz-UnXQqeCf4LDg,1086
-pryvx-2.4.1.dist-info/METADATA,sha256=omTieHCpjPJe3yoCkJZ0SoX2sFj0IUMRfHIQB-8TUzI,1066
-pryvx-2.4.1.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
-pryvx-2.4.1.dist-info/top_level.txt,sha256=1iHoeevu_FoFjdPg8HyMlgvHNUm1--9QEpFMaJEa4hw,6
-pryvx-2.4.1.dist-info/RECORD,,