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.