validate_prediction_size function

Ensure that predictions have the correct number of rows