supertree – outil de visualisation interactive d’arbres de décision

• Package Python pour visualiser de manière interactive des arbres de décision directement dans des notebooks Jupyter, JupyterLab et Google Colab
• Permet de zoomer, déplacer (pan) et replier (collapse) les nœuds, ainsi que de suivre le chemin d’un échantillon, directement dans le notebook
• S’intègre aux principales bibliothèques de machine learning, notamment scikit-learn, XGBoost, LightGBM et ONNX
• Prend en charge un large éventail de classifieurs et de modèles de régression, et permet de visualiser non seulement des arbres individuels, mais aussi des ensembles d’arbres
  • Dans l’exemple Random Forest Regressor, il est possible d’indiquer l’arbre d’un index donné à afficher
• Pour scikit-learn, prend largement en charge des classifieurs et modèles de régression tels que DecisionTreeClassifier, RandomForestClassifier et GradientBoostingRegressor
• Compatible avec LGBMClassifier, LGBMRegressor et Booster pour LightGBM, ainsi qu’avec XGBClassifier, XGBRegressor et Booster pour XGBoost
• Open source sous licence Apache License 2.0