def setUp(self):
     # use default setting
     eggroll.init("123")
     logistic_param = LogisticParam()
     self.model = BaseLogisticRegression(logistic_param)
     self.model.header = []
     self.data_instance = self.__prepare_data()
Beispiel #2
0
 def _initialize_model(self, config):
     logistic_param = LogisticParam()
     local_model_param = LocalModelParam()
     self.logistic_param = ParamExtract.parse_param_from_config(logistic_param, config)
     local_model_param = ParamExtract.parse_param_from_config(local_model_param, config)
     self.local_model = self._create_local_model(local_model_param)
     self.model = HeteroDNNLRHost(self.local_model, self.logistic_param)
     self.model.set_data_shape(local_model_param.encode_dim)
Beispiel #3
0
 def _initialize_model(self, config):
     logistic_param = LogisticParam()
     local_model_param = LocalModelParam()
     self.logistic_param = ParamExtract.parse_param_from_config(
         logistic_param, config)
     local_model_param = ParamExtract.parse_param_from_config(
         local_model_param, config)
     self.local_model = self._create_local_model(local_model_param)
     self.model = HeteroDNNLRGuest(self.local_model, self.logistic_param)
     self.model.set_feature_shape(local_model_param.encode_dim)
     self.model.set_header(self._create_header(
         local_model_param.encode_dim))
     self.model.set_local_model_update_proxy(
         SemiEncryptedLocalModelUpdateProxy())
Beispiel #4
0
 def _initialize_model(self, config):
     logistic_param = LogisticParam()
     self.logistic_param = ParamExtract.parse_param_from_config(
         logistic_param, config)
     self.model = HeteroLRHost(self.logistic_param)
Beispiel #5
0
 def _initialize_model(self, runtime_conf_path):
     logistic_param = LogisticParam()
     self.logistic_param = ParamExtract.parse_param_from_config(
         logistic_param, runtime_conf_path)
     self.model = HomoLRArbiter(self.logistic_param)