File tree Expand file tree Collapse file tree 2 files changed +1
-22
lines changed
mllib/src/main/scala/org/apache/spark/ml/classification Expand file tree Collapse file tree 2 files changed +1
-22
lines changed Original file line number Diff line number Diff 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 *
Original file line number Diff line number Diff line change @@ -19,8 +19,8 @@ package org.apache.spark.ml.classification
1919
2020import org .apache .spark .annotation .Since
2121import org .apache .spark .ml .linalg .{DenseVector , Vector , VectorUDT }
22- import org .apache .spark .ml .param .shared ._
2322import org .apache .spark .ml .param .ParamMap
23+ import org .apache .spark .ml .param .shared ._
2424import org .apache .spark .ml .util .SchemaUtils
2525import org .apache .spark .sql .{DataFrame , Dataset }
2626import org .apache .spark .sql .functions ._
You can’t perform that action at this time.
0 commit comments