fix bug in ExtractFeaturesFromMemory when predidct_fun_ is used (#3721)

31bc196a · shiyu1994 · GitHub · 753b0e9c · 31bc196a
Unverified Commit 31bc196a authored Jan 07, 2021 by shiyu1994 Committed by GitHub Jan 07, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

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

No files found.
--- a/src/io/dataset_loader.cpp
+++ b/src/io/dataset_loader.cpp
@@ -1205,7 +1205,7 @@ void DatasetLoader::ExtractFeaturesFromMemory(std::vector<std::string>* text_dat
      // set label
      dataset->metadata_.SetLabelAt(i, static_cast<label_t>(tmp_label));
      // free processed line:
-      text_data[i].clear();
+      ref_text_data[i].clear();
      // shrink_to_fit will be very slow in linux, and seems not free memory, disable for now
      // text_reader_->Lines()[i].shrink_to_fit();
      // push data