mlr での機械学習

R によるハイパーパラメータチューニング

Dr. Shirin Elsinghorst

Data Scientist

mlr パッケージ

  • mlr は R のもう一つの機械学習フレームワークです。

学習は3ステップで行います:

  1. タスクを定義
  2. 学習器を定義
  3. モデルを学習

https://mlr-org.github.io/mlr

R によるハイパーパラメータチューニング

新データセット: User Knowledge Data

library(tidyverse)
glimpse(knowledge_data)
Observations: 150
Variables: 6
$ STG <dbl> 0.080, 0.000, 0.180, 0.100, 0.120, 0.090, 0.080, 0.150, ...
$ SCG <dbl> 0.080, 0.000, 0.180, 0.100, 0.120, 0.300, 0.325, 0.275, ...
$ STR <dbl> 0.100, 0.500, 0.550, 0.700, 0.750, 0.680, 0.620, 0.800, ...
$ LPR <dbl> 0.24, 0.20, 0.30, 0.15, 0.35, 0.18, 0.94, 0.21, 0.19, ...
$ PEG <dbl> 0.90, 0.85, 0.81, 0.90, 0.80, 0.85, 0.56, 0.81, 0.82, ...
$ UNS <chr> "High", "High", "High", "High", "High", "High", ...

 

knowledge_data %>%
    count(UNS)

 

# A tibble: 3 x 2
  UNS        n
  <chr>  <int>
1 High      50                                                         
2 Low       50
3 Middle    50
R によるハイパーパラメータチューニング

教師あり学習のタスク(mlr)

  • 回帰には RegrTask()
  • 二値・多クラス分類には ClassifTask()
  • マルチラベル分類には MultilabelTask()
  • コスト考慮の分類には CostSensTask()

 

学生知識データで分類器を作れます:

task <- makeClassifTask(data = knowledge_train_data, 
                        target = "UNS")
R によるハイパーパラメータチューニング
listLearners()
                            class                   package
1                     classif.ada                 ada,rpart
2              classif.adaboostm1                     RWeka
3             classif.bartMachine               bartMachine
4                classif.binomial                     stats
5                classif.boosting              adabag,rpart
6                     classif.bst                 bst,rpart
7                     classif.C50                       C50
8                 classif.cforest                     party
9              classif.clusterSVM        SwarmSVM,LiblineaR
10                  classif.ctree                     party
...
# 学習器を定義
lrn <- makeLearner("classif.h2o.deeplearning", 
                   fix.factors.prediction = TRUE,
                   predict.type = "prob")
R によるハイパーパラメータチューニング

mlr でのモデル学習

tic()
# タスクを定義
task <- makeClassifTask(data = knowledge_train_data, 
                        target = "UNS")
# 学習器を定義
lrn <- makeLearner("classif.h2o.deeplearning",
                   fix.factors.prediction = TRUE)
# モデルを学習
model <- train(lrn, 
               task)
toc()
3.782 sec elapsed
R によるハイパーパラメータチューニング

練習しましょう!

R によるハイパーパラメータチューニング

Preparing Video For Download...