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

Bug + documentation

Browse Source
This commit is contained in:
El Potaeto 2015-02-15 17:46:12 +01:00
parent f84cc0843f
commit f0eaac2174
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -59,7 +59,7 @@
#' xgb.importance(train$data@@Dimnames[[2]], model = bst)
#'
#' # Same thing with co-occurence computation this time
#' xgb.importance(agaricus.test$data@@Dimnames[[2]], model = bst, data = train$data, label = train$label)
#' xgb.importance(train$data@@Dimnames[[2]], model = bst, data = train$data, label = train$label)
#'
#' @export
xgb.importance <- function(feature_names = NULL, filename_dump = NULL, model = NULL, data = NULL, label = NULL, target = function(x) ((x + label) == 2)){
@ -102,7 +102,7 @@ xgb.importance <- function(feature_names = NULL, filename_dump = NULL, model = N
}
treeDump <- function(feature_names, text){
result <- xgb.model.dt.tree(feature_names = feature_names, text = text)[Feature!="Leaf",.(Gain = sum(Quality), Cover = sum(Cover), Frequence = .N), by = Feature][,`:=`(Gain = Gain/sum(Gain), Cover = Cover/sum(Cover), Frequence = Frequence/sum(Frequence))]
result <- xgb.model.dt.tree(feature_names = feature_names, text = text)[Feature!="Leaf",.(Gain = sum(Quality), Cover = sum(Cover), Frequence = .N), by = Feature][,`:=`(Gain = Gain/sum(Gain), Cover = Cover/sum(Cover), Frequence = Frequence/sum(Frequence))][order(Gain, decreasing = T)]
result
}

2
R-package/man/xgb.importance.Rd
View File

@ -64,6 +64,6 @@ bst <- xgboost(data = train$data, label = train$label, max.depth = 2,
xgb.importance(train$data@Dimnames[[2]], model = bst)
# Same thing with co-occurence computation this time
xgb.importance(agaricus.test$data@Dimnames[[2]], model = bst, data = train$data, label = train$label)
xgb.importance(train$data@Dimnames[[2]], model = bst, data = train$data, label = train$label)
}