[ 
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

Reply via email to