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

[R] allow xgb.plot.importance() calls to fill a grid (#6294)

Browse Source
This commit is contained in:
James Lamb 2020-10-28 21:37:28 +00:00 committed by GitHub
parent d261ba029a
commit 6383757dca
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
R-package/R/xgb.plot.importance.R
View File

@ -99,8 +99,12 @@ xgb.plot.importance <- function(importance_matrix = NULL, top_n = NULL, measure
} }
if (plot) { if (plot) {
op <- par(no.readonly = TRUE) original_mar <- par()$mar
mar <- op$mar
# reset margins so this function doesn't have side effects
on.exit({par(mar = original_mar)})
mar <- original_mar
if (!is.null(left_margin)) if (!is.null(left_margin))
mar[2] <- left_margin mar[2] <- left_margin
par(mar = mar) par(mar = mar)
@ -109,11 +113,6 @@ xgb.plot.importance <- function(importance_matrix = NULL, top_n = NULL, measure
importance_matrix[rev(seq_len(nrow(importance_matrix))), importance_matrix[rev(seq_len(nrow(importance_matrix))),
barplot(Importance, horiz = TRUE, border = NA, cex.names = cex, barplot(Importance, horiz = TRUE, border = NA, cex.names = cex,
names.arg = Feature, las = 1, ...)] names.arg = Feature, las = 1, ...)]
grid(NULL, NA)
# redraw over the grid
importance_matrix[rev(seq_len(nrow(importance_matrix))),
barplot(Importance, horiz = TRUE, border = NA, add = TRUE)]
par(op)
} }
invisible(importance_matrix) invisible(importance_matrix)