def test_feature_binning(self): binning_guest = HeteroFeatureBinningGuest() guest_param = self._make_param_dict('fit') binning_guest.run(guest_param, self.args) result_data = binning_guest.save_data() local_data = result_data.collect() print("data in fit") for k, v in local_data: print("k: {}, v: {}".format(k, v.features)) guest_model = {self.model_name: binning_guest.export_model()} guest_args = { 'data': { self.model_name: { 'data': self.table } }, 'model': guest_model } binning_guest = HeteroFeatureBinningGuest() guest_param = self._make_param_dict('transform') binning_guest.run(guest_param, guest_args) result_data = binning_guest.save_data() local_data = result_data.collect() print("data in transform") for k, v in local_data: print("k: {}, v: {}".format(k, v.features))
def run_data(self, table_args, run_type='fit'): if self.binning_obj is not None: return self.binning_obj if self.role == GUEST: binning_obj = HeteroFeatureBinningGuest() else: binning_obj = HeteroFeatureBinningHost() guest_param = self._make_param_dict(run_type) binning_obj.run(guest_param, table_args) self.binning_obj = binning_obj return binning_obj
def run_data(self, table_args, run_type='fit'): if self.binning_obj is not None: return self.binning_obj if self.role == GUEST: binning_obj = HeteroFeatureBinningGuest() else: binning_obj = HeteroFeatureBinningHost() # param_obj = FeatureBinningParam(method=consts.QUANTILE) # binning_obj.model_param = param_obj guest_param = self._make_param_dict(run_type) binning_obj.run(guest_param, table_args) print("current binning method: {}, split_points: {}".format( binning_obj.model_param.method, binning_obj.binning_obj.split_points)) self.binning_obj = binning_obj return binning_obj