[ https://issues.apache.org/jira/browse/SPARK-4156?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=14242486#comment-14242486 ]
Meethu Mathew commented on SPARK-4156: -------------------------------------- [~tgaloppo] The current version of the code has no predict function to return the cluster labels, i.e, the index of the cluster to which the point has maximum membership.We have written a predict function to return the cluster labels and the membership values.We would be happy to contribute this to your code. cc [~mengxr] > Add expectation maximization for Gaussian mixture models to MLLib clustering > ---------------------------------------------------------------------------- > > Key: SPARK-4156 > URL: https://issues.apache.org/jira/browse/SPARK-4156 > Project: Spark > Issue Type: New Feature > Components: MLlib > Reporter: Travis Galoppo > Assignee: Travis Galoppo > > As an additional clustering algorithm, implement expectation maximization for > Gaussian mixture models -- 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