supertree - herramienta interactiva de visualización de árboles de decisión
(github.com/mljar)- Paquete de Python para visualizar árboles de decisión de forma interactiva dentro de notebooks de Jupyter, JupyterLab y Google Colab
- Permite zoom, paneo y colapso de nodos, además de rastrear la ruta de muestras, directamente dentro del notebook
- Se integra con bibliotecas principales de machine learning como scikit-learn, XGBoost, LightGBM y ONNX
- Ofrece amplio soporte para clasificadores y modelos de regresión, y puede visualizar no solo árboles individuales sino también ensambles de árboles
- En el ejemplo de Random Forest Regressor, se puede especificar y mostrar el árbol de un índice determinado
- En scikit-learn, ofrece amplio soporte para clasificadores y modelos de regresión como
DecisionTreeClassifier,RandomForestClassifieryGradientBoostingRegressor - En LightGBM, es compatible con
LGBMClassifier,LGBMRegressoryBooster; en XGBoost, conXGBClassifier,XGBRegressoryBooster - Open source bajo licencia Apache License 2.0
Aún no hay comentarios.