Skip to content

Files

Latest commit

 

History

History

i_Clustering

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 

Clustering

Implemented KMeans

Visualization

KMeans

Example

from Util.Util import DataUtil
from i_Clustering.KMeans import KMeans

x, y = DataUtil.gen_two_clusters()      # Generate two clusters
k_means = KMeans()
k_means.fit(x)                          # Train KMeans (n_clusters: 2)
k_means.visualize2d(x, y, dense=400, extra=k_means["centers"])
                                        # Visualize result (2d)
                                        # Rendering KMeans centers by setting 'extra=k_means["centers"]')

Result

KMeans