bug fix

gtesei · Oct 2, 2015 · f70d138 · f70d138
1 parent 433a886
commit f70d138
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/R-package/R/featureEncode.R b/R-package/R/featureEncode.R
@@ -89,8 +89,8 @@ ff.encodeCategoricalFeature = function(data.train ,
   }
 
   ## reassembling 
-  testdata = mm[1:(length(data.test)),]
-  traindata = mm[((length(data.test))+1):(dim(mm)[1]),]
+  testdata = mm[1:(length(data.test)),,drop=F]
+  traindata = mm[((length(data.test))+1):(dim(mm)[1]),,drop=F]
 
   return(list(traindata = traindata ,testdata = testdata))
 }