Allow users to set initial model in logistic regression

GayathriMurali · GayathriMurali · commit ab05aa61b21a · 2016-03-01T20:22:44.000-08:00
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
   }

Original file line number	Diff line number	Diff line change
`@@ -250,7 +250,7 @@ class LogisticRegression @Since("1.2.0") (`
`250`	`250`	`private var optInitialModel: Option[LogisticRegressionModel] = None`
`251`	`251`
`252`	`252`	`/** @group setParam */`
`253`		`- private[spark] def setInitialModel(model: LogisticRegressionModel): this.type = {`
	`253`	`+ def setInitialModel(model: LogisticRegressionModel): this.type = {`
`254`	`254`	`this.optInitialModel = Some(model)`
`255`	`255`	`this`
`256`	`256`	`}`