parrt / dtreeviz

A python library for decision tree visualization and model interpretation.
MIT License
2.89k stars 333 forks source link

Support arbitrary target label values instead of 0..n-1 #246

Open parrt opened 1 year ago

parrt commented 1 year ago

See https://github.com/parrt/dtreeviz/issues/238

parrt commented 1 year ago

See comment in decision_boundaries():

    TODO: assumes classes are contiguous and 0..k-1
parrt commented 1 year ago

This looks to be an issue even in view()