run_rf {ubair} | R Documentation |
Run random forest model with ranger
Description
This function trains a random forest model (ranger) on the specified training dataset and makes predictions on the test dataset in a counterfactual scenario. The model uses meteorological variables and temporal features.
Usage
run_rf(train, test, model_params, alpha, calc_shaps)
Arguments
train |
Dataframe of train data as returned by the |
test |
Dataframe of test data as returned by the |
model_params |
list of hyperparameters to use in ranger call. See |
alpha |
Confidence level of the prediction interval between 0 and 1. |
calc_shaps |
Boolean value. If TRUE, calculate SHAP values for the
method used and format them so they can be visualised with |
Details
Note: Runs the random forest model for individualised use with own data pipeline.
Otherwise use run_counterfactual()
to call this function.
Value
List with data frame of predictions and model