Skip to content

Commit 55b52bd

Browse files
committed
rebase
1 parent 59ced0a commit 55b52bd

File tree

2 files changed

+1
-22
lines changed

2 files changed

+1
-22
lines changed

mllib/src/main/scala/org/apache/spark/ml/classification/LinearSVC.scala

Lines changed: 0 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -394,27 +394,6 @@ class LinearSVCModel private[classification] (
394394
@Since("3.1.0")
395395
override def summary: LinearSVCTrainingSummary = super.summary
396396

397-
/**
398-
* If the rawPrediction and prediction columns are set, this method returns the current model,
399-
* otherwise it generates new columns for them and sets them as columns on a new copy of
400-
* the current model
401-
*/
402-
private[classification] def findSummaryModel(): (LinearSVCModel, String, String) = {
403-
val model = if ($(rawPredictionCol).isEmpty && $(predictionCol).isEmpty) {
404-
copy(ParamMap.empty)
405-
.setRawPredictionCol("rawPrediction_" + java.util.UUID.randomUUID.toString)
406-
.setPredictionCol("prediction_" + java.util.UUID.randomUUID.toString)
407-
} else if ($(rawPredictionCol).isEmpty) {
408-
copy(ParamMap.empty).setRawPredictionCol("rawPrediction_" +
409-
java.util.UUID.randomUUID.toString)
410-
} else if ($(predictionCol).isEmpty) {
411-
copy(ParamMap.empty).setPredictionCol("prediction_" + java.util.UUID.randomUUID.toString)
412-
} else {
413-
this
414-
}
415-
(model, model.getRawPredictionCol, model.getPredictionCol)
416-
}
417-
418397
/**
419398
* Evaluates the model on a test dataset.
420399
*

mllib/src/main/scala/org/apache/spark/ml/classification/ProbabilisticClassifier.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,8 @@ package org.apache.spark.ml.classification
1919

2020
import org.apache.spark.annotation.Since
2121
import org.apache.spark.ml.linalg.{DenseVector, Vector, VectorUDT}
22-
import org.apache.spark.ml.param.shared._
2322
import org.apache.spark.ml.param.ParamMap
23+
import org.apache.spark.ml.param.shared._
2424
import org.apache.spark.ml.util.SchemaUtils
2525
import org.apache.spark.sql.{DataFrame, Dataset}
2626
import org.apache.spark.sql.functions._

0 commit comments

Comments
 (0)