Coalesce labels with test data

Also fix test prediction to output Array[Array[Double]]