Adding a parameter to plot_confusion_matrix() to hide overlaid counts

[echan5]

Hi @reiinakano,

Thank you for this great repo! I am using plot_confusion_matrix() but my counts are quite large so the overlaid counts end up overlapping each other and result in a cluttered plot. I was wondering if I could submit a pull request to update this function to add a hide_counts parameter to give the option to not plot the counts? I've already forked and created a branch with the changes. Thank you!

[reiinakano]

That sounds good to me. Could you also provide an example image before and after?

[echan5]

Great, and sure, an example image of before vs after would be:

[echan5]

pull request is now here: #90

[echan5]

I tried pip install scikit-plot (after pip uninstall scikit-plot) but it seems like it's not installing the updated version with the new parameter (tried the function but it's not recognizing hide_counts). It however did work if I installed via pip install git+https://github.com/reiinakano/scikit-plot. What needs to be done to make it work the first way? I realized I forgot to increase the version in setup.py and I'm not sure if that's related. Thanks!

[reiinakano]

Hi @echan5, I haven't released a new version of scikit plot yet so you won't get it on pypi. I'll probably do a release over the weekend. For now, please stick with the solution you found, which is installing directly from master

[echan5]

Got it, thank you!

[reiinakano]

New version with your patch is now out. Thanks!