fix #984 .

d4bec4f2 · Guolin Ke · 7577a363 · d4bec4f2
Commit d4bec4f2 authored Oct 27, 2017 by Guolin Ke
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

src/boosting/dart.hpp src/boosting/dart.hpp +2 -2

No files found.
--- a/src/boosting/dart.hpp
+++ b/src/boosting/dart.hpp
@@ -95,7 +95,7 @@ private:
        }
        for (int i = 0; i < iter_; ++i) {
          if (random_for_drop_.NextFloat() < drop_rate * tree_weight_[i] * inv_average_weight) {
-            drop_index_.push_back(i);
+            drop_index_.push_back(num_init_iteration_ + i);
          }
        }
      } else {
@@ -104,7 +104,7 @@ private:
        }
        for (int i = 0; i < iter_; ++i) {
          if (random_for_drop_.NextFloat() < drop_rate) {
-            drop_index_.push_back(i);
+            drop_index_.push_back(num_init_iteration_ + i);
          }
        }
      }