mtry.guide: Recommended mtry range

View source: R/RandomForest_Functions.R

mtry.guideR Documentation

Recommended mtry range

Description

Recommended mtry range based on number of paramerets and tree type

Usage

mtry.guide(npred, rf.type, spread.upper = 5, spread.lower = 4)

Arguments

npred

Numeric. Input number of predictor variables.

rf.type

One of: "class" for classification trees or "reg" for regression trees.

spread.upper

Upper range of mtry to use above optimal mtry value, Default: 5

spread.lower

Lower range of mtry to use below optimal mtry value: 4

Details

Written primarily for rf wrapper function

Value

Returns a list:
ideal.mtry is optimal mtry based on tree type and number of predictor variables
upper.mtry is upper range of mtry for downstream hyperparameter tuning
lower.mtry is lowest value of mtry for downstream hyperparameter tuning
range is range of mtry values to try in downstream hyperparameter tuning given in the format: upper.mtry:lower.mtry


SarahAsbury/BioDataTools documentation built on Feb. 5, 2024, 4:01 p.m.