[SPARK-1406] Fixed extreme cases for logit

ThalesGroup · Apr 28, 2015 · 30165c4 · 30165c4
1 parent 7a5e0ec
commit 30165c4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...c/main/scala/org/apache/spark/mllib/pmml/export/BinaryClassificationPMMLModelExport.scala b/...c/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)
        }