最良のモデルを選ぶ
この演習では、複数の分類器を比較し、最も良い性能のものを選びます。
ここで扱うデータセット(すでに読み込み済みで、学習用とテスト用に分割されています)は、Pokémon の各ポケモンについて、ステータス、タイプ、そして伝説かどうかの情報で構成されています。分類器の目的は、この 'Legendary' 変数を予測することです。
学習データには3つの個別の分類器が学習済みです。
clf_lrはロジスティック回帰です。clf_dtは決定木です。clf_knnは 5-nearest neighbors 分類器です。
このデータではクラス不均衡があり(800体のうち伝説は65体のみ)、評価指標には F1-Score を用います。Scikit-learn の f1_score() はインポート済みです。
この演習はコースの一部です
Pythonで学ぶアンサンブル手法
実践的なインタラクティブ演習
このサンプルコードを完成させて、この演習に挑戦してみましょう。
# Predict the labels of the test set
pred_lr = ____
pred_dt = ____
pred_knn = ____