fix a bug of filter count.

db3d1f89 · Guolin Ke · GitHub · 5b355b00 · db3d1f89
Commit db3d1f89 authored Apr 09, 2017 by Guolin Ke Committed by GitHub Apr 09, 2017
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

src/io/dataset_loader.cpp src/io/dataset_loader.cpp +2 -2

No files found.
--- a/src/io/dataset_loader.cpp
+++ b/src/io/dataset_loader.cpp
@@ -707,7 +707,7 @@ void DatasetLoader::ConstructBinMappersFromTextData(int rank, int num_machines,
  dataset->feature_names_ = feature_names_;
  std::vector<std::unique_ptr<BinMapper>> bin_mappers(sample_values.size());
  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
  if (num_machines == 1) {