[ https://issues.apache.org/jira/browse/SPARK-5886?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=14334137#comment-14334137 ]
Apache Spark commented on SPARK-5886: ------------------------------------- User 'mengxr' has created a pull request for this issue: https://github.com/apache/spark/pull/4735 > Add LabelIndexer > ---------------- > > Key: SPARK-5886 > URL: https://issues.apache.org/jira/browse/SPARK-5886 > Project: Spark > Issue Type: Sub-task > Components: ML > Reporter: Xiangrui Meng > Assignee: Xiangrui Meng > > `LabelIndexer` takes a column of labels (raw categories) and outputs an > integer column with labels indexed by their frequency. > {code} > va li = new LabelIndexer() > .setInputCol("country") > .setOutputCol("countryIndex") > {code} > In the output column, we should store the label to index map as an ML > attribute. The index should be ordered by frequency, where the most frequent > label gets index 0, to enhance sparsity. > We can discuss whether this should index multiple columns at the same time. -- This message was sent by Atlassian JIRA (v6.3.4#6332) --------------------------------------------------------------------- To unsubscribe, e-mail: issues-unsubscr...@spark.apache.org For additional commands, e-mail: issues-h...@spark.apache.org