View source: R/HDI_quantregforest.R
This function implements an HDI RF prediction interval method.
1 2 3 4 5 6 7 8 9 10 11 12 13 |
formula |
Object of class formula or character describing the model to fit. Interaction terms supported only for numerical variables. |
train_data |
Training data of class data.frame, matrix, dgCMatrix (Matrix) or gwaa.data (GenABEL). Matches ranger() requirements. |
test_data |
Test data of class data.frame, matrix, dgCMatrix (Matrix) or gwaa.data (GenABEL). Utilizes ranger::predict() to get prediction intervals for test data. |
alpha |
Significance level for prediction intervals. |
num_tree |
Number of trees. |
mtry |
Number of variables to randomly select from at each split. |
min_node_size |
Minimum number of observations before split at a node. |
max_depth |
maximum depth of each tree in RF. ranger parameter. |
replace |
Sample with replacement, or not. Utilized for the two different variants outlined in Ghosal, Hooker 2018. Currently variant 2 not implemented. |
num_threads |
The number of threads to use in parallel. Default is the current number of cores. |
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.