Skip to content

GitLab

Commit 5c5dce37 authored by Guolin Ke's avatar Guolin Ke
Browse files

clean code

parent 4f232570
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/treelearner/data_parallel_tree_learner.cpp
View file @ 5c5dce37
...@@ -166,7 +166,7 @@ void DataParallelTreeLearner::FindBestThresholds() { ...@@ -166,7 +166,7 @@ void DataParallelTreeLearner::FindBestThresholds() {
smaller_leaf_histogram_array_[feature_index].FindBestThreshold( smaller_leaf_histogram_array_[feature_index].FindBestThreshold(
smaller_leaf_splits_->sum_gradients(), smaller_leaf_splits_->sum_gradients(),
smaller_leaf_splits_->sum_hessians(), smaller_leaf_splits_->sum_hessians(),
smaller_leaf_splits_->num_data_in_leaf(), GetGlobalDataCountInLeaf(smaller_leaf_splits_->LeafIndex()),
&smaller_leaf_splits_->BestSplitPerFeature()[feature_index]); &smaller_leaf_splits_->BestSplitPerFeature()[feature_index]);
// only root leaf // only root leaf
...@@ -180,7 +180,7 @@ void DataParallelTreeLearner::FindBestThresholds() { ...@@ -180,7 +180,7 @@ void DataParallelTreeLearner::FindBestThresholds() {
larger_leaf_histogram_array_[feature_index].FindBestThreshold( larger_leaf_histogram_array_[feature_index].FindBestThreshold(
larger_leaf_splits_->sum_gradients(), larger_leaf_splits_->sum_gradients(),
larger_leaf_splits_->sum_hessians(), larger_leaf_splits_->sum_hessians(),
larger_leaf_splits_->num_data_in_leaf(), GetGlobalDataCountInLeaf(larger_leaf_splits_->LeafIndex()),
&larger_leaf_splits_->BestSplitPerFeature()[feature_index]); &larger_leaf_splits_->BestSplitPerFeature()[feature_index]);
} }
......
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