Should we make the return type RDD[Boolean] for classification ? If not, this looks identical to the RegressionModel and we should unify them.