correctionModel: Correction Model Function

In Alessandra23/pineR: Modelling the life cycle of the pine weevil

Correction Model Function

Description

This function modifies the class of the input data.frame to match the model specified and then attempts to generate predictions.

Usage

correctionModel(
  data,
  model = c("ranger", "bart", "xgboost", "nn", "svm", "lasso", "lm"),
  seed,
  ...
)

Arguments

data	A data.frame containing the dataset to be used.
model	Character. The model to be used for correction. The available options are 'ranger', 'bart', 'xgboost', 'nn', 'svm', 'lasso', and 'lm'. The default model is 'ranger'.
seed	Numeric. The seed for reproducibility. Not used in this function but might be used in the 'getPredicts' function.
...	Additional arguments to be passed to the model's function.

Value

This function returns the result of calling 'getPredicts' on the input data.

Examples

## Not run: 
corrected_data <- correctionModel(data = mtcars, model = "xgboost", seed = 123)

## End(Not run)

Alessandra23/pineR documentation built on Aug. 26, 2023, 8 a.m.