从 GMM 模型中绘制高斯子群体答案

【问题标题】：Plot gaussian sub-populations from GMM model从 GMM 模型中绘制高斯子群体
【发布时间】：2020-02-22 22:36:08
【问题描述】：

我使用 scikit learn on python 制作了一个 GMM 模型，如下所述：

x = df1['DNA_2']
y = df1['DNA_1']
X = np.column_stack((x, y)) # create a 2D array from the two lists
mod2 = GaussianMixture(n_components=5, covariance_type='tied', random_state=2) # build the gmm
mod2.fit(X)

然后我使用这个模型进行预测，然后绘制：

df1['pred2'] = labels
fig, ax = plt.subplots(1,1)
ax.scatter(x, y, c=df1['pred2'].apply(lambda x: colors[x]), s = 0.5, alpha=0.2)
H,X,Y = density_estimation(x,y)
ax.contour(H, X, Y, 8, linewidths=0.5, cmap='viridis')

得到：

我想知道如何绘制 5 个总体的高斯曲线。我知道我可以使用 mod1.means_ 获得平均值，使用 mod1.covariances_ 获得方差（均为 2D），但我如何绘制它以获得每个群体的曲线？

希望得到类似的东西：

【问题讨论】：

标签： python gaussian gmm

【解决方案1】：

如果是像图片这样的2D GMM，唯一的办法就是绘制2D密度图如：https://pythonmachinelearning.pro/clustering-with-gaussian-mixture-models/ 所附折线图适用于具有三个分量的一维 GMM。要绘制此图，您需要绘制每个集群/组的概率密度分量。

【讨论】：