from miniML.clustering import KMeans
kmeans = KMeans(n_clusters = 3)
kmeans.fit(X)
# predict to which cluster each value belongs
predictions = kmeans.predict(X)
# n_cluster center points
centers = kmeans.cluster_centers_-
Notifications
You must be signed in to change notification settings - Fork 0
yonas-g/miniML
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
mini machine learning library