From 3b67028ad62a60a8e98b78b6332c3a6a8242449c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Micha=C3=ABl=20Benesty?= Date: Sat, 5 Dec 2015 19:02:05 +0100 Subject: [PATCH] remove intersect column in sparse Matrix --- R-package/demo/predict_leaf_indices.R | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/R-package/demo/predict_leaf_indices.R b/R-package/demo/predict_leaf_indices.R index 6cde561c2..fcde3438d 100644 --- a/R-package/demo/predict_leaf_indices.R +++ b/R-package/demo/predict_leaf_indices.R @@ -29,10 +29,10 @@ create.new.tree.features <- function(model, original.features){ cols <- list() for(i in 1:length(trees)){ # max is not the real max but it s not important for the purpose of adding features - max <- max(pred_with_leaf[,i]) - cols[[i]] <- factor(x = pred_with_leaf[,i], level = seq(to = max)) + leaf.id <- sort(unique(pred_with_leaf[,i])) + cols[[i]] <- factor(x = pred_with_leaf[,i], level = leaf.id) } - cBind(original.features, sparse.model.matrix( ~ ., as.data.frame(cols))) + cBind(original.features, sparse.model.matrix( ~ . -1, as.data.frame(cols))) } # Convert previous features to one hot encoding