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 (
250
250
251
251
/**
252
252
* 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.
254
253
*
255
254
* @param documents RDD of documents, which are term (word) count vectors paired with IDs.
256
255
* The term count vectors are "bags of words" with a fixed-size vocabulary
@@ -444,7 +443,7 @@ private[clustering] object LDA {
444
443
private val D = documents.count().toInt
445
444
val actualBatchNumber = Math .ceil(D .toDouble / batchSize).toInt
446
445
447
- // Initialize the variational distribution q(beta|lambda)
446
+ // Initialize the variational distribution q(beta|lambda)
448
447
var lambda = getGammaMatrix(k, vocabSize) // K * V
449
448
private var Elogbeta = dirichlet_expectation(lambda) // K * V
450
449
private var expElogbeta = exp(Elogbeta ) // K * V
You can’t perform that action at this time.
0 commit comments