Skip to content

Commit c3c8da5

Browse files
committed
small cleanup
1 parent 934f97b commit c3c8da5

File tree

1 file changed

+0
-13
lines changed

1 file changed

+0
-13
lines changed

mllib/src/main/scala/org/apache/spark/ml/impl/estimator/Predictor.scala

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -80,23 +80,10 @@ abstract class Predictor[
8080
M <: PredictionModel[FeaturesType, M]]
8181
extends Estimator[M] with PredictorParams {
8282

83-
// TODO: Eliminate asInstanceOf and see if that works.
8483
def setLabelCol(value: String): Learner = set(labelCol, value).asInstanceOf[Learner]
8584
def setFeaturesCol(value: String): Learner = set(featuresCol, value).asInstanceOf[Learner]
8685
def setPredictionCol(value: String): Learner = set(predictionCol, value).asInstanceOf[Learner]
8786

88-
/*
89-
// This will be useful for boosting.
90-
protected def selectLabelColumn(dataset: SchemaRDD, paramMap: ParamMap): RDD[Double] = {
91-
import dataset.sqlContext._
92-
val map = this.paramMap ++ paramMap
93-
dataset.select(map(labelCol).attr).map {
94-
case Row(label: Double) => label
95-
case Row(label: Int) => label.toDouble
96-
}
97-
}
98-
*/
99-
10087
/**
10188
* :: DeveloperApi ::
10289
*

0 commit comments

Comments
 (0)