train_model5_outer_fold: Train model 5 - Process the outer fold

View source: R/train_model5.R

train_model5_outer_foldR Documentation

Train model 5 - Process the outer fold

Description

Train model 5 - Process the outer fold

Usage

train_model5_outer_fold(
  dat,
  response_name,
  outer_train_index,
  random_state,
  inner_cv_folds = 3,
  calibration_youden_index_threshold = 0.9,
  calibration_lambda_min_ratio = 1e-06,
  rf_grid = NULL,
  result_file = "outer_fold.Rda",
  verbose = TRUE,
  ...
)

Arguments

dat

a data.frame of input data.

response_name

column name of the response.

outer_train_index

outer train sample indexes.

random_state

random seed.

inner_cv_folds

inner cross-validation fold number.

calibration_youden_index_threshold

a float of Youden index threhold. Default to 0.9.

calibration_lambda_min_ratio

see lambda.min.ratio of glmnet. Default to 1e-6.

rf_grid

A data frame with possible tuning values. See train.

result_file

save result in Rda file.

verbose

A bool.

Value

a list of two attributes, a data.frame of calibrated probabilities and response variable, and a vector of selected features.


markgene/yamatClassifier documentation built on Oct. 14, 2024, 2:36 a.m.