get_partial_dependence_xgb_2d: get_partial_dependence_xgb_2d()

Description Usage Arguments

View source: R/get_partial_dependence_xgb_2d.R

Description

Cette fonction calcule les dépendances partielles en 2D pour 2 variables d'un modèles xgboost après avoir enlevé les outliers.

Usage

1
2
get_partial_dependence_xgb_2d(xgb_model, feature_vars, model_data,
  pdp_var1, pdp_var2, offset_var = NULL, nquantiles = 5)

Arguments

xgb_model

le modèle xgboost qui va être prédit

feature_vars

un vecteur caractère avec les noms des variables prédictives du modèle

model_data

un data frame avec les observations à prédire

pdp_var1

la premiere variable pour la pdp (caractère)

pdp_var2

la deuxième variable pour la pdp (caractère)

offset_var

variable à mettre en offset, si nécessaire

nquantiles

le nombre de quantiles à calculer.


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