def plot_clustering_ellipses(self, ESS=None, dims=[0, 1], scale=1):
"""Make a scatterplot of the data with error ellipses
"""
if ESS is None:
ESS = self.ESS
plt.plot(self.data[:, dims[0]], self.data[:, dims[1]], 'o', ls='none')
for k in range(ESS.num_comp):
pos = ESS.smm_mean[k, :]
cov = scale * ESS.smm_covar[k, :, :]
plt.plot(pos[0], pos[1], 'r+')
plot_ellipse(pos, cov, edge='red')
plt.show()
def plot_clustering_ellipses(self, data, ess, dims=[0, 1], scale=1):
'''
Make a scatterplot of the data with error ellipses
'''
plt.plot(data[:, dims[0]],
data[:, dims[1]],
'o',
ls='none')
for k in range(ess.num_comp):
pos = ess.smm_mean[k, :]
cov = scale * ess.smm_covar[k, :, :]
plt.plot(pos[0], pos[1], 'r+')
plot_ellipse(pos, cov, edge='red')
plt.show()
