始める無料で始める

アプリの評価を予測する

前の演習で Google アプリのデータセットを確認しました。ここでは、その特徴量の一部からアプリの評価を予測するモデルを作成します。

そのために、scikit-learnDecisionTreeRegressor を使います。決定木は多くのアンサンブルモデルの基本構成要素です。この仕組みを復習しておくと、コース全体で必ず役に立ちます。

評価指標には MAE(平均絶対誤差)を使います。これは、実際の評価と予測された評価の絶対差の平均を表すため、とても解釈しやすい指標です。

必要なモジュールはすべてインポート済みです。特徴量と目的変数は、それぞれ変数 Xy に用意されています。

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

Pythonで学ぶアンサンブル手法

コースを見る

演習の手順

  • train_test_split() を使って、Xy を学習用とテスト用に分割します。テストサイズは 20%(0.2)にしてください。
  • 次のハイパーパラメータを指定して DecisionTreeRegressor() をインスタンス化し、reg_dt として作成します: min_samples_leaf = 3 および min_samples_split = 9
  • .fit() を使って学習データに回帰器を学習させます。
  • .predict() を使ってテストセットのラベルを予測します。

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

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

# Split into train (80%) and test (20%) sets
X_train, X_test, y_train, y_test = ____(____, ____, ____, random_state=42)

# Instantiate the regressor
reg_dt = ____(____, ____, random_state=500)

# Fit to the training set
____

# Evaluate the performance of the model on the test set
y_pred = ____
print('MAE: {:.3f}'.format(mean_absolute_error(y_test, y_pred)))
コードを編集して実行