始める無料で始める

制約あり・なしの決定木

この演習では、前のチャプターで扱ったポケモンのデータセットに戻ります。目的は、与えられたポケモンが伝説(legendary)かどうかを予測することでした。

ここでは、2つの別々の決定木分類器を作成します。1つ目では、min_samples_leafmin_samples_split のパラメータを指定しますが、最大深さは設定せず、木が制約なく十分に成長できるようにします。

2つ目では、決定木の深さを制限していくつかの制約を設けます。2つのモデルを比較することで、「弱い」学習器という概念をよりよく理解できるようになります。

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

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

コースを見る

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

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

# Build unrestricted decision tree
clf = ____
clf.fit(X_train, y_train)

# Predict the labels
pred = clf.predict(X_test)

# Print the confusion matrix
cm = confusion_matrix(y_test, pred)
print('Confusion matrix:\n', cm)

# Print the F1 score
score = f1_score(y_test, pred)
print('F1-Score: {:.3f}'.format(score))
コードを編集して実行