[ https://issues.apache.org/jira/browse/SPARK-9231?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=14647144#comment-14647144 ]
yuhao yang commented on SPARK-9231: ----------------------------------- I'd like to send a PR on this soon since it's targeting 1.5. > DistributedLDAModel method for top topics per document > ------------------------------------------------------ > > Key: SPARK-9231 > URL: https://issues.apache.org/jira/browse/SPARK-9231 > Project: Spark > Issue Type: New Feature > Components: MLlib > Reporter: Joseph K. Bradley > Priority: Minor > Original Estimate: 48h > Remaining Estimate: 48h > > Helper method in DistributedLDAModel of this form: > {code} > /** > * For each document, return the top k weighted topics for that document. > * @return RDD of (doc ID, topic indices, topic weights) > */ > def topTopicsPerDocument(k: Int): RDD[(Long, Array[Int], Array[Double])] > {code} > I believe the above method signature will be Java-friendly. -- 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