rmweather (version 0.2.62)

rmw_predict: Function to predict using a ranger random forest.

Description

Function to predict using a ranger random forest.

Usage

rmw_predict(model, df = NA, se = FALSE, n_cores = NULL, verbose = FALSE)

Value

Numeric vector or a named list containing two numeric vectors.

Arguments

model

A ranger model object from rmw_train_model.

df

Input data to be used for predictions.

se

If df is supplied, should the standard error of the prediction be calculated too? The standard error method is the "infinitesimal jackknife for bagging" and will slow down the predictions significantly.

n_cores

Number of CPU cores to use for the model predictions.

verbose

Should the function give messages?

Author

Stuart K. Grange

Examples

# Load package
library(dplyr)

# Prepare example data
data_london_prepared <- data_london %>% 
  filter(variable == "no2") %>% 
  rmw_prepare_data()

# Make a prediction with the examples
vector_prediction <- rmw_predict(
  model_london, 
  df = data_london_prepared
)


# Make a prediction with standard errors too
list_prediction <- rmw_predict(
  model_london, 
  df = data_london_prepared,
  se = TRUE
)