instead of shuffle, np.random.choice, max 10000 samples