Enhanced ctree_feature_space() that accepts a features arg rather than requiring a model with just 1 or 2 features.

[parrt]

Signed-off-by: Terence Parr parrt@antlr.org

[parrt]

ok, ready for review! I updated sklearn notebook.

[mepland]

I think we should turn the y-axis on for univar when gtype='barstacked' otherwise you can't tell anything from the magnitude.

[parrt]

I guess it’s just a count, right?

[tlapusan]

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 :

[mepland]

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?

[tlapusan]

I took the code from this PR and there is reinitialization for both classification and regression... @mepland can you check this also please ?

[parrt]

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

[mepland]

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]

@parrt looks good to me now, but I would still like the option of having a y-axis for gtype='barstacked'.

Fixed

[parrt]

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.

[parrt]

Cool. I will push out a new version today.