def get_model_meta(self): model_meta = BoostingTreeModelMeta() model_meta.tree_meta.CopyFrom(self.booster_meta) model_meta.num_trees = self.boosting_round model_meta.quantile_meta.CopyFrom(QuantileMeta(bin_num=self.bin_num)) model_meta.boosting_strategy = self.boosting_strategy model_meta.module = "HeteroSecureBoost" meta_name = "HeteroSecureBoostingTreeHostMeta" return meta_name, model_meta
def get_model_meta(self): model_meta = BoostingTreeModelMeta() model_meta.tree_meta.CopyFrom(self.booster_meta) model_meta.learning_rate = self.learning_rate model_meta.num_trees = self.boosting_round model_meta.quantile_meta.CopyFrom(QuantileMeta(bin_num=self.bin_num)) model_meta.objective_meta.CopyFrom( ObjectiveMeta(objective=self.objective_param.objective, param=self.objective_param.params)) model_meta.use_missing = self.use_missing model_meta.zero_as_missing = self.zero_as_missing model_meta.task_type = self.task_type model_meta.n_iter_no_change = self.n_iter_no_change model_meta.tol = self.tol model_meta.boosting_strategy = self.boosting_strategy model_meta.module = "HeteroSecureBoost" meta_name = consts.HETERO_SBT_GUEST_MODEL + "Meta" return meta_name, model_meta