Actually, dont do a loop. But please make sure that training and test data does not overlap