Skip to content

GitLab

"src/io/vscode:/vscode.git/clone" did not exist on "5ffd7571193f7814eb2e30a0bc648e648bad0bf3"
Commit db3d1f89 authored by Guolin Ke's avatar Guolin Ke Committed by GitHub
Browse files

fix a bug of filter count.

parent 5b355b00
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/io/dataset_loader.cpp
View file @ db3d1f89
...@@ -707,7 +707,7 @@ void DatasetLoader::ConstructBinMappersFromTextData(int rank, int num_machines, ...@@ -707,7 +707,7 @@ void DatasetLoader::ConstructBinMappersFromTextData(int rank, int num_machines,
dataset->feature_names_ = feature_names_; dataset->feature_names_ = feature_names_;
std::vector<std::unique_ptr<BinMapper>> bin_mappers(sample_values.size()); std::vector<std::unique_ptr<BinMapper>> bin_mappers(sample_values.size());
const data_size_t filter_cnt = static_cast<data_size_t>( const data_size_t filter_cnt = static_cast<data_size_t>(
static_cast<double>(io_config_.min_data_in_leaf* sample_values.size()) / dataset->num_data_); static_cast<double>(io_config_.min_data_in_leaf* sample_data.size()) / dataset->num_data_);
// start find bins // start find bins
if (num_machines == 1) { if (num_machines == 1) {
......
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