diff options
author | Yuhao Yang <yuhao.yang@intel.com> | 2016-07-06 01:30:47 -0700 |
---|---|---|
committer | Yanbo Liang <ybliang8@gmail.com> | 2016-07-06 01:30:47 -0700 |
commit | 5497242c769b40338bfa57d64f2c64996dfa57e8 (patch) | |
tree | 2685f49aec90a0d7d2d12ce970db5999aacdd16f | |
parent | 5f342049cce9102fb62b4de2d8d8fa691c2e8ac4 (diff) | |
download | spark-5497242c769b40338bfa57d64f2c64996dfa57e8.tar.gz spark-5497242c769b40338bfa57d64f2c64996dfa57e8.tar.bz2 spark-5497242c769b40338bfa57d64f2c64996dfa57e8.zip |
[SPARK-16249][ML] Change visibility of Object ml.clustering.LDA to public for loading
## What changes were proposed in this pull request?
jira: https://issues.apache.org/jira/browse/SPARK-16249
Change visibility of Object ml.clustering.LDA to public for loading, thus users can invoke LDA.load("path").
## How was this patch tested?
existing ut and manually test for load ( saved with current code)
Author: Yuhao Yang <yuhao.yang@intel.com>
Author: Yuhao Yang <hhbyyh@gmail.com>
Closes #13941 from hhbyyh/ldapublic.
-rw-r--r-- | mllib/src/main/scala/org/apache/spark/ml/clustering/LDA.scala | 10 |
1 files changed, 6 insertions, 4 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 b333d59258..778cd0fee7 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 @@ -880,11 +880,13 @@ class LDA @Since("1.6.0") ( } } - -private[clustering] object LDA extends DefaultParamsReadable[LDA] { +@Since("2.0.0") +object LDA extends DefaultParamsReadable[LDA] { /** Get dataset for spark.mllib LDA */ - def getOldDataset(dataset: Dataset[_], featuresCol: String): RDD[(Long, OldVector)] = { + private[clustering] def getOldDataset( + dataset: Dataset[_], + featuresCol: String): RDD[(Long, OldVector)] = { dataset .withColumn("docId", monotonicallyIncreasingId()) .select("docId", featuresCol) @@ -894,6 +896,6 @@ private[clustering] object LDA extends DefaultParamsReadable[LDA] { } } - @Since("1.6.0") + @Since("2.0.0") override def load(path: String): LDA = super.load(path) } |