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()