Skip to content

Instantly share code, notes, and snippets.

@StuartGordonReid
Created June 15, 2015 14:38
Show Gist options
  • Save StuartGordonReid/06016f399f44913c6834 to your computer and use it in GitHub Desktop.
Save StuartGordonReid/06016f399f44913c6834 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