Closed parrt closed 1 year ago
ok, ready for review! I updated sklearn notebook.
I think we should turn the y-axis on for univar
when gtype='barstacked'
otherwise you can't tell anything from the magnitude.
I guess it’s just a count, right?
Is there a need to reinitialize the decision tree/dtreeviz in case we want to use one and after that two features ? I think what @mepland initially suggested is to initialize a decision tree/dtreeviz model and after that to choose if you want to make a viz with one or two features. Like in the bellow screenshot :
Is there a need to reinitialize the decision tree/dtreeviz in case we want to use one and after that two features ?
I think that is the current behavior, the screenshot looks fine to me?
I took the code from this PR and there is reinitialization for both classification and regression... @mepland can you check this also please ?
Good catch @tlapusan. Sorry about that. cleaned up and resubmitted the notebook so that it creates a single decision tree and model, then reuses that for visualization
I see what you mean now @tlapusan.
@parrt looks good to me now, but I would still like the option of having a y-axis for gtype='barstacked'
.
@parrt looks good to me now, but I would still like the option of having a y-axis for
gtype='barstacked'
.
Fixed
OK, can you guys do one last quick check? If it looks good I can push it for a quick release so I can use it within colab.
Cool. I will push out a new version today.
Signed-off-by: Terence Parr parrt@antlr.org