Searched refs:is_regression (Results 1 – 6 of 6) sorted by relevance
55 self.is_regression = params.regression83 output_size = 1 if self.is_regression else self.params.num_classes113 if self.is_regression:
298 param_proto_.is_regression() ? (std::max(1, label_dim)) : 1; in Compute()416 void FinalizeLeaf(bool is_regression, bool drop_final_class, in FinalizeLeaf() argument420 if (is_regression) { in FinalizeLeaf()486 FinalizeLeaf(param_proto_.is_regression(), in Compute()
250 if (!param_proto_.is_regression() && sum > 0 && sum != 1) { in set_output_value()351 param_proto_.is_regression() ? (std::max(1, label_dim)) : 1; in Compute()
57 output_size = 1 if self.is_regression else self.params.num_classes
130 bool is_regression = 8; field
92 proto.is_regression = params.regression