Skip to content

GitLab

"python-package/vscode:/vscode.git/clone" did not exist on "581d53c6af6c72a221944bbcd656b608044ee9b0"
Unverified Commit 3d25e373 authored by James Lamb's avatar James Lamb Committed by GitHub
Browse files

fix typo in CEGB method name (#5168)

parent f68b8541
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/treelearner/cost_effective_gradient_boosting.hpp
View file @ 3d25e373
...@@ -63,7 +63,7 @@ class CostEfficientGradientBoosting { ...@@ -63,7 +63,7 @@ class CostEfficientGradientBoosting {
} }
init_ = true; 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) { int num_data_in_leaf, SplitInfo split_info) {
auto config = tree_learner_->config_; auto config = tree_learner_->config_;
double delta = double delta =
......
src/treelearner/serial_tree_learner.cpp
View file @ 3d25e373
...@@ -772,7 +772,7 @@ void SerialTreeLearner::ComputeBestSplitForFeature( ...@@ -772,7 +772,7 @@ void SerialTreeLearner::ComputeBestSplitForFeature(
new_split.feature = real_fidx; new_split.feature = real_fidx;
if (cegb_ != nullptr) { if (cegb_ != nullptr) {
new_split.gain -= 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); num_data, new_split);
} }
if (new_split.monotone_type != 0) { if (new_split.monotone_type != 0) {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment