show_just_path_prediction about dtreeviz HOT 15 CLOSED

Indeed, if I look again at above visualisation, it looks 'scattered'. I will adjust it to the vertical stack version. Let's see how it'll look ;)

from dtreeviz.

Looks pretty cool! I think our goal, though, is to show the path from root to predictor leaf in a big tree as tightly as possible. I wonder what it looks like as a vertical stack of just the decision nodes entered and then the leaf. That would be the smallest footprint, which I think is kind of the goal here. what do you think?

from dtreeviz.

It looks much cleaner now :)

from dtreeviz.

heh, nice work! I like it. Does it work horizontally too? I think we have an option for that.

from dtreeviz.

cool. By horizontally.... do you mean a horizontal stack instead of vertical stack ?

from dtreeviz.

yep, left to right

from dtreeviz.

wouldn't be a problem if the prediction path is very deep ? I think we will end with a 'crowdy' plot.

from dtreeviz.

well, sometimes you want very tall and sometimes very wide. seems like original code can change orientation so this should inherit that; i assume you just cut/paste, right?

from dtreeviz.

I wanted to use as much functionality as possible from dtreeviz(), so I didn't cut/paste. I will take a look for horizontal view. There is little work left also for vertical view, I made a little 'hack' to create the visualisation with vertical view for decision nodes only :)

from dtreeviz.

We can skip horizontal if it gets messy in the code :)

from dtreeviz.

For horizontal view I didn't have to do any extra logic, it was already in dtreeviz() code ;)

Should I make a PR ?

from dtreeviz.

Looks great! Only question is, how do we activate this? Seems like it should be an arg to existing tree viz functions. Oh, I see you have show_just_path. Sounds good! Please do make a PR. Does it work for regression trees too?

from dtreeviz.

I didn't try it on regression trees. I will try it tomorrow.

from dtreeviz.

yey, it works also for regression trees. I will create a PR for this.

from dtreeviz.

great job!

from dtreeviz.