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]与我们联系。