KMeans.train actually returns a KMeansModel so you can use predict() method of
the model
e.g. clusters.predict(pointToPredict)
or
clusters.predict(pointsToPredict)
first is a single Vector, 2nd is RDD[Vector]
Robin
On 12 Feb 2015, at 06:37, Shi Yu shiyu@gmail.com wrote:
Hi there,
I
Thanks Robin, got it.
On Thu, Feb 12, 2015 at 2:21 AM, Robin East robin.e...@xense.co.uk wrote:
KMeans.train actually returns a KMeansModel so you can use predict()
method of the model
e.g. clusters.predict(pointToPredict)
or
clusters.predict(pointsToPredict)
first is a single Vector,
Hi there,
I am new to spark. When training a model using K-means using the
following code, how do I obtain the cluster assignment in the next
step?
val clusters = KMeans.train(parsedData, numClusters, numIterations)
I searched around many examples but they mostly calculate the WSSSE.
I am