Learn R Programming

mlr (version 2.3)

train: Train a learning algorithm.

Description

Given a Task, creates a model for the learning machine which can be used for predictions on new data.

Usage

train(learner, task, subset, weights = NULL)

Arguments

Value

[WrappedModel].

See Also

predict.WrappedModel

Examples

Run this code
training.set = sample(1:nrow(iris), nrow(iris) / 2)

## use linear discriminant analysis to classify iris data
task = makeClassifTask(data = iris, target = "Species")
learner = makeLearner("classif.lda", method = "mle")
mod = train(learner, task, subset = training.set)
print(mod)

## use random forest to classify iris data
task = makeClassifTask(data = iris, target = "Species")
learner = makeLearner("classif.rpart", minsplit = 7, predict.type = "prob")
mod = train(learner, task, subset = training.set)
print(mod)

Run the code above in your browser using DataLab