search

src-d / kmcuda

Large scale K-means and K-nn implementation on NVIDIA GPU / CUDA
Other
784 stars 144 forks source link

hi, i want to use custom initial cluster center to #41

Open luobo11 opened 6 years ago

luobo11 commented 6 years ago

initCluCenter=np.array([[0,0],[4,8],[6,3]]).astype(np.float32) centroids, assignments, avg_distance = kmeanscuda(samplesMatrix,len(initCluCenter), init=initCluCenter, verbosity=0, seed=3)

vmarkovtsev commented 6 years ago

What if you try

initCluCenter=np.array([[0,0],[4,8],[6,3]]).astype(np.float32)