get_xgboost_fold_pred_model: get_xgboost_fold_pred_model()

In SimonCoulombe/pkgsimon: Les fonctions de Simon

Description

get_xgboost_fold_pred_model() retourne les prédictions et les modèles de chaque fold d'un xgboost. Pourquoi utiliser ça plutôt que caret? Parce que je gère le offset et les contraintes de monotonicité.

Usage

get_xgboost_fold_pred_model(params = list(), data, label_var,
  feature_vars, offset_var = NULL, nb_fold = 10,
  fold_group_var = NULL, seed = 8484, nround = 1000,
  contraintes = NULL)

Arguments

params	les paramètre du xgboost,
data	la base de données (categorical variables have been replaced by dummies.)
label_var	le nom de la variable à prédire
feature_vars	le vecteur des noms des variables explicatives
offset_var	le nom de la variable qui sera mise dans le base_margin (ex:exposure), default= NULL
nb_fold	le nombre de fold dans notre validation croisée
seed	le numéro de seed
nround	le nombre d'arbres à créer avec le xgboost
contraintes	vecteur des contraints de monotonicité

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