Use featurizedData.numRows to calculate trainError.

This avoids expensive recomputation of trainVsTest.