library(randomForest)
library(forestFloor)
#X y could be a training set
X = data.frame(numeric = c(1,5,2,7,-4.3),
factor1 = factor(c("jim","freddy","marley","marley","alfred")),
factor2 = factor(c("jill","ann","liz","leila","vicky")))
y = factor(1:5)
set.seed(1)
rf = randomForest(X,y,keep.inbag=TRUE,ntree=7)
#should not raise any error
test = data.frame(numeric = rnorm(5),
factor1 = factor(c("jim","jim","jim","freddy","freddy")),
factor2 = factor(c("jill","jill","vicky","leila","vicky"))
)
out = Xtestmerger(X,test,inbag=rf$inbag,y=y)
Run the code above in your browser using DataLab