在 Python 中传递返回 NaN 的函数的输入

Question

我正在使用EMCEE。我在下面写了一部分，问题从那里出现。我遇到了这个错误 ValueError: lnprob returned NaN.

此错误中断了计算，我不知道如何克服。因此，我应该做一些事情来传递此错误，以便继续其余的计算。

我唯一想到的是lnprob函数中添加一行，例如：

 if not np.isfinite(lp):
        return -np.inf
 if  np.isnan(lp):
        return -np.inf

但这是不正确的

代码为：

def log_prior(H0, od0, c, b, Orc, M):
    if  0.4 < od0 < 0.9 and  50 < H0 < 90  and  0 < c < 3 and  0 < b < 1  and  -0.3 < M < 0.2 and  0 < Orc < 0.1:
        return 0.0
    return -np.inf
def lnlike(H0, od0, c, b, Orc, M):
    lg = -chi2(H0, od0, c, b, Orc, M)/2.
    return lg
def lnprob(H0, od0, c, b, Orc, M):
    lp = log_prior(H0, od0, c, b, Orc, M)
    if not np.isfinite(lp):
        return -np.inf
    return lp + lnlike(H0, od0, c, b, Orc, M)
def func(theta):
    H0, od0, c, b, Orc, M = theta
    return -2. * lnprob(H0, od0, c, b, Orc, M)

我感谢您的帮助。

Answer 1

我只想说python，如果lnprob是NaN，没关系，继续其他值

如果这个^对你来说仍然如此，你应该能够做到：

def func(theta):
    H0, od0, c, b, Orc, M = theta
    try:
       lnprobval = -2. * lnprob(H0, od0, c, b, Orc, M)
    except ValueError: # NaN value case
       lnprobval = -np.inf # just set to negative infinity 
    return lnprobval

该代码应替换def func(theta)中的 return 语句