dtreeviz决策树可视化

dtreeviz 是一个用于决策树可视化和模型解释的 Python 库。目前支持scikit-learn、XGBoost、Spark MLlib和LightGBM树。

可视化的灵感来自于 R2D3 的一个教育动画；机器学习的可视化介绍。通过 dtreeviz，你可以看到特征空间是如何在决策节点分割的，训练样本是如何分布在叶子节点的，树是如何对一个特定的观察进行预测的等等。这些操作对于理解分类或回归决策树的工作原理至关重要。如果你对决策树不熟悉，可以看看 fast.ai 的 Introduction to Machine Learning for Coders MOOC。

使用此 Colab notebook 直接进入示例