def __init__(self):
super(HeteroKmeansArbiter, self).__init__()
self.model_param = KmeansParam()
# self.dist_aggregator = secure_sum_aggregator.Server(enable_secure_aggregate=False)
# self.cluster_dist_aggregator = secure_sum_aggregator.Server(enable_secure_aggregate=False)
self.DBI = 0
self.aggregator = table_aggregator.Server(enable_secure_aggregate=True)
def fit_split_points(self, data=None):
if self.aggregator is None:
self.aggregator = table_aggregator.Server(
enable_secure_aggregate=True)
self.get_total_count()
self.get_min_max()
self.get_missing_count()
self.query_values()
def fit_split_points(self, data=None):
if self.aggregator is None:
self.aggregator = table_aggregator.Server(
enable_secure_aggregate=False)
self.get_total_count()
self.get_min_max()
self.get_missing_count()
self.set_suffix(-1)
self.query_values()
n_iter = 0
while n_iter < self.params.max_iter:
self.set_suffix(n_iter)
is_converge = self.transfer_variable.is_converge.get(
suffix=self.suffix)[0]
if is_converge:
break
self.query_values()
n_iter += 1