SRxgboost_h2o_automl: SRxgboost_h2o_automl

In samuelreuther/SRxgboost: Helper and meta-functions for modelling with XGBOOST

SRxgboost_h2o_automl

Description

Runs several models with h2o.automl. Install latestest version of h2o: install.packages("h2o", type="source", repos=(c("http://h2o-release.s3.amazonaws.com/h2o/latest_stable_R")))

Usage

SRxgboost_h2o_automl(
  y = NULL,
  data_train = NULL,
  data_test = NULL,
  metric = NULL,
  kfold = 5,
  max_runtime_sec = 5 * 60,
  run_shap = FALSE,
  nthreads = NULL,
  max_mem_size = "16g"
)

Arguments

y	character
data_train	data.frame
data_test	data.frame
metric	character: "AUTO", "RMSE", "MAE", "RMSLE", "AUC", "AUCPR", "logloss", "lift_top_group", "misclassification", "mean_per_class_error"
kfold	integer
max_runtime_sec	integer
run_shap	boolean

Value

writes a lot of files

samuelreuther/SRxgboost documentation built on March 30, 2025, 12:48 a.m.