Merge branch 'main' of github.com:ampel-leipzig/rusranger

DESCRIPTION

@@ -1,7 +1,7 @@
 Package: rusranger
 Title:
     Modified ranger implementation to support random-under-sampling
-Version: 0.0.1
+Version: 0.0.2
 Date: 2022-03-24
 Description:
     The random forest implementation of the ranger package is modified to

NEWS.md

@@ -1,5 +1,9 @@
 # rusranger 0.0
 
+## Changes in 0.0.2
+
+- Fix `.caseweights` for 0/1 binary class coding (instead of 1/2).
+
 ## Changes in 0.0.1
 
 - Initial version.

R/utils-rusranger.R

@@ -28,6 +28,7 @@
 #' @return `double(length(y))` with case weights
 #' @noRd
 .caseweights <- function(y, replace = FALSE) {
+    y <- as.factor(y)
     if (isTRUE(replace))
         ## these case weights work just for replace = TRUE (=> bootstrap)
         c(1 / table(y))[y]

tests/testthat/test_utils-rusranger.R

@@ -23,6 +23,12 @@ test_that(".caseweights", {
         .caseweights(cl, replace = TRUE),
         setNames(rep(c(0.5, 0.25), c(2, 4)), cl)
     )
+    ## fix error if classes are numerics other as 1:2
+    cl <- rep(0:1, c(2, 4))
+    expect_equal(
+        unname(.caseweights(cl, replace = FALSE)),
+        unname(.caseweights(cl + 1, replace = FALSE)),
+    )
 })
 
 test_that(".samplefraction", {