parrt / dtreeviz

A python library for decision tree visualization and model interpretation.
MIT License
2.88k stars 332 forks source link

Add support for Catboost decision tree #116

Open oleg-savko opened 3 years ago

oleg-savko commented 3 years ago

It would be great if support for catboost model would be added.

tlapusan commented 3 years ago

thanks @oleg-savko for mentioning this. We will see when we can implement/add catboost, because the roadmap for dtreeviz is already pretty busy. Meanwhile, if you have time, we will be happy to accept a PR from your side with the implementation :)

yanhong-zhao-ef commented 3 years ago

Would love to have Catboost supported through dtreeviz - will model the xgboost shadow tree class and see if a Catboost shadow tree class can be quickly spun up since Catboost is already using graphviz to plot trees at different nodes https://catboost.ai/docs/concepts/python-reference_catboostclassifier_plot_tree.html

yanhong-zhao-ef commented 3 years ago

having a go at this https://github.com/yanhong-zhao-ef/dtreeviz/tree/catboost-dtree will keep on at it when I got some more time and hope to raise a PR soon :)

yanhong-zhao-ef commented 3 years ago

spent a few hours this morning working on this again and it is almost working for symmetric trees in Catboost https://github.com/yanhong-zhao-ef/dtreeviz/tree/catboost-dtree. Will attach a similar notebook once it runs at least all the functionalities for xgboost as outlined in that notebook!

tlapusan commented 1 year ago

Hi @yanhong-zhao-ef, do you have any updates on this ? Thanks!

mkleinbort-ic commented 1 year ago

Hi all, any updates? Would love this feature

parrt commented 1 year ago

No updates from our side...

Thegongyx commented 8 months ago

+1