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

Add comments

Browse Source
This commit is contained in:
El Potaeto 2015-02-17 23:30:36 +01:00
parent 08493c2b3d
commit 412a6e1085
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
R-package/R/xgb.importance.R
View File

@ -94,15 +94,19 @@ xgb.importance <- function(feature_names = NULL, filename_dump = NULL, model = N
result <- treeDump(feature_names, text = text, keepDetail = !is.null(data)) result <- treeDump(feature_names, text = text, keepDetail = !is.null(data))
# Co-occurence computation # Co-occurence computation
if(!is.null(data) & !is.null(label) & nrow(result) > 0) { if(!is.null(data) & !is.null(label) & nrow(result) > 0) {
# Apply split
a <- data[, result[,Feature],drop=FALSE] < as.numeric(result[,Split]) a <- data[, result[,Feature],drop=FALSE] < as.numeric(result[,Split])
# Take care of missing column
b <- data[, result[No == Missing,Feature],drop=FALSE] != 0 b <- data[, result[No == Missing,Feature],drop=FALSE] != 0
# Do nothing if missing should be included in Yes
c <- data[, result[No != Missing,Feature],drop=FALSE] c <- data[, result[No != Missing,Feature],drop=FALSE]
# Bind the two previous Matrix and reorder columns
d <- cBind(b,c) %>% .[,result[,Feature]] d <- cBind(b,c) %>% .[,result[,Feature]]
apply(a & d, 2, . %>% target %>% sum) -> vec apply(a & d, 2, . %>% target %>% sum) -> vec
result <- result[, "RealCover":= as.numeric(vec), with = F][, "RealCover %" := RealCover / sum(label)] result <- result[, "RealCover":= as.numeric(vec), with = F][, "RealCover %" := RealCover / sum(label)][,`:=`(No = NULL, Missing = NULL)]
} }
} }
result result