supertree - 디시젼 트리 인터랙티브 시각화 도구
요약
supertree는 Jupyter, Google Colab 환경에서 의사결정 트리를 인터랙티브하게 시각화할 수 있는 Python 패키지입니다. scikit-learn, XGBoost, LightGBM 등 주요 라이브러리와 연동되어 노드 확대, 이동, 접기 기능을 제공합니다.
핵심 포인트
- Jupyter 및 Colab 환경 내 인터랙티브 시각화 지원
- scikit-learn, XGBoost, LightGBM 등 주요 ML 라이브러리 호환
- 단일 트리뿐만 아니라 트리 앙상블 모델 시각화 가능
- 노드 확대, 이동, 접기 및 샘플 경로 추적 기능 제공
-
의사결정 트리를
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 기반 오픈소스
댓글과 토론
AI 자동 생성 콘텐츠
본 콘텐츠는 GeekNews의 원문을 AI가 자동으로 요약·번역·분석한 것입니다. 원 저작권은 원저작자에게 있으며, 정확한 내용은 반드시 원문을 확인해 주세요.
원문 바로가기0