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