始める無料で始める

ペナルティのチューニング

Lasso は、許容できる性能を保ちながらモデルの特徴量数を減らすのに有効だと確信できたので、最適なペナルティ値を選んでモデルをチューニングしたいと考えています。基本的な recipe と、traintest の分割は環境に読み込まれています。

この演習はコースの一部です

Rで学ぶ特徴量エンジニアリング

コースを見る

演習の手順

  • ペナルティが自動でチューニングされるようにモデルを設定します。
  • レベル数が30のペナルティグリッドを設定します。

実践的なインタラクティブ演習

このサンプルコードを完成させて、この演習に挑戦してみましょう。

# Set up your model so that the penalty is tuned automatically
model_lasso_tuned <- logistic_reg() %>% set_engine("glmnet") %>%
  set_args(mixture = 1, ___ = ___) 
workflow_lasso_tuned <- workflow() %>%
  add_model(model_lasso_tuned) %>%
  add_recipe(recipe)

# Configure a penalty grid with 30 levels
penalty_grid <- grid_regular(penalty(range = c(-3, 1)), ___ = ___)

tune_output <- tune_grid(workflow_lasso_tuned,
  resamples = vfold_cv(train, v = 5),
  metrics = metric_set(roc_auc),grid = penalty_grid)

autoplot(tune_output)
コードを編集して実行