【发布时间】:2020-03-16 20:54:03
【问题描述】:
我需要在脚本中使用多元正态分布。我注意到我的版本给出了与 scipy 方法不同的答案。我真的不知道为什么......
这是我的功能:
def gauss(x, mu, sigma):
assert np.linalg.det(sigma)!=0, "determinant of sigma is 0"
y = np.exp((-1/2)*(x-mu).T.dot(np.linalg.inv(sigma)).dot(x-mu))/np.sqrt(
np.power(2*np.pi, len(x))*np.linalg.det(sigma)
)
return y
以下是结果对比:
from scipy.stats import multivariate_normal
import numpy as np
x = np.array([-0.54849176, 6.39530657])
mu = np.array([15,20])
sigma = np.array([
[2,3],
[4,10]
])
print(gauss(x, mu, sigma))
# output is 1.8781656851138248e-37
print(multivariate_normal.pdf(x, mu, sigma))
# output is 2.698549423643947e-61
有人注意到了吗?我的功能错了吗?任何帮助将不胜感激!
【问题讨论】:
-
当
x和mu是一维数组时,(x-mu).T不会像您认为的那样做。如果您想安全,请使用reshape(1, -1)。 -
另外,不是断言数组,而是转换为数组而不需要额外的副本。这实际上是相当标准的。
-
@MadPhysicist 感谢 cmets。我试过
reshape(1, -1),但似乎没有帮助 -
我会玩它,当我使用桌面时会通知你
-
密度非常低,因此先验地,人们可能会猜测您所看到的只是数字问题:以不同的方式设置一对括号就足以导致差异。但是,这里发生了其他事情,因为您也有
x = np.array([15.054849176, 20.39530657])的差异@
标签: python python-3.x numpy scipy probability