[ https://issues.apache.org/jira/browse/SPARK-5886?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=14387554#comment-14387554 ]
Joseph K. Bradley commented on SPARK-5886: ------------------------------------------ Was there any discussion about this indexing multiple columns at the same time? I think it should be able to since it sounds easier and more efficient when indexing many columns. > 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