Searched refs:right_stats (Results 1 – 6 of 6) sorted by relevance
47 const NodeStats& left_stats, const NodeStats& right_stats) in SplitStats()50 right_node_stats(right_stats), in SplitStats()51 gain(left_stats.gain + right_stats.gain - root_stats.gain - in SplitStats()
490 int best_split_index, LeafStat* left_stats, LeafStat* right_stats) const { in InitLeafClassStats()497 auto* right_class_stats = right_stats->mutable_classification(); in InitLeafClassStats()593 int best_split_index, LeafStat* left_stats, LeafStat* right_stats) const { in InitLeafClassStats()597 auto* right_class_stats = right_stats->mutable_classification(); in InitLeafClassStats()742 const auto& right_stats = in ExtractFromProto() local743 cand.right_stats().classification().sparse_counts(); in ExtractFromProto()746 right_counts_[split_num].ExtractFromProto(right_stats); in ExtractFromProto()763 auto* right_stats = cand->mutable_right_stats() in PackToProto() local766 right_counts_[split_num].PackToProto(right_stats); in PackToProto()784 int best_split_index, LeafStat* left_stats, LeafStat* right_stats) const { in InitLeafClassStats()[all …]
191 LeafStat* right_stats) const = 0;313 LeafStat* right_stats) const override;378 LeafStat* right_stats) const override;471 LeafStat* right_stats) const override;
71 model_op_->ExportModel(best->right_stats(), right_leaf); in SplitNode()
276 NodeStats right_stats = state->ComputeNodeStats(right_gradient_stats); in ComputeNormalDecisionTree() local277 if (left_stats.gain + right_stats.gain > best_gain) { in ComputeNormalDecisionTree()278 best_gain = left_stats.gain + right_stats.gain; in ComputeNormalDecisionTree()280 best_right_node_stats = right_stats; in ComputeNormalDecisionTree()848 NodeStats right_stats = state->ComputeNodeStats(right_gradient_stats); in ComputeNormalDecisionTree() local850 left_stats.gain + right_stats.gain > best_gain) { in ComputeNormalDecisionTree()851 best_gain = left_stats.gain + right_stats.gain; in ComputeNormalDecisionTree()853 best_right_node_stats = right_stats; in ComputeNormalDecisionTree()
88 LeafStat right_stats = 5; field