I can do that yes, it will take a few days though just a fair warning. I don’t have an equal number of samples because as k approaches zero the diversity in images increases exponentially, so the number of samples with k<0.6 is heavily over populated whereas the number of samples for say k~0.9 is less because the network has been shown to predict these images well in unseen data.