supertree - 디시젼 트리 인터랙티브 시각화 도구

2 days ago 6

▲

(github.com/mljar)

1P by xguru 1분전 | ★ favorite | 댓글과 토론

의사결정 트리를 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 기반 오픈소스