Skip to content

Commit bb4539b

Browse files
committed
nit
1 parent b050973 commit bb4539b

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/KMeansModel.scala

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -90,10 +90,10 @@ class KMeansModel (@Since("1.0.0") val clusterCenters: Array[Vector],
9090
@Since("1.0.0")
9191
def predict(points: RDD[Vector]): RDD[Int] = {
9292
val bcCentersWithNorm = points.context.broadcast(clusterCentersWithNorm)
93-
val bcRadii = points.context.broadcast(statistics)
93+
val bcStatistics = points.context.broadcast(statistics)
9494
points.map(p =>
9595
distanceMeasureInstance.findClosest(bcCentersWithNorm.value,
96-
bcRadii.value, new VectorWithNorm(p))._1)
96+
bcStatistics.value, new VectorWithNorm(p))._1)
9797
}
9898

9999
/**

0 commit comments

Comments
 (0)