始める無料で始める

カスタムの学習/テストインデックスを作る

動画で見たとおり、この章では前の章で扱った概念を総動員して、実世界のデータセットに取り組みます。

churn データセットには、さまざまな通信会社の顧客データが含まれており、どの顧客が解約(churn)するかを予測することが課題です。

この章では、glmnetrf という2種類の予測モデルを比較します。まずは、それらを安定して比較できるよう、再利用可能な trainControl オブジェクトを作成しましょう。

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

Rで学ぶ caret を使った Machine Learning

コースを見る

演習の手順

churn_xchurn_y がワークスペースに読み込まれています。

  • createFolds() を使って、目的変数である churn_y に対して 5 分割の CV フォールドを作成します。
  • 作成したフォールドを trainControl() に渡し、モデル比較に使える再利用可能な trainControl を作成します。

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

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

# Create custom indices: myFolds
myFolds <- createFolds(___, k = 5)

# Create reusable trainControl object: myControl
myControl <- trainControl(
  summaryFunction = twoClassSummary,
  classProbs = TRUE, # IMPORTANT!
  verboseIter = TRUE,
  savePredictions = TRUE,
  index = ___
)
コードを編集して実行