% Generated by roxygen2 (4.1.0): do not edit by hand % Please edit documentation in R/xgb.importance.R \name{xgb.importance} \alias{xgb.importance} \title{Show importance of features in a model} \usage{ xgb.importance(feature_names = NULL, filename_dump = NULL, model = NULL) } \arguments{ \item{feature_names}{names of each feature as a character vector. Can be extracted from a sparse matrix (see example). If model dump already contains feature names, this argument should be \code{NULL}.} \item{filename_dump}{the path to the text file storing the model. Model dump must include the gain per feature and per tree (\code{with.stats = T} in function \code{xgb.dump}).} \item{model}{generated by the \code{xgb.train} function. Avoid the creation of a dump file.} } \value{ A \code{data.table} of the features used in the model with their average gain (and their weight for boosted tree model) in the model. } \description{ Read a xgboost model text dump. Can be tree or linear model (text dump of linear model are only supported in dev version of \code{Xgboost} for now). } \details{ This is the function to understand the model trained (and through your model, your data). Results are returned for both linear and tree models. \code{data.table} is returned by the function. There are 3 columns : \itemize{ \item \code{Features} name of the features as provided in \code{feature_names} or already present in the model dump. \item \code{Gain} contribution of each feature to the model. For boosted tree model, each gain of each feature of each tree is taken into account, then average per feature to give a vision of the entire model. Highest percentage means important feature to predict the \code{label} used for the training ; \item \code{Cover} metric of the number of observation related to this feature (only available for tree models) ; \item \code{Weight} percentage representing the relative number of times a feature have been taken into trees. \code{Gain} should be prefered to search the most important feature. For boosted linear model, this column has no meaning. } } \examples{ data(agaricus.train, package='xgboost') data(agaricus.test, package='xgboost') #Both dataset are list with two items, a sparse matrix and labels (labels = outcome column which will be learned). #Each column of the sparse Matrix is a feature in one hot encoding format. train <- agaricus.train test <- agaricus.test bst <- xgboost(data = train$data, label = train$label, max.depth = 2, eta = 1, nround = 2,objective = "binary:logistic") #agaricus.test$data@Dimnames[[2]] represents the column names of the sparse matrix. xgb.importance(agaricus.test$data@Dimnames[[2]], model = bst) }