From 8f1112f370032d3cbfbe0edf83eb72187d904664 Mon Sep 17 00:00:00 2001
From: guptesp <guptesp@users.noreply.github.com>
Date: Thu, 21 Jul 2016 18:45:20 -0400
Subject: [PATCH] Changes the numClasses check to !=2 (Line 306)

If numClasses is less than 2, it gives a array out of index error.
---
 .../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 7694773c816b..ba5f8d70c819 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
@@ -303,7 +303,7 @@ class LogisticRegression @Since("1.2.0") (
         throw new SparkException(msg)
       }
 
-      if (numClasses > 2) {
+      if (numClasses != 2) {
         val msg = s"Currently, LogisticRegression with ElasticNet in ML package only supports " +
           s"binary classification. Found $numClasses in the input dataset."
         logError(msg)