xgb_treino_ensemble: Treino ensemble de modelos XGBoost

In zegkreist/magikaRp: Serie de funções úteis a mim

Description

O objetivo desta função é treinar vários modelos por meio de sessões. Cada sessão possui uma parte dos dados, um modelo ali então é treinado e salvo numa pasta especificada.

Usage

xgb_treino_ensemble(dados, n_samples, nthreads = 3,
  parametros_treino = NULL, n_models = 100, save_importance = F,
  folder_to_save = NULL)

Arguments

dados	Objeto data.table com os dados já pre processados e com uma coluna target.
n_samples	O tamanho do sorteio par acada classe. Se regressão o tamanbho total da amostra
nthreads	A quantidade de CPU threads disponíveis par ao XGBoost. DEFAULT: 3
parametros_treino	A lista de parâmetros retornado pela função xgb_select_params
n_models	A quantidade de modelos gerados. Também é o número de sessões realizado. DEFAULT: 100
save_importance	Indicador para o salvamento das importâncias das variáveis para cada modelo. DEFAULT: F
folder_to_save	Pasta em que será salvo os n_models gerados

Value

A função não retorna nada, apenas salva os modelos na pasta

zegkreist/magikaRp documentation built on Oct. 16, 2019, 4:16 a.m.