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

issue #368, data.table problems

Browse Source
This commit is contained in:
orenov 2015-07-22 12:03:01 +03:00
parent 80b6ec4478
commit d8fc16538e
1 changed files with 18 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -137,29 +137,28 @@ xgb.model.dt.tree <- function(feature_names = NULL, filename_dump = NULL, model
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
} }