Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

remove intersect column in sparse Matrix

Browse Source
This commit is contained in:
Michaël Benesty 2015-12-05 19:02:05 +01:00
parent 4f4a5409d7
commit 3b67028ad6
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
R-package/demo/predict_leaf_indices.R
View File

@ -29,10 +29,10 @@ create.new.tree.features <- function(model, original.features){
cols <- list() cols <- list()
for(i in 1:length(trees)){ for(i in 1:length(trees)){
# max is not the real max but it s not important for the purpose of adding features # max is not the real max but it s not important for the purpose of adding features
max <- max(pred_with_leaf[,i]) leaf.id <- sort(unique(pred_with_leaf[,i]))
cols[[i]] <- factor(x = pred_with_leaf[,i], level = seq(to = max)) 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 # Convert previous features to one hot encoding