Allow users to set initial model in logistic regression

apache · Mar 2, 2016 · ab05aa6 · ab05aa6
1 parent e42724b
commit ab05aa6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mllib/src/main/scala/org/apache/spark/ml/classification/LogisticRegression.scala b/mllib/src/main/scala/org/apache/spark/ml/classification/LogisticRegression.scala
@@ -250,7 +250,7 @@ class LogisticRegression @Since("1.2.0") (
   private var optInitialModel: Option[LogisticRegressionModel] = None
 
   /** @group setParam */
-  private[spark] def setInitialModel(model: LogisticRegressionModel): this.type = {
+  def setInitialModel(model: LogisticRegressionModel): this.type = {
     this.optInitialModel = Some(model)
     this
   }