1. 学ぶ
    2. /
  2. コース
    3. /
  3. Pythonで学ぶstatsmodelsによる回帰入門
Connected

演習

混同行列のモザイクプロットを描く

パフォーマンス行列を手計算するのは楽しいかもしれませんが、複数のモデルで混同行列が必要になると手間がかかります。幸い、.pred_table() メソッドを使えば混同行列を自動で計算できます。

さらに、.pred_table() の出力を mosaic() 関数に渡して、混同行列を可視化できます。

churn と mdl_churn_vs_relationship が利用可能です。

指示

100 XP
  • statsmodels.graphics.mosaicplot から mosaic() 関数をインポートします。
  • .pred_table() メソッドで conf_matrix を作成し、印字します。
  • 混同行列のモザイクプロットを描画します。