[jira] [Created] (SPARK-17151) Decide how to handle inferring number of classes in Multinomial logistic regression

["Seth Hendrickson (JIRA)" <ji...@apache.org>]

Seth Hendrickson created SPARK-17151:
----------------------------------------

             Summary: Decide how to handle inferring number of classes in Multinomial logistic regression
                 Key: SPARK-17151
                 URL: https://issues.apache.org/jira/browse/SPARK-17151
             Project: Spark
          Issue Type: Sub-task
            Reporter: Seth Hendrickson
            Priority: Minor


This JIRA is to discuss how the number of label classes should be inferred in multinomial logistic regression. Currently, MLOR checks the dataframe metadata and if the number of classes is not specified then it uses the maximum value seen in the label column. If the labels are not properly indexed, then this can cause a large number of zero coefficients and potentially produce instabilities in model training.



--
This message was sent by Atlassian JIRA
(v6.3.4#6332)

---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscribe@spark.apache.org
For additional commands, e-mail: issues-help@spark.apache.org