prise6
/
aVirtualTwins
mirror of https://github.com/prise6/aVirtualTwins.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Fix list bug

This commit is contained in:
prise6 2015-07-25 02:59:16 +02:00
parent 12e4ab74fd
commit 263e1d6d22
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
R/forest.wrapper.R
View File

@ -26,12 +26,12 @@ vt.forest <- function(forest.type = "one", vt.data, interactions = T, method = "
params <- list(...)
if (forest.type == "one"){
if(! "rf" %in% names(params) ){
if(! "model" %in% names(params) ){
rf <- randomForest(x = vt.data$getX(interactions = interactions, trt = NULL),
y = vt.data$getY(),
...)
} else{
rf <- params["rf"]
rf <- params[["model"]]
}
vt.difft <- VT.forest.one(vt.object = vt.data, model = rf, interactions = interactions, method = method)
@ -41,19 +41,19 @@ vt.forest <- function(forest.type = "one", vt.data, interactions = T, method = "
y = vt.data$getY(1),
...)
} else
rf_trt1 <- params["model_trt1"]
rf_trt1 <- params[["model_trt1"]]
if(! "model_trt0" %in% names(params) ){
rf_trt0 <- randomForest(x = vt.data$getX(trt = 1, interactions = interactions),
y = vt.data$getY(1),
...)
} else
rf_trt0 <- params["rf_trt0"]
rf_trt0 <- params[["model_trt0"]]
vt.difft <- VT.forest.double(vt.object = vt.data, model_trt1 = rf_trt1, model_trt0 = rf_trt0, method = method, ...)
vt.difft <- VT.forest.double(vt.object = vt.data, model_trt1 = rf_trt1, model_trt0 = rf_trt0, method = method)
} else if (forest.type == "fold"){
fold <- ifelse(! "fold" %in% names(params) , 5, params["fold"])
fold <- ifelse(! "ratio" %in% names(params) , 1, params["ratio"])
fold <- ifelse(! "fold" %in% names(params) , 5, as.numeric(params["fold"]))
ratio <- ifelse(! "ratio" %in% names(params) , 1, as.numeric(params["ratio"]))
vt.difft <- aVirtualTwins:::VT.forest.fold(vt.object = vt.data, fold = fold, ratio = ratio,
interactions = interactions, method = method)