Searched refs:split_num (Results 1 – 2 of 2) sorted by relevance
60 void GrowStats::RemoveSplit(int split_num) { in RemoveSplit() argument61 splits_.erase(splits_.begin() + split_num); in RemoveSplit()62 evaluators_.erase(evaluators_.begin() + split_num); in RemoveSplit()63 RemoveSplitStats(split_num); in RemoveSplit()432 int split_num = 0; in ExtractFromProto() local438 mutable_left_count(split_num, i) = val; in ExtractFromProto()439 MaybeInitializeRunningCount(split_num, val); in ExtractFromProto()441 ++split_num; in ExtractFromProto()456 for (int split_num = 0; split_num < num_splits(); ++split_num) { in PackToProto() local458 *cand->mutable_split() = splits_[split_num]; in PackToProto()[all …]
52 const decision_trees::BinaryNode& Split(int split_num) const { in Split() argument53 return splits_[split_num]; in Split()74 void RemoveSplit(int split_num);92 virtual void RemoveSplitStats(int split_num) = 0;137 void RemoveSplitStats(int split_num) override {} in RemoveSplitStats() argument319 void ClassificationRemoveSplitStats(int split_num) override { in ClassificationRemoveSplitStats() argument320 left_counts_.erase(left_counts_.begin() + num_outputs_ * split_num, in ClassificationRemoveSplitStats()321 left_counts_.begin() + num_outputs_ * (split_num + 1)); in ClassificationRemoveSplitStats()384 void ClassificationRemoveSplitStats(int split_num) override { in ClassificationRemoveSplitStats() argument385 left_counts_.erase(left_counts_.begin() + split_num, in ClassificationRemoveSplitStats()[all …]