supertree - Interaktives Visualisierungstool für Decision Trees

• Ein Python-Paket zur interaktiven Visualisierung von Entscheidungsbäumen direkt in Jupyter-, JupyterLab- und Google-Colab-Notebooks
• Zoom, Pan und Collapse von Knoten sowie das Nachverfolgen von Sample-Pfaden direkt innerhalb des Notebooks
• Integration mit wichtigen Machine-Learning-Bibliotheken wie scikit-learn, XGBoost, LightGBM, ONNX
• Unterstützt Klassifikatoren und Regressionsmodelle umfassend; visualisieren lassen sich nicht nur einzelne Bäume, sondern auch Tree Ensembles
  • Im Beispiel mit dem Random Forest Regressor kann ein Baum mit einem bestimmten Index ausgewählt und ausgegeben werden
• Für scikit-learn werden unter anderem DecisionTreeClassifier, RandomForestClassifier, GradientBoostingRegressor sowie weitere Klassifikations- und Regressionsmodelle breit unterstützt
• Kompatibel mit LightGBM-Modellen wie LGBMClassifier, LGBMRegressor, Booster sowie mit XGBoost-Modellen wie XGBClassifier, XGBRegressor, Booster
• Open Source unter der Apache License 2.0