aboutsummaryrefslogtreecommitdiff
path: root/mllib
diff options
context:
space:
mode:
authorYuhao Yang <yuhao.yang@intel.com>2016-07-06 01:30:47 -0700
committerYanbo Liang <ybliang8@gmail.com>2016-07-06 01:30:47 -0700
commit5497242c769b40338bfa57d64f2c64996dfa57e8 (patch)
tree2685f49aec90a0d7d2d12ce970db5999aacdd16f /mllib
parent5f342049cce9102fb62b4de2d8d8fa691c2e8ac4 (diff)
downloadspark-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.
Diffstat (limited to 'mllib')
-rw-r--r--mllib/src/main/scala/org/apache/spark/ml/clustering/LDA.scala10
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)
}