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 기반 오픈소스