diff options
| author | Xusen Yin <yinxusen@gmail.com> | 2016-01-26 13:18:01 -0800 |
|---|---|---|
| committer | Joseph K. Bradley <joseph@databricks.com> | 2016-01-26 13:18:01 -0800 |
| commit | fbf7623d49525e3aa6b08f482afd7ee8118d80cb (patch) | |
| tree | f4f93ab3fba2787f6a9e3fb22219e4a599413fcc /mllib | |
| parent | 8beab68152348c44cf2f89850f792f164b06470d (diff) | |
| download | spark-fbf7623d49525e3aa6b08f482afd7ee8118d80cb.tar.gz spark-fbf7623d49525e3aa6b08f482afd7ee8118d80cb.tar.bz2 spark-fbf7623d49525e3aa6b08f482afd7ee8118d80cb.zip | |
[SPARK-12952] EMLDAOptimizer initialize() should return EMLDAOptimizer other than its parent class
https://issues.apache.org/jira/browse/SPARK-12952
Author: Xusen Yin <yinxusen@gmail.com>
Closes #10863 from yinxusen/SPARK-12952.
Diffstat (limited to 'mllib')
| -rw-r--r-- | mllib/src/main/scala/org/apache/spark/mllib/clustering/LDAOptimizer.scala | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/clustering/LDAOptimizer.scala b/mllib/src/main/scala/org/apache/spark/mllib/clustering/LDAOptimizer.scala index c19595e6cd..7a41f74191 100644 --- a/mllib/src/main/scala/org/apache/spark/mllib/clustering/LDAOptimizer.scala +++ b/mllib/src/main/scala/org/apache/spark/mllib/clustering/LDAOptimizer.scala @@ -95,7 +95,9 @@ final class EMLDAOptimizer extends LDAOptimizer { /** * Compute bipartite term/doc graph. */ - override private[clustering] def initialize(docs: RDD[(Long, Vector)], lda: LDA): LDAOptimizer = { + override private[clustering] def initialize( + docs: RDD[(Long, Vector)], + lda: LDA): EMLDAOptimizer = { // EMLDAOptimizer currently only supports symmetric document-topic priors val docConcentration = lda.getDocConcentration |