dtreeviz icon indicating copy to clipboard operation
dtreeviz copied to clipboard

Add support for Catboost decision tree

Open oleg-savko opened this issue 4 years ago • 8 comments

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

oleg-savko avatar Jan 04 '21 20:01 oleg-savko

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 :)

tlapusan avatar Jan 06 '21 13:01 tlapusan

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 avatar Jan 07 '21 16:01 yanhong-zhao-ef

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 avatar Jan 11 '21 23:01 yanhong-zhao-ef

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!

yanhong-zhao-ef avatar Jan 20 '21 18:01 yanhong-zhao-ef

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

tlapusan avatar Aug 17 '22 08:08 tlapusan

Hi all, any updates? Would love this feature

mkleinbort-ic avatar Jan 19 '23 16:01 mkleinbort-ic

No updates from our side...

parrt avatar Jan 22 '23 18:01 parrt

+1

Thegongyx avatar Oct 11 '23 04:10 Thegongyx