Skip to content

Commit 32b23b8

Browse files
committed
fixed usage of checkpointer in lda
1 parent 0b3dbc0 commit 32b23b8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

mllib/src/main/scala/org/apache/spark/mllib/clustering/LDAOptimizer.scala

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -142,8 +142,8 @@ final class EMLDAOptimizer extends LDAOptimizer {
142142
this.k = k
143143
this.vocabSize = docs.take(1).head._2.size
144144
this.checkpointInterval = lda.getCheckpointInterval
145-
this.graphCheckpointer = new
146-
PeriodicGraphCheckpointer[TopicCounts, TokenCount](graph, checkpointInterval)
145+
this.graphCheckpointer = new PeriodicGraphCheckpointer[TopicCounts, TokenCount](
146+
checkpointInterval, graph.vertices.sparkContext)
147147
this.globalTopicTotals = computeGlobalTopicTotals()
148148
this
149149
}

0 commit comments

Comments
 (0)