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 | |
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.
-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 |