fix creating folds in sdm()

tiemvanderdeure · Nov 28, 2023 · 7d79e40 · 7d79e40
1 parent feda310
commit 7d79e40
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ensemble.jl b/src/ensemble.jl
@@ -116,7 +116,7 @@ function sdm(
 
     trained_models = mapreduce(vcat, keys(resamplers_)) do resampler_key
         resampler = resamplers_[resampler_key]
-        folds = MLJ.MLJBase.train_test_pairs(resampler, 1:1100, response_values) ## get indices
+        folds = MLJ.MLJBase.train_test_pairs(resampler, 1:n_total, response_values) ## get indices
         mapreduce(vcat, keys(models_)) do model_key
             model = models_[model_key]
             map(enumerate(folds)) do (f, (train, test))