Esempio n. 1
0
 def __init__(self):
     super(HomoBoostingClient, self).__init__()
     self.transfer_inst = HomoBoostingTransferVariable()
     self.aggregator = HomoBoostClientAggregator()
     self.model_param = HomoSecureBoostParam()
     self.binning_obj = HomoFeatureBinningClient()
     self.mode = consts.H**O
Esempio n. 2
0
 def __init__(self):
     super(HomoSecureBoostingTreeArbiter, self).__init__()
     self.model_name = 'HomoSecureBoost'
     self.tree_param = None  # decision tree param
     self.use_missing = False
     self.zero_as_missing = False
     self.cur_epoch_idx = -1
     self.grad_and_hess = None
     self.feature_importances_ = {}
     self.model_param = HomoSecureBoostParam()
Esempio n. 3
0
    def __init__(self):
        super(HomoSecureBoostingTreeClient, self).__init__()
        self.model_name = 'HomoSecureBoost'
        self.tree_param = None  # decision tree param
        self.use_missing = False
        self.zero_as_missing = False
        self.cur_epoch_idx = -1
        self.grad_and_hess = None
        self.feature_importance_ = {}
        self.model_param = HomoSecureBoostParam()

        # memory back end
        self.backend = consts.DISTRIBUTED_BACKEND
        self.bin_arr, self.sample_id_arr = None, None