extract.coef.xgb.Booster | R Documentation |
Extract Coefficient Information from Models
## S3 method for class 'xgb.Booster'
extract.coef(
model,
feature_names = NULL,
removeNonSelected = TRUE,
zero_threshold = 0.001,
...
)
model |
Model object from which to extract information. |
feature_names |
Names of coefficients |
removeNonSelected |
If |
zero_threshold |
Since |
... |
Further arguments |
Gets the coefficient values and variable names from a model. Since xgboost does not have standard errors, those will just be NA.
A data.frame
containing the coefficient, the standard
error and the variable name.
Jared P. Lander
library(xgboost)
data(diamonds, package='ggplot2')
diaX <- useful::build.x(price ~ carat + cut + x, data=diamonds, contrasts=FALSE)
diaY <- useful::build.y(price ~ carat + cut + x, data=diamonds)
xg1 <- xgb.train(data=xgb.DMatrix(data=diaX, label=diaY),
booster='gblinear',
objective='reg:squarederror', eval_metric='rmse',
nrounds=50
)
extract.coef(xg1)
extract.coef(xg1, zero_threshold=0)
extract.coef(xg1, feature_names=colnames(diaX))
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.