correct code to have training set of random subsample of images 1 and 2 and image 3 as test set