From ab05aa61b21a23531e43d082757a40bf2ab750d8 Mon Sep 17 00:00:00 2001 From: GayathriMurali Date: Tue, 1 Mar 2016 20:22:44 -0800 Subject: [PATCH] Allow users to set initial model in logistic regression --- .../org/apache/spark/ml/classification/LogisticRegression.scala | 2 +- 1 file changed, 1 insertion(+), 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 index 0d329d2c08d5..9aca2625c77c 100644 --- 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 }