fit: Fit model

In tonyelhabr/xgbh: xgboost helpers

Description

Fit model.

Usage

fit(
  data = NULL,
  x = NULL,
  y = NULL,
  col_y = NULL,
  id = NULL,
  col_id = NULL,
  overwrite = TRUE,
  export = TRUE,
  f_import = readr::read_rds,
  f_export = readr::write_rds,
  ...,
  drop = TRUE,
  dir = .get_dir_data(),
  file = "fit",
  ext = "rds",
  path = NULL,
  sep = "_",
  suffix = NULL,
  print_every_n = 10,
  cols_extra = NULL,
  n_param = 10,
  nrounds = 2000,
  booster = "gbtree",
  objective = "reg:squarederror",
  eval_metrics = list("rmse"),
  .params = list(NULL),
  early_stopping_rounds = 10
)

tonyelhabr/xgbh documentation built on Dec. 23, 2021, 11:59 a.m.