pryvx 2.4.0__py3-none-any.whl → 2.4.2__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,10 +2,11 @@ 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)
 
@@ -13,6 +14,13 @@ def train(features, labels):
 
     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
+
 
 def send_params(serialized_model, connection_url):
 

pryvx/fl_ensemble.py

@@ -0,0 +1,16 @@
+from sklearn.ensemble import VotingClassifier
+
+# estimators = [('client_1', model_1), ('client_2', model_2), ('client_3', model_3)]
+
+def federated_ensemble(estimators, X_test):
+    # Create an ensemble using soft voting
+    ensemble_model = VotingClassifier(
+        estimators=estimators,
+        voting='soft'  # Soft voting (probability-based)
+    )
+
+    # Make predictions
+    y_pred_ensemble = ensemble_model.predict(X_test)
+
+    return y_pred_ensemble
+

pryvx/fl_server.py

@@ -3,21 +3,39 @@ 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:
+            # Deserialize model received from client
+            model = pickle.loads(request.params)
+
+            timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
+
+            # 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_{timestamp}.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/phe.py

@@ -26,6 +26,14 @@ class PHE:
         private_key = (lambda_n, mu)
         return public_key, private_key
 
+    @staticmethod
+    def encode(value, scale_factor=1000):
+        return int(round(value * scale_factor))
+
+    @staticmethod
+    def decode(value, scale_factor=1000):
+        return value / scale_factor
+
     @staticmethod
     def encrypt(public_key, plaintext, r=None):
         n, g = public_key
@@ -56,6 +64,22 @@ class PHE:
         n2 = n * n
         return (c1 * c2) % n2
     
+    @staticmethod
+    def homomorphic_add_plaintext(ciphertext, plaintext, public_key):
+        n, g = public_key
+        n2 = n * n
+        c_plain = pow(g, plaintext, n2)
+        return (ciphertext * c_plain) % n2
+    
+    @staticmethod
+    def homomorphic_sub_plaintext(ciphertext, plaintext, public_key):
+        n, g = public_key
+        n2 = n * n
+        
+        # Encrypt the plaintext as its negative (-plaintext) mod n
+        c_plain_neg = pow(g, -plaintext % n, n2)  # Modular inverse for subtraction
+        return (ciphertext * c_plain_neg) % n2
+    
     @staticmethod
     def homomorphic_sub(c1, c2, public_key):
         n, _ = public_key

{pryvx-2.4.0.dist-info → pryvx-2.4.2.dist-info}/METADATA

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

pryvx-2.4.2.dist-info/RECORD

@@ -0,0 +1,16 @@
+pryvx/__init__.py,sha256=QvpYISJ8MH7XnbxC5iTMTycq7Svsld0NIbMozcndYjo,111
+pryvx/fl_client.py,sha256=axJKdQgXDMs1FT8rgI2HzG8keyEG__U9Y4GtiWWeHO4,986
+pryvx/fl_ensemble.py,sha256=cUPPx3QpnalRKtIBtoE4PCTk34E7940W7ZxGtUisDgo,475
+pryvx/fl_server.py,sha256=bJ_dm-UZvmq-JUB58b_F_V5-IsmB3eC2RS0JjF5CPs8,1739
+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.2.dist-info/LICENSE,sha256=pbIMXbaorAIVW-fDch2tvtZRkVA3mz-UnXQqeCf4LDg,1086
+pryvx-2.4.2.dist-info/METADATA,sha256=axWXanMU4AAmUsGygBZlLx7BYDT6v6LNqW9pGaw02Rk,1066
+pryvx-2.4.2.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
+pryvx-2.4.2.dist-info/top_level.txt,sha256=1iHoeevu_FoFjdPg8HyMlgvHNUm1--9QEpFMaJEa4hw,6
+pryvx-2.4.2.dist-info/RECORD,,

pryvx-2.4.0.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=NLYaLFCNN5aQfL0kC1YTUi962e930p2ae5dw3flJ0qo,2178
-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.0.dist-info/LICENSE,sha256=pbIMXbaorAIVW-fDch2tvtZRkVA3mz-UnXQqeCf4LDg,1086
-pryvx-2.4.0.dist-info/METADATA,sha256=PhZ10Rw7ngp_mhVp2Au9YED-g_KMVKtHmUpDp4e0Zqg,1066
-pryvx-2.4.0.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
-pryvx-2.4.0.dist-info/top_level.txt,sha256=1iHoeevu_FoFjdPg8HyMlgvHNUm1--9QEpFMaJEa4hw,6
-pryvx-2.4.0.dist-info/RECORD,,