-> as.factor (#101)

grf-labs · Jun 24, 2021 · 22d2de9 · 22d2de9
1 parent 5a23a51
commit 22d2de9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/r-package/policytree/vignettes/policytree.Rmd b/r-package/policytree/vignettes/policytree.Rmd
@@ -57,7 +57,7 @@ head(data.frame(data)[1:6])
 
 X <- data$X
 Y <- data$Y
-W <- data$action
+W <- as.factor(data$action)
 
 multi.forest <- multi_arm_causal_forest(X, Y, W)