fix bug:

cross-validation last fold missing some instances
Xtra-Computing · Nov 20, 2017 · ee7e28e · ee7e28e
1 parent c59b502
commit ee7e28e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/thundersvm/model/svmmodel.cpp b/src/thundersvm/model/svmmodel.cpp
@@ -40,7 +40,11 @@ vector<float_type> SvmModel::cross_validation(DataSet dataset, SvmParam param, i
             vector<int> class_idx = dataset.original_index(i);
             auto idx_begin = class_idx.begin() + fold_test_count * k;
             auto idx_end = idx_begin;
-            while (idx_end != class_idx.end() && idx_end - idx_begin < fold_test_count) idx_end++;
+            if (k == n_fold - 1) {
+                idx_end = class_idx.end();
+            } else {
+                while (idx_end != class_idx.end() && idx_end - idx_begin < fold_test_count) idx_end++;
+            }
             for (int j: vector<int>(idx_begin, idx_end)) {
                 x_test.push_back(dataset.instances()[j]);
                 y_test.push_back(dataset.y()[j]);