Decision Tree Visualizer
v1.0.1
決策樹可視化器是一個功能強大的庫,可讓您輕鬆地可視化Sklearn決策樹分類器。它提供了提取有關樹結構和規則的有用信息的功能,並生成HTML文件以可視化決策樹。
要安裝庫,請使用PIP:
pip install d-treevis
要開始,請導入庫並使用create_tree和create_sankey函數:
import d_treevis as dtv接下來,將數據集上的Sklearn決策樹分類器安裝到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 )有關詳細的文檔和示例,請參閱決策樹可視化器文檔。
該圖書館已獲得MIT許可證的許可。
歡迎捐款!在提交拉動請求之前,請閱讀貢獻指南。
如果您有任何疑問或建議,請隨時通過[email protected]與我們聯繫。