Feature Request: Confusion Matrix

[3h4]

Amazing library! Would it be possible to plot a confusion matrix? Perhaps I can do it with the matrix_plot and then put text on top of that?

[piccolomo]

Thanks for the appreciation:-)

Would that be the confusion matrix connected to machine learning?

[3h4]

Yes indeed, that is the case :)

[piccolomo]

Thanks, will answer relatively soon!

[piccolomo]

Hi @3h4, finally working on your idea. Here is a partial result:

I will upload new code soon. Any feedback for now is welcomed.

Thanks

[3h4]

@piccolomo Wow looks perfect! Would it be possible to have more classes than 2, for example 15 different animals or something like that? So you can put a percentage or number in the square and class on the axes. Perhaps there will be not enough room so only numbers can be used? Or single letters. Great work so far!

[piccolomo]

Hi @3h4 , I finally updated plotext including the changes you asked for.

I introduced the function confusion_matrix(), but haven't tested properly so any feedback is welcomed.

The new version is available on GitHib for now and soon also on PyPi. To install follow the indications here.

Your changes are documented here and you have been credited here.

Any feedback is welcomed.

Thanks a lot for your inputs and all the best,
Savino

[3h4]

@piccolomo Omg, Perfect! Is it possible to put it in a subplot as well? What happens if the space becomes too small?

[piccolomo]

I think you can, if the space diminishes the plot tries its best to adapt, if too small some features disappear. Try it if you want and can and see if there are any issues. No worries otherwise, I really haven't played around it much. Thanks :-)

[piccolomo]

I was trying to get some nice blue shaded rectangles, as gray is a bit dull, but could not find a way around it. If anyone can please send some code. Thanks :-)