我正在尝试在 python 中实现高斯混合模型的期望最大化算法。
给定高斯分布的平均mu和协方差sigma ,我有以下行来计算我的数据X的高斯概率p:
for i in range(len(X[0])):
p[i] = scipy.stats.multivariate_normal.pdf(X[:,i],mu,sigma)
我想知道我是否可以以某种方式摆脱 for 循环以获得类似的东西
p[:] = scipy.stats.multivariate_normal.pdf(X[:,:]??)
我对广播进行了一些研究,并正在考虑使用该numpy.einsum
功能,但无法弄清楚在这种情况下它将如何工作。