Changed PredictBatch to take in RowMatrix types