mlr: Machine Learning in R - basics

How to train and tune machine learning algorithms in a unified way?
With mlr R package 😄

I am currently keen on automated machine learning, especially hyperparameter optimization. Therefore, recently I mainly explore frameworks for unified model training. In this post, I will show how to train ML algorithms and tune them using grid-search. I am going to show only basics, but mlr package has more sophisticated features, I strongly encourage you to visit mlr webpage and explore all tutorials.

Data set

We will use BreastCancer data set from mlbench package and will perform binary classification. The aim of the model is to predict whether a cancer is benign or malignant (variable Class). It is worth to remove the first column that contains the id of a patient as it is redundant for modeling. To read more about the data set, see the documentation (?BreastCancer).

library("mlbench")
data("BreastCancer")
bc <- na.omit(BreastCancer[ ,-1])

head(bc)
##   Cl.thickness Cell.size Cell.shape Marg.adhesion Epith.c.size Bare.nuclei
## 1            5         1          1             1            2           1
## 2            5         4          4             5            7          10
## 3            3         1          1             1            2           2
## 4            6         8          8             1            3           4
## 5            4         1          1             3            2           1
## 6            8        10         10             8            7          10
##   Bl.cromatin Normal.nucleoli Mitoses     Class
## 1           3               1       1    benign
## 2           3               2       1    benign
## 3           3               1       1    benign
## 4           3               7       1    benign
## 5           3               1       1    benign
## 6           9               7       1 malignant

Installation

First of all, make sure that you have installed mlr, It is on CRAN, so you can simply use install.packages() function.

install.packages("mlr")

After installation, load mlr and set seed to make results reproducible.

library(mlr)
set.seed(1)

Modeling

Fitting a model

First, you need to define a task. The task is the definition of a machine learning problem. Our problem is classification, therefor we use makeClassifTask() function. For regression, it would be makeRegrTask() and for clustering makeClusterTask().

Parameter id define the name of the task, data is the data model will be trained on and target indicates the target variable.

classif_task = makeClassifTask(id = "bc", data = bc, target = "Class")
classif_task
## Supervised task: bc
## Type: classif
## Target: Class
## Observations: 683
## Features:
##    numerics     factors     ordered functionals 
##           0           4           5           0 
## Missings: FALSE
## Has weights: FALSE
## Has blocking: FALSE
## Has coordinates: FALSE
## Classes: 2
##    benign malignant 
##       444       239 
## Positive class: benign

The second step is defining a model. Please, note that we do not train a model yet. We only create an object that describes our algorithm.

classif_lrn = makeLearner("classif.randomForest", par.vals = list(ntree = 200))

In the example above, we have created an object that defines classification random forest with 200 trees. To get names of hyperparameters, their ranges, and default values use function getParamSet().

getParamSet(classif_lrn)
##                      Type  len   Def   Constr Req Tunable Trafo
## ntree             integer    -   500 1 to Inf   -    TRUE     -
## mtry              integer    -     - 1 to Inf   -    TRUE     -
## replace           logical    -  TRUE        -   -    TRUE     -
## classwt     numericvector <NA>     - 0 to Inf   -    TRUE     -
## cutoff      numericvector <NA>     -   0 to 1   -    TRUE     -
## strata            untyped    -     -        -   -   FALSE     -
## sampsize    integervector <NA>     - 1 to Inf   -    TRUE     -
## nodesize          integer    -     1 1 to Inf   -    TRUE     -
## maxnodes          integer    -     - 1 to Inf   -    TRUE     -
## importance        logical    - FALSE        -   -    TRUE     -
## localImp          logical    - FALSE        -   -    TRUE     -
## proximity         logical    - FALSE        -   -   FALSE     -
## oob.prox          logical    -     -        -   Y   FALSE     -
## norm.votes        logical    -  TRUE        -   -   FALSE     -
## do.trace          logical    - FALSE        -   -   FALSE     -
## keep.forest       logical    -  TRUE        -   -   FALSE     -
## keep.inbag        logical    - FALSE        -   -   FALSE     -

Now, we are ready to fit a model. We can just simply use function train() with specified model and task.

model = train(classif_lrn, classif_task)

Tuning a model

To tune hyperparameters, we need to specify a space fo search. For defining space for integer parameters we use function makeIntegerParam(). All of this is pinned together with the function makeParamSet().

params = makeParamSet(
  makeIntegerParam("mtry", lower = 1, upper = 100),
  makeIntegerParam("ntree", lower = 1L, upper = 500L)
)

Now, we use function makeTuneControlRandom() to create an object that define random search. Parameter maxit defines the number of iterations. Function makeResampleDesc() creates an object for a resampling strategy, in this case cross-validation. Finally, we can combine all of the previous pieces with function tuneParams() and tune random forest.

ctrl = makeTuneControlRandom(maxit = 10L)
rdesc = makeResampleDesc("CV", iters = 3L)

res = tuneParams(classif_lrn, task = classif_task, 
                 resampling = rdesc, 
                 par.set = params,
                 control = ctrl,
                 measures = list(acc),
                 show.info = FALSE)
res
## Tune result:
## Op. pars: mtry=49; ntree=464
## acc.test.mean=0.9707409

As a result of tuning, we have obtained hyperparameters mtry=49, ntree=464.

More

If you would like to learn more about mlr, you can visit mlr webpage.
What is more, a new version of this package is coming up. I highly recommended the following information about mlr3.

comments powered by Disqus