Skip to content

GitLab

"...git@developer.sourcefind.cn:tianlh/lightgbm-dcu.git" did not exist on "bbeecc09af946c5ff9b84d1ada4749a9f26bca31"
Commit db3d1f89 authored by Guolin Ke's avatar Guolin Ke Committed by GitHub
Browse files

fix a bug of filter count.

parent 5b355b00
Hide 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) {
...@@ -1031,4 +1031,4 @@ std::string DatasetLoader::CheckCanLoadFromBin(const char* filename) { ...@@ -1031,4 +1031,4 @@ std::string DatasetLoader::CheckCanLoadFromBin(const char* filename) {
} }
} }
\ No newline at end of file
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