dtreeviz是一个用于决策树可视化和模型解释的Python库。目前支持scikit-learn、XGBoost、SparkMLlib和LightGBM树。
可视化的灵感来自于R2D3的一个教育动画；机器学习的可视化介绍。通过dtreeviz，你可以看到特征空间是如何在决策节点分割的，训练样本是如何分布在叶子节点的，树是如何对一个特定的观察进行预测的等等。这些操作对于理解分类或回归决策树的工作原理至关重要。如果你对决策树不熟悉，可以看看fast.ai的IntroductiontoMachineLearningforCodersMOOC。
使用此Colabnotebook 直接进入示例
评论