diff options
| author | Zheng RuiFeng <ruifengz@foxmail.com> | 2017-02-22 16:33:14 +0200 |
|---|---|---|
| committer | Nick Pentreath <nickp@za.ibm.com> | 2017-02-22 16:33:14 +0200 |
| commit | ef3c73535f0b521905839f2050a77a99fd4c36cd (patch) | |
| tree | 3380577c9934bf5e1715e13f40bfb74baf0ee296 /mllib | |
| parent | 1a45d2b2cc6466841fb73da21a61b61f14a5d5fb (diff) | |
| download | spark-ef3c73535f0b521905839f2050a77a99fd4c36cd.tar.gz spark-ef3c73535f0b521905839f2050a77a99fd4c36cd.tar.bz2 spark-ef3c73535f0b521905839f2050a77a99fd4c36cd.zip | |
[SPARK-19694][ML] Add missing 'setTopicDistributionCol' for LDAModel
## What changes were proposed in this pull request?
Add missing 'setTopicDistributionCol' for LDAModel
## How was this patch tested?
existing tests
Author: Zheng RuiFeng <ruifengz@foxmail.com>
Closes #17021 from zhengruifeng/lda_outputCol.
Diffstat (limited to 'mllib')
| -rw-r--r-- | mllib/src/main/scala/org/apache/spark/ml/clustering/LDA.scala | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/mllib/src/main/scala/org/apache/spark/ml/clustering/LDA.scala b/mllib/src/main/scala/org/apache/spark/ml/clustering/LDA.scala index bbcef3502d..55720e2d61 100644 --- a/mllib/src/main/scala/org/apache/spark/ml/clustering/LDA.scala +++ b/mllib/src/main/scala/org/apache/spark/ml/clustering/LDA.scala @@ -437,6 +437,9 @@ abstract class LDAModel private[ml] ( @Since("1.6.0") def setFeaturesCol(value: String): this.type = set(featuresCol, value) + @Since("2.2.0") + def setTopicDistributionCol(value: String): this.type = set(topicDistributionCol, value) + /** @group setParam */ @Since("1.6.0") def setSeed(value: Long): this.type = set(seed, value) |