diff --git a/src/main/java/ca/joeltherrien/randomforest/covariates/numeric/NumericCovariate.java b/src/main/java/ca/joeltherrien/randomforest/covariates/numeric/NumericCovariate.java index 1f00d51..9b23d08 100644 --- a/src/main/java/ca/joeltherrien/randomforest/covariates/numeric/NumericCovariate.java +++ b/src/main/java/ca/joeltherrien/randomforest/covariates/numeric/NumericCovariate.java @@ -52,7 +52,7 @@ public final class NumericCovariate implements Covariate { if(number == 0){ dataIterator = new UniqueValueIterator<>(sortedDataIterator); } - else{ // random splitting; we will not weight based on how many times a Row appears in the bootstrap sample + else{ final TreeSet indexSet = new TreeSet<>(); final int maxIndex = data.size();