タグ

ブックマーク / analytics-note.xyz (1)

  • scikit-learnの学習済み決定木モデルから学習結果を抽出する | 分析ノート

    scikit-learnで学習した決定木の学習結果を確認するにはライブラリを使うのが便利ですが、 自分でも直接取得してみたかったので方法を調べてみました。 参考: dtreevizで決定木の可視化 graphvizで決定木を可視化 とりあえず、 iris を学習しておきます。dtreevizの記事とパラメーターを揃えたので、 この後の結果はそちらと見比べていただくとわかりやすいです。 ただし、最初の分岐が2パターンあって乱数でどちらになるか決まるので、運が悪いと結果が変わります。 from sklearn.datasets import load_iris from sklearn.tree import DecisionTreeClassifier iris = load_iris() clf = DecisionTreeClassifier(min_samples_split=5) c

  • 1