Try different size training and testsets