fixed CvBoost

481d0b2c · Maria Dimashova · 428aef52 · 481d0b2c
Commit 481d0b2c authored Apr 30, 2011 by Maria Dimashova
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

boost.cpp modules/ml/src/boost.cpp +2 -0

No files found.
--- a/modules/ml/src/boost.cpp
+++ b/modules/ml/src/boost.cpp
@@ -1072,6 +1072,8 @@ CvBoost::train( const CvMat* _train_data, int _tflag,
        cvSeqPush( weak, &tree );
        update_weights( tree );
        trim_weights();
+        if( cvCountNonZero(subsample_mask) == 0 )
+            break;
    }

    get_active_vars(); // recompute active_vars* maps and condensed_idx's in the splits.