diff --git a/R/Xtestmerger.R b/R/Xtestmerger.R index 3f61dd2..f20860a 100644 --- a/R/Xtestmerger.R +++ b/R/Xtestmerger.R @@ -22,8 +22,8 @@ Xtestmerger = function(X,test,inbag=NULL,y=NULL) { #return unmatched levels of test in X FUN = function(xl,tl) tl[which(is.na(match(tl,xl)))], #test levels not in x #for factor column, get all used levels - lapply(lapply( X[factor.ind],droplevels),levels),#xl - lapply(lapply(test[factor.ind],droplevels),levels),#tl + lapply( X[factor.ind],levels),#xl + lapply(test[factor.ind],levels),#tl SIMPLIFY = FALSE, USE.NAMES = TRUE ) unmatchedLevelsCount = sapply(unmatchedTestLevels,length) @@ -55,4 +55,4 @@ Xtestmerger = function(X,test,inbag=NULL,y=NULL) { }, isTrain = c(rep(T,dim(X)[1]),rep(F,dim(test)[1])) )) -} \ No newline at end of file +}