search

oegedijk / explainerdashboard

Quickly build Explainable AI dashboards that show the inner workings of so-called "blackbox" machine learning models.
http://explainerdashboard.readthedocs.io
MIT License
2.29k stars 328 forks source link

Contribution: tree visualisation for lightgbm models #118

Open oegedijk opened 3 years ago

oegedijk commented 3 years ago

Recently dtreeviz has added support for lightgbm models: https://github.com/parrt/dtreeviz/

So it would be cool to add the same tree visualization in explainerdashboard that already exist for RandomForest, ExtraTrees and XGBoost models.

If someone wants to pitch in to help extract individual predictions of decision trees inside lightgbm boosters and then get them in shape to be used by the dashboard that would be great!

kurucan commented 2 years ago

This feature would be great !