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

Merge pull request #399 from orenov/master

Browse Source 
issue #368, data.table problems
This commit is contained in:
Michaël Benesty 2015-07-22 21:21:34 +02:00
commit 1fb5c127b5
1 changed files with 18 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
R-package/R/xgb.model.dt.tree.R
View File

@ -133,33 +133,32 @@ xgb.model.dt.tree <- function(feature_names = NULL, filename_dump = NULL, model
allTrees <- rbindlist(list(allTrees, dt), use.names = T, fill = F) allTrees <- rbindlist(list(allTrees, dt), use.names = T, fill = F)
} }
yes <- allTrees[!is.na(Yes),Yes] yes <- allTrees[!is.na(Yes), Yes]
set(allTrees, i = which(allTrees[,Feature]!= "Leaf"), set(allTrees, i = which(allTrees[, Feature] != "Leaf"),
j = "Yes.Feature", j = "Yes.Feature",
value = allTrees[ID == yes,Feature]) value = allTrees[ID %in% yes, Feature])
set(allTrees, i = which(allTrees[,Feature]!= "Leaf"), set(allTrees, i = which(allTrees[, Feature] != "Leaf"),
j = "Yes.Cover", j = "Yes.Cover",
value = allTrees[ID == yes,Cover]) value = allTrees[ID %in% yes, Cover])
set(allTrees, i = which(allTrees[,Feature]!= "Leaf"), set(allTrees, i = which(allTrees[, Feature] != "Leaf"),
j = "Yes.Quality", j = "Yes.Quality",
value = allTrees[ID == yes,Quality]) value = allTrees[ID %in% yes, Quality])
no <- allTrees[!is.na(No), No]
no <- allTrees[!is.na(No),No] set(allTrees, i = which(allTrees[, Feature] != "Leaf"),
set(allTrees, i = which(allTrees[,Feature]!= "Leaf"),
j = "No.Feature", j = "No.Feature",
value = allTrees[ID == no,Feature]) value = allTrees[ID %in% no, Feature])
set(allTrees, i = which(allTrees[,Feature]!= "Leaf"), set(allTrees, i = which(allTrees[, Feature] != "Leaf"),
j = "No.Cover", j = "No.Cover",
value = allTrees[ID == no,Cover]) value = allTrees[ID %in% no, Cover])
set(allTrees, i = which(allTrees[,Feature]!= "Leaf"), set(allTrees, i = which(allTrees[, Feature] != "Leaf"),
j = "No.Quality", j = "No.Quality",
value = allTrees[ID == no,Quality]) value = allTrees[ID %in% no, Quality])
allTrees allTrees
} }