supertree - デシジョンツリーのインタラクティブ可視化ツール
(github.com/mljar)- Jupyter、JupyterLab、Google Colab ノートブック内で決定木をインタラクティブに可視化する Python パッケージ
- ノードの 拡大(Zoom)・移動(Pan)・折りたたみ(collapse) やサンプル経路の追跡をノートブック内でそのまま実行可能
- scikit-learn、XGBoost、LightGBM、ONNX など主要な機械学習ライブラリと連携
- 分類器と回帰モデルを幅広くサポートし、単一ツリーだけでなく ツリーアンサンブル も可視化可能
- Random Forest Regressor の例では、特定インデックスのツリーを指定して出力可能
- scikit-learn は
DecisionTreeClassifier、RandomForestClassifier、GradientBoostingRegressorなど分類器・回帰モデルを幅広くサポート - LightGBM は
LGBMClassifier、LGBMRegressor、Booster、XGBoost はXGBClassifier、XGBRegressor、Boosterなどと互換 - Apache License 2.0 ベースのオープンソース
まだコメントはありません。