supertree - ferramenta interativa de visualização de árvores de decisão
(github.com/mljar)- Pacote Python para visualizar árvores de decisão de forma interativa dentro de notebooks Jupyter, JupyterLab e Google Colab
- Permite fazer zoom, pan e recolher (collapse) nós, além de rastrear o caminho de amostras diretamente no notebook
- Integração com as principais bibliotecas de machine learning, como scikit-learn, XGBoost, LightGBM e ONNX
- Oferece amplo suporte a classificadores e modelos de regressão, e consegue visualizar não só árvores individuais, mas também ensembles de árvores
- No exemplo com Random Forest Regressor, é possível especificar e exibir a árvore de um índice específico
- No scikit-learn, há amplo suporte a classificadores e modelos de regressão como
DecisionTreeClassifier,RandomForestClassifiereGradientBoostingRegressor - No LightGBM, é compatível com
LGBMClassifier,LGBMRegressoreBooster; no XGBoost, comXGBClassifier,XGBRegressoreBooster - Open source sob a licença Apache License 2.0
Ainda não há comentários.