oegedijk / explainerdashboard

Quickly build Explainable AI dashboards that show the inner workings of so-called "blackbox" machine learning models.
http://explainerdashboard.readthedocs.io
MIT License
2.31k stars 332 forks source link

Add support for string labels #274

Open AlexanderZender opened 1 year ago

AlexanderZender commented 1 year ago

I got some datasets which have labels that are string based instead of int. Currently, when these are passed down to explainer dashboard it will break, as the target is converted to a float values. I convert these labels using a label encoder, but this occurs in my model.

It would be a great addition if you could support string labels, it would give more readability to the dashboard with such datasets. As the user would not have to know which value was cast to what in the label encoder, and would also allow the support for label encoding in models/pipeline.