train_model1_outer_fold: Train model 1 - Process the outer fold

View source: R/train_model1.R

train_model1_outer_foldR Documentation

Train model 1 - Process the outer fold

Description

Train model 1 - Process the outer fold

Usage

train_model1_outer_fold(
  dat,
  response_name,
  outer_train_index,
  random_state,
  feature_selection = c("Boruta"),
  inner_cv_folds = 5,
  rf_grid = NULL,
  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.

feature_selection

feature selection method.

inner_cv_folds

inner cross-validation fold number.

rf_grid

A data frame with possible tuning values. See train.

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.