train_model2_outer_fold: Train model 2 - Process the outer fold

View source: R/train_model2.R

train_model2_outer_foldR Documentation

Train model 2 - Process the outer fold

Description

Train model 2 - Process the outer fold

Usage

train_model2_outer_fold(
  dat,
  response_name,
  outer_train_index,
  random_state,
  inner_cv_folds = 5,
  rf_grid = NULL,
  top_n = 10000,
  feature_selection_ranger_num_trees = 100,
  importance = "permutation",
  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.

rf_grid

A data frame with possible tuning values. See train.

top_n

top N loci/features.

importance

importance of ranger.

verbose

A bool.

feature_selection_ranger_num_tree

num.trees of ranger.

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.