Skip to content

Commit 97c3f6a

Browse files
committed
fix validateParams
1 parent 378ad6c commit 97c3f6a

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

mllib/src/main/scala/org/apache/spark/ml/regression/GeneralizedLinearRegression.scala

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -69,9 +69,11 @@ private[regression] trait GeneralizedLinearRegressionParams extends PredictorPar
6969

7070
@Since("2.0.0")
7171
override def validateParams(): Unit = {
72-
require(GeneralizedLinearRegression.supportedFamilyLinkPairs.contains($(family) -> $(link)),
73-
s"Generalized Linear Regression with ${$(family)} family does not support ${$(link)} " +
74-
s"link function.")
72+
if (isDefined(link)) {
73+
require(GeneralizedLinearRegression.supportedFamilyLinkPairs.contains($(family) -> $(link)),
74+
s"Generalized Linear Regression with ${$(family)} family does not support ${$(link)} " +
75+
s"link function.")
76+
}
7577
}
7678
}
7779

0 commit comments

Comments
 (0)