File tree Expand file tree Collapse file tree 1 file changed +0
-13
lines changed
mllib/src/main/scala/org/apache/spark/ml/impl/estimator Expand file tree Collapse file tree 1 file changed +0
-13
lines changed Original file line number Diff line number Diff 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 *
You can’t perform that action at this time.
0 commit comments