get_partial_dependence_xgb: get_partial_dependence_xgb()

In SimonCoulombe/pkgsimon: Les fonctions de Simon

Description

Cette fonction calcule les dépendances partielles d'un modèles xgboost après avoir enlevé les outliers. défaut = toutes les variables. elle retourne un data frame avec les valeurs (Mod) de la variable explicative (Variable)

Usage

get_partial_dependence_xgb(xgb_model, pdp_vars = NULL, feature_vars,
  offset_var = NULL, model_data, nquantiles = 21)

Arguments

xgb_model	le modèle xgboost qui va être prédit
pdp_vars	un vecteur caractère avec les noms des variables pour lesquelles on souhaite calculer la pdp
feature_vars	un vecteur caractère avec les noms des variables prédictives du modèle
offset_var	variable à mettre en offset, si nécessaire
model_data	un data frame avec les observations à prédire
nquantiles	le nombre de quantiles à calculer.

Examples

zzzz1 <- get_partial_dependence_xgb(xgb_model = resBoost,
                                    feature_vars = c("brake_intensity", "number_of_brake", "number_of_accel", "distraction"),
                                    model_data = model_data,
                                    pdp_vars = c("brake_intensity"),
                                    offset_var = "pred_saison",
                                    nquantiles = 11)

SimonCoulombe/pkgsimon documentation built on Sept. 5, 2019, 9:12 p.m.