Skip to content

Commit

Permalink
fix typo in CEGB method name (#5168)
Browse files Browse the repository at this point in the history
  • Loading branch information
jameslamb authored Apr 24, 2022
1 parent f68b854 commit 3d25e37
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion src/treelearner/cost_effective_gradient_boosting.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -63,7 +63,7 @@ class CostEfficientGradientBoosting {
}
init_ = true;
}
double DetlaGain(int feature_index, int real_fidx, int leaf_index,
double DeltaGain(int feature_index, int real_fidx, int leaf_index,
int num_data_in_leaf, SplitInfo split_info) {
auto config = tree_learner_->config_;
double delta =
Expand Down
2 changes: 1 addition & 1 deletion src/treelearner/serial_tree_learner.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -772,7 +772,7 @@ void SerialTreeLearner::ComputeBestSplitForFeature(
new_split.feature = real_fidx;
if (cegb_ != nullptr) {
new_split.gain -=
cegb_->DetlaGain(feature_index, real_fidx, leaf_splits->leaf_index(),
cegb_->DeltaGain(feature_index, real_fidx, leaf_splits->leaf_index(),
num_data, new_split);
}
if (new_split.monotone_type != 0) {
Expand Down

0 comments on commit 3d25e37

Please sign in to comment.