reiinakano / scikit-plot

An intuitive library to add plotting functionality to scikit-learn objects.
MIT License
2.43k stars 285 forks source link

plot_pca_2d_projection, order of classes/labels is not preserved #107

Open moritzschaefer opened 4 years ago

moritzschaefer commented 4 years ago

The use of np.unique (which lexicographically sorts the array) in plot_pca_2d_projection leads to a different order of the classes in the plot without any control for the user on the order of labels/colors.