How to plot the pdf of a 1D Gaussian Mixture Model with matplotlib

穿精又带淫゛_ 提交于 2019-12-20 04:29:15

问题


I want to plot a Gaussian Mixture Model. The following code allows me to plot 2 separate Gaussians, but where they intersect, the line is very sharp and not smooth enough. Is there a way to plot the pdf of a 1D GMM?

def plot_data():
    mu = [-6, 5]
    var = [2, 3]
    sigma = [np.sqrt(var[0]), np.sqrt(var[1])]
    x = np.linspace(-10, 10, 100)
    curve_0 = mlab.normpdf(x, mu[0], sigma[0])
    curve_1 = mlab.normpdf(x, mu[1], sigma[1])
    import ipdb; ipdb.set_trace()
    plt.plot(x, curve_0, color='grey')
    plt.plot(x, curve_1, color='grey')
    plt.fill_between(x,curve_0 , color='grey')
    plt.fill_between(x,curve_1, color='grey')
    plt.show()
    plt.savefig('data_t0.jpg')

回答1:


You can literally draw samples from a Gaussian mixture model and plot the empirical density / histogram too:

import matplotlib.pyplot as plt
import numpy as np
import seaborn as sns
n = 10000 # number of sample to be drawn
mu = [-6, 5]
sigma = [2, 3]
samples = []
for i in range(n): # iteratively draw samples
    Z = np.random.choice([0,1]) # latent variable
    samples.append(np.random.normal(mu[Z], sigma[Z], 1))
sns.distplot(samples, hist=False)
plt.show()
sns.distplot(samples)
plt.show()




回答2:


You have to form a convex combination of the densities.

curve = p*curve_0 + (1-p)*curve_1

where p is the probability that a sample comes from the first Gaussian.



来源:https://stackoverflow.com/questions/42339786/how-to-plot-the-pdf-of-a-1d-gaussian-mixture-model-with-matplotlib

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!