Re: obtain cluster assignment in K-means

2015-02-12 Thread Shi Yu
Thanks Robin, got it. On Thu, Feb 12, 2015 at 2:21 AM, Robin East 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, 2nd is RDD[Vect

Re: obtain cluster assignment in K-means

2015-02-12 Thread Robin East
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 wrote: > Hi there, > > I am new to spark.