dtreeviz icon indicating copy to clipboard operation
dtreeviz copied to clipboard

Support for tensorflow - decision forest

Open tlapusan opened this issue 3 years ago • 9 comments

https://blog.tensorflow.org/2021/05/introducing-tensorflow-decision-forests.html

tlapusan avatar Mar 09 '22 14:03 tlapusan

Implemented the basic necessary methods to be able to initialize a shadow tree without errors :) . Still work to be done to add the first visualisations

Screenshot 2022-03-30 at 10 26 43

tlapusan avatar Mar 30 '22 07:03 tlapusan

awesome!

parrt avatar Mar 30 '22 16:03 parrt

First visualisation for RandomForest classification model !

Screenshot 2022-05-02 at 09 56 27

tlapusan avatar May 02 '22 06:05 tlapusan

Woohoo!!!

parrt avatar May 02 '22 15:05 parrt

even more visualisations ! Screenshot 2022-05-07 at 20 06 08

tlapusan avatar May 07 '22 17:05 tlapusan

woot!

parrt avatar May 07 '22 17:05 parrt

And we have the main visualisation implemented also :)

Screenshot 2022-08-03 at 10 52 46

tlapusan avatar Aug 03 '22 07:08 tlapusan

By the way I think in your examples you tend to pass the shadow tree around but nobody knows what that is so I think our example should avoid that unless you are specifically doing something for efficiency reasons.

parrt avatar Aug 05 '22 20:08 parrt

By examples you mean this screenshots or visualisations from the notebook ?

In the notebook I initialize the shadow tree like this : tfdf_shadow = ShadowTFDFTree(model, tree_index=0, x_data=train[features], y_data=train[target], feature_names=features, target_name=target, class_names=[0, 1]) and send it to the visualisations instead of using the other method of sending a long list of parameters for each.

tlapusan avatar Aug 08 '22 14:08 tlapusan

I just meant no one knows what a shadow tree is. Why expose that in normal circumstances to the user?

parrt avatar Aug 13 '22 22:08 parrt

Indeed, if somebody is looking to the above screenshoot, they wouldn't know what a tfdf_shadow is.

But if the user is looking through the notebook visualisations, he/she will have access to the initialization step for tfdf_shadow.

tlapusan avatar Aug 16 '22 06:08 tlapusan

True, but I just want to make sure we show the simplest possible interface for most people. Those were efficiency issues can dig deeper.

parrt avatar Aug 19 '22 00:08 parrt