fix #984 .

microsoft · Oct 27, 2017 · d4bec4f · d4bec4f
1 parent 7577a36
commit d4bec4f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/boosting/dart.hpp b/src/boosting/dart.hpp
@@ -95,7 +95,7 @@ class DART: public GBDT {
         }
         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 @@ class DART: public GBDT {
         }
         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);
           }
         }
       }