[SPARK-9231] DistributedLDAModel method for top topics per document

• Type: New Feature
• Status: Resolved
• Priority: Minor
• Resolution: Fixed
• Fix Version/s: 1.5.0
• Component/s: MLlib
• Target Version/s: 1.5.0
• Assignee: yuhao yang
• Reporter: Joseph K. Bradley
• Shepherd: Joseph K. Bradley
• Created: 21/Jul/15 20:38
• Updated: 31/Jul/15 18:50
• Resolved: 31/Jul/15 18:50

Description

Helper method in DistributedLDAModel of this form:

/**
 * 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])]

I believe the above method signature will be Java-friendly.

Issue Links

• is required by: SPARK-5572 LDA improvement listing
• links to: [Github] Pull Request #7785 (hhbyyh)