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

Make the train and test input with same colnames. (#4329)

Browse Source 
Fix the bug report of https://github.com/dmlc/xgboost/issues/4328.
I am the beginner of the Git so just try my best to follows the guide, https://xgboost.readthedocs.io/en/latest/contribute.html#r-package.
I find there is no `dev`  branch, so I pull this fix from my master branch to the original master branch.
This commit is contained in:
Jiaxiang Li 2019-04-05 06:59:27 +08:00 committed by Philip Hyunsu Cho
parent 70be1e38c2
commit 1ca5698221
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -38,6 +38,7 @@ create.new.tree.features <- function(model, original.features){
# Convert previous features to one hot encoding # Convert previous features to one hot encoding
new.features.train <- create.new.tree.features(bst, agaricus.train$data) new.features.train <- create.new.tree.features(bst, agaricus.train$data)
new.features.test <- create.new.tree.features(bst, agaricus.test$data) new.features.test <- create.new.tree.features(bst, agaricus.test$data)
colnames(new.features.test) <- colnames(new.features.train)
# learning with new features # learning with new features
new.dtrain <- xgb.DMatrix(data = new.features.train, label = agaricus.train$label) new.dtrain <- xgb.DMatrix(data = new.features.train, label = agaricus.train$label)