Skip to content

Commit 2808e7f

Browse files
committed
fix case weight entries
1 parent 3facc66 commit 2808e7f

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

R/rand_forest_data.R

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -727,7 +727,7 @@ set_fit(
727727
value = list(
728728
interface = "data.frame",
729729
data = c(x = "X", y = "Y", weights = "sample.weights"),
730-
protect = c("x", "y", "weights"),
730+
protect = c("X", "Y", "weights"),
731731
func = c(pkg = "grf", fun = "probability_forest"),
732732
defaults = list(
733733
num.threads = 1
@@ -753,8 +753,8 @@ set_fit(
753753
mode = "regression",
754754
value = list(
755755
interface = "data.frame",
756-
data = c(x = "X", y = "Y", weights = "case.weights"),
757-
protect = c("x", "y", "weights"),
756+
data = c(x = "X", y = "Y", weights = "sample.weights"),
757+
protect = c("X", "Y", "weights"),
758758
func = c(pkg = "grf", fun = "regression_forest"),
759759
defaults = list(
760760
num.threads = 1
@@ -860,8 +860,8 @@ set_fit(
860860
mode = "quantile regression",
861861
value = list(
862862
interface = "data.frame",
863-
data = c(x = "X", y = "Y", weights = "case.weights"),
864-
protect = c("x", "y", "weights"),
863+
data = c(x = "X", y = "Y"),
864+
protect = c("X", "Y"),
865865
func = c(pkg = "grf", fun = "quantile_forest"),
866866
defaults = list(
867867
num.threads = 1,

0 commit comments

Comments
 (0)