-
Notifications
You must be signed in to change notification settings - Fork 3
/
Users.py
44 lines (36 loc) · 1.35 KB
/
Users.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
import numpy as np
from util_functions import featureUniform, gaussianFeature, fileOverWriteWarning
import json
from random import choice, randint
class User():
def __init__(self, id, theta = None):
self.id = id
self.theta = theta
class UserManager():
def __init__(self, dimension, userNum, thetaFunc, argv = None):
self.dimension = dimension
self.thetaFunc = thetaFunc
self.userNum = userNum
self.argv = argv
self.signature = "A-"+"+PA"+"+TF-"+self.thetaFunc.__name__
def simulateThetaForHomoUsers(self):
users = []
thetaVector = self.thetaFunc(self.dimension, argv=self.argv)
l2_norm = np.linalg.norm(thetaVector, ord=2)
thetaVector = thetaVector/l2_norm
for key in range(self.userNum):
users.append(User(key, thetaVector))
return users
def simulateThetaForHeteroUsers(self, global_dim):
local_dim = self.dimension-global_dim
users = []
thetaVector_g = self.thetaFunc(global_dim, argv=self.argv)
l2_norm = np.linalg.norm(thetaVector_g, ord=2)
thetaVector_g = thetaVector_g/l2_norm
for key in range(self.userNum):
thetaVector_l = self.thetaFunc(local_dim, argv=self.argv)
l2_norm = np.linalg.norm(thetaVector_l, ord=2)
thetaVector_l = thetaVector_l/l2_norm
thetaVector = np.concatenate([thetaVector_g, thetaVector_l])
users.append(User(key, thetaVector))
return users