fix setParent

apache · Feb 9, 2016 · 3082686 · 3082686
1 parent a37e285
commit 3082686
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mllib/src/main/scala/org/apache/spark/ml/regression/GeneralizedLinearRegression.scala b/mllib/src/main/scala/org/apache/spark/ml/regression/GeneralizedLinearRegression.scala
@@ -211,7 +211,7 @@ class GeneralizedLinearRegression @Since("2.0.0") (@Since("2.0.0") override val
     val irlsModel = optimizer.fit(instances)
 
     val model = copyValues(new GeneralizedLinearRegressionModel(uid,
-      irlsModel.coefficients, irlsModel.intercept))
+      irlsModel.coefficients, irlsModel.intercept).setParent(this))
     model
   }