File tree Expand file tree Collapse file tree 1 file changed +1
-2
lines changed
mllib/src/main/scala/org/apache/spark/mllib/clustering Expand file tree Collapse file tree 1 file changed +1
-2
lines changed Original file line number Diff line number Diff line change @@ -250,7 +250,6 @@ class LDA private (
250250
251251 /**
252252 * Learn an LDA model using the given dataset, using online variational Bayes (VB) algorithm.
253- * Hoffman, Blei and Bach, “Online Learning for Latent Dirichlet Allocation.” NIPS, 2010.
254253 *
255254 * @param documents RDD of documents, which are term (word) count vectors paired with IDs.
256255 * The term count vectors are "bags of words" with a fixed-size vocabulary
@@ -444,7 +443,7 @@ private[clustering] object LDA {
444443 private val D = documents.count().toInt
445444 val actualBatchNumber = Math .ceil(D .toDouble / batchSize).toInt
446445
447- // Initialize the variational distribution q(beta|lambda)
446+ // Initialize the variational distribution q(beta|lambda)
448447 var lambda = getGammaMatrix(k, vocabSize) // K * V
449448 private var Elogbeta = dirichlet_expectation(lambda) // K * V
450449 private var expElogbeta = exp(Elogbeta ) // K * V
You can’t perform that action at this time.
0 commit comments