Decision Tree Visualizer
v1.0.1
Decision Tree Visualizerは、Sklearn決定ツリー分類子を簡単に視覚化できる強力なライブラリです。ツリー構造とルールに関する有用な情報を抽出するための関数を提供し、決定ツリーを視覚化するためのHTMLファイルを生成します。
ライブラリをインストールするには、PIPを使用してください。
pip install d-treevis
開始するには、ライブラリをインポートし、 create_treeとcreate_sankey関数を使用します。
import d_treevis as dtv次に、データセットにSklearn Decision Tree分類子を取り付けて、 create_tree関数に渡します。
from sklearn . tree import DecisionTreeClassifier
# Fit a Decision Tree Classifier on your dataset
tree_model = DecisionTreeClassifier ()
tree_model . fit ( X , y )
# Visualize the decision tree
visualizer . visualize ( tree_model )ターゲット名と色をカスタマイズして、より良い視覚化：
# Define target names and colors
target_names = [ 'Survived' , 'Not Survived' ]
target_colors = [ 'red' , 'yellow' ]
# Visualize the decision tree with custom target names and colors
visualizer . visualize ( tree_model , target_names = target_names , target_colors = target_colors )詳細なドキュメントと例については、Decision Tree Visualizerドキュメントを参照してください。
このライブラリは、MITライセンスに基づいてライセンスされています。
貢献は大歓迎です！プルリクエストを送信する前に、寄付ガイドラインをお読みください。
ご質問や提案がある場合は、[email protected]でお気軽にお問い合わせください。
