Issue with plotting.stage_histogram

[raghav90]

I think the "stage_histogram" function has a minor bug. For a given class, the sum of its fractions across the assigned stages exceeds 1. I think this can be addressed by fixing the input "bins" parameter when calling the "ax.hist" function.

[noxtoby]

Thanks. Does this also happen in the kde_ebm package? It's more up to date than this one

[raghav90]

Haven't tried it yet but I suspect it may happen there too because the issue seems to be with the function "ax.hist" (and how it behaves when the "bins" parameter is an integer). I set bins to np.arange(max_stage+2) and it works fine at my end.