diff --git a/mllib/src/main/scala/org/apache/spark/mllib/pmml/export/BinaryClassificationPMMLModelExport.scala b/mllib/src/main/scala/org/apache/spark/mllib/pmml/export/BinaryClassificationPMMLModelExport.scala index 58f4b52f1b497..e469a7ac9bc85 100644 --- a/mllib/src/main/scala/org/apache/spark/mllib/pmml/export/BinaryClassificationPMMLModelExport.scala +++ b/mllib/src/main/scala/org/apache/spark/mllib/pmml/export/BinaryClassificationPMMLModelExport.scala @@ -49,9 +49,9 @@ private[mllib] class BinaryClassificationPMMLModelExport( var interceptNO = threshold if (RegressionNormalizationMethodType.LOGIT == normalizationMethod) { if (threshold <= 0) - interceptNO = 1000 - else if (threshold >= 1) interceptNO = -1000 + else if (threshold >= 1) + interceptNO = 1000 else interceptNO = -math.log(1/threshold -1) }