Skip to content

Instantly share code, notes, and snippets.

@DFoly
Last active March 5, 2019 16:01
Show Gist options
  • Save DFoly/eb40ed58a2e378e180865e06fbb74879 to your computer and use it in GitHub Desktop.
Save DFoly/eb40ed58a2e378e180865e06fbb74879 to your computer and use it in GitHub Desktop.
def _initialise_parameters(self, X):
"""Implement k-means to find starting
parameter values.
Parameters:
------------
X: numpy array of data points
Returns:
----------
tuple containing initial means and covariance
_initial_means: numpy array: (C*d)
_initial_cov: numpy array: (C,d*d)
"""
n_clusters = self.C
kmeans = KMeans(n_clusters= n_clusters, init="k-means++", max_iter=500, algorithm = 'auto')
fitted = kmeans.fit(X)
prediction = kmeans.predict(X)
self._initial_means, self._initial_cov, self._initial_pi = self.calculate_mean_covariance(X, prediction)
return (self._initial_means, self._initial_cov, self._initial_pi)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment