This learner wraps train, providing a procedure to fit the algorithms available in package caret.

Format

R6Class object.

Value

Learner object with methods for training and prediction. See Lrnr_base for documentation on learners.

Parameters

algorithm

The algorithm to use. This argument is passed as method to train. For a detailed description of the methods available in the package, please consult the documentation for train.

metric

An optional argument specifying the summary metric to be used to select the optimal model. If not specified, it will be inferred depending on the outcome type. For a detailed description, please consult the documentation for train.

trControl

An optional trainControl object controlling the computational nuances. If not specified, it will be initialized with method = "cv" instead of the default "boot". For a detailed description, please consult the documentation for train.

...

Other parameters passed to train.