get_model_specs {shapr} | R Documentation |
Fetches feature information from natively supported models
Description
This function is used to extract the feature information from the model to be checked against the
corresponding feature information in the data passed to explain()
.
NOTE: You should never need to call this function explicitly. It is exported just to be easier accessible for users, see details.
Usage
get_model_specs(x)
## Default S3 method:
get_model_specs(x)
## S3 method for class 'ar'
get_model_specs(x)
## S3 method for class 'Arima'
get_model_specs(x)
## S3 method for class 'forecast_ARIMA'
get_model_specs(x)
## S3 method for class 'glm'
get_model_specs(x)
## S3 method for class 'lm'
get_model_specs(x)
## S3 method for class 'gam'
get_model_specs(x)
## S3 method for class 'ranger'
get_model_specs(x)
## S3 method for class 'workflow'
get_model_specs(x)
## S3 method for class 'xgb.Booster'
get_model_specs(x)
Arguments
x |
Model object for the model to be explained. |
Details
If you are explaining a model not supported natively, you may (optionally) enable such checking by
creating this function yourself and passing it on to explain()
.
Value
A list with the following elements:
- labels
character vector with the feature names to compute Shapley values for
- classes
a named character vector with the labels as names and the class type as elements
- factor_levels
a named list with the labels as names and character vectors with the factor levels as elements (NULL if the feature is not a factor)
Author(s)
Martin Jullum
See Also
For model classes not supported natively, you NEED to create an analogue to predict_model()
. See it's
help file for details.
Examples
# Load example data
data("airquality")
airquality <- airquality[complete.cases(airquality), ]
# Split data into test- and training data
x_train <- head(airquality, -3)
x_explain <- tail(airquality, 3)
# Fit a linear model
model <- lm(Ozone ~ Solar.R + Wind + Temp + Month, data = x_train)
get_model_specs(model)