决策树可视化 dtreeviz

MIT
跨平台
2021-12-09
白开水不加糖

dtreeviz 是一个用于决策树可视化和模型解释的 Python 库。目前支持scikit-learnXGBoostSpark MLlibLightGBM树。

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

使用此 Colab notebook 直接进入示例

加载中

暂无资讯

暂无问答

暂无博客

返回顶部
顶部