Increase prediction plot batch size