Initialize any_nan_ property of LinearTreeLearner (#3709)

1abc2e06 · Belinda Trotta · GitHub · a86a211b · 1abc2e06
Unverified Commit 1abc2e06 authored Jan 11, 2021 by Belinda Trotta Committed by GitHub Jan 11, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

src/treelearner/linear_tree_learner.cpp src/treelearner/linear_tree_learner.cpp +1 -0

No files found.
--- a/src/treelearner/linear_tree_learner.cpp
+++ b/src/treelearner/linear_tree_learner.cpp
@@ -34,6 +34,7 @@ void LinearTreeLearner::InitLinear(const Dataset* train_data, const int max_leav
      }
    }
  }
+  any_nan_ = false;
  for (int feat = 0; feat < train_data->num_features(); ++feat) {
    if (contains_nan_[feat]) {
      any_nan_ = true;