parrt / dtreeviz

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

Handle string values for target variable #272

Open tlapusan opened 1 year ago

tlapusan commented 1 year ago

This issue was created based on https://github.com/parrt/dtreeviz/issues/271

There could be situations when the target variable is a string in the pandas dataframe, but dtreeviz requires to be numeric. We need to find a solution for this, either to check if we can support string values or raise an exception in case the target variable is a string and guide the user to convert it to numeric.