Skip to content

Instantly share code, notes, and snippets.

Show Gist options
  • Save vitasiku/c77bda6421b8adeb812b7e3b7522bd0e to your computer and use it in GitHub Desktop.
Save vitasiku/c77bda6421b8adeb812b7e3b7522bd0e to your computer and use it in GitHub Desktop.
class Clustering:
def k_means_clustering(self, n, s=1.0):
"""
This method performs the K-means clustering algorithm on the data for n iterations. This involves updating the
centroids using the mean-shift heuristic n-times and reassigning the patterns to their closest centroids.
:param n: number of iterations to complete
:param s: the scaling factor to use when updating the centroids
pick on which has a better solution (according to some measure of cluster quality)
"""
for i in range(n):
self.assign_patterns()
self.update_centroids(s)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment