dtreeviz決策樹可視化

dtreeviz 是一個(gè)用于決策樹可視化和模型解釋的 Python 庫。目前支持scikit-learn、XGBoost、Spark MLlib和LightGBM樹。

可視化的靈感來自于 R2D3 的一個(gè)教育動(dòng)畫；機(jī)器學(xué)習(xí)的可視化介紹。通過 dtreeviz，你可以看到特征空間是如何在決策節(jié)點(diǎn)分割的，訓(xùn)練樣本是如何分布在葉子節(jié)點(diǎn)的，樹是如何對一個(gè)特定的觀察進(jìn)行預(yù)測的等等。這些操作對于理解分類或回歸決策樹的工作原理至關(guān)重要。如果你對決策樹不熟悉，可以看看 fast.ai 的 Introduction to Machine Learning for Coders MOOC。

使用此 Colab notebook 直接進(jìn)入示例