fix #596

582ded51 · Guolin Ke · 591efb6d · 582ded51
Commit 582ded51 authored Jun 07, 2017 by Guolin Ke
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

src/application/predictor.hpp src/application/predictor.hpp +3 -1

No files found.
--- a/src/application/predictor.hpp
+++ b/src/application/predictor.hpp
@@ -172,7 +172,9 @@ private:
    } else {
      int loop_size = static_cast<int>(features.size());
      for (int i = 0; i < loop_size; ++i) {
-        pred_buf[features[i].first] = 0.0f;
+        if (features[i].first < num_feature_) {
+          pred_buf[features[i].first] = 0.0f;
+        }
      }
    }
  }