supertree - Công cụ trực quan hóa tương tác cây quyết định
(github.com/mljar)- Gói Python trực quan hóa tương tác cây quyết định ngay bên trong notebook Jupyter, JupyterLab, Google Colab
- Có thể thực hiện trực tiếp trong notebook các thao tác phóng to (Zoom)·di chuyển (Pan)·thu gọn (collapse) nút và theo dõi đường đi của mẫu
- Tích hợp với các thư viện máy học phổ biến như scikit-learn, XGBoost, LightGBM, ONNX
- Hỗ trợ rộng rãi các mô hình phân loại và hồi quy, đồng thời có thể trực quan hóa không chỉ cây đơn mà cả tổ hợp cây
- Trong ví dụ Random Forest Regressor, có thể chỉ định cây ở một chỉ mục cụ thể để hiển thị
- Với scikit-learn, hỗ trợ rộng rãi các mô hình phân loại và hồi quy như
DecisionTreeClassifier,RandomForestClassifier,GradientBoostingRegressor - Tương thích với LightGBM như
LGBMClassifier,LGBMRegressor,Booster, và với XGBoost nhưXGBClassifier,XGBRegressor,Booster - Mã nguồn mở theo giấy phép Apache License 2.0
Chưa có bình luận nào.