scipy负对数似然极小化不收敛的指数衰减拟合

2024-07-05 09:32:59 发布

您现在位置:Python中文网/ 问答频道 /正文

enter image description here我试图用负对数似然极小化来拟合指数衰减函数,但即使有好的初始参数x0作为最小值,我似乎也无法使其收敛。为什么?我写错了吗?在

编辑以包括传统的二进制似然,即“曲线”拟合

import numpy as np
from scipy.optimize import minimize, curve_fit
import matplotlib.pyplot as plt

np.random.seed(1)

def exp_fit(x, N, L):
    return N * np.exp(- L * x)

def negloglik(args, func, data):
    """Negative log likelihood"""
    return - np.sum(np.log(func(data, *args)))

def histpoints_w_err(X, bins):
    counts, bin_edges = np.histogram(X, bins = bins, normed = False)
    bin_centers       = (bin_edges[1:] + bin_edges[:-1]) / 2
    bin_err           = np.sqrt(counts)

    # Generate fitting points
    x  = bin_centers[counts > 0]  # filter along counts, to remove any value in the same position as an empty bin
    y  = counts[counts > 0]
    sy = bin_err[counts > 0]

    return x, y, sy

data = np.random.exponential(0.5, 1000)
bins = np.arange(0, 3, 0.1)

x, y, sy = histpoints_w_err(data, bins)
popt, pcov = curve_fit(exp_fit, x, y, sigma = sy)

xpts = np.linspace(0, 3, 100)


# All variables must be positive
bnds = ((0, None),
        (0, None))

result = minimize(negloglik,
                  args = (exp_fit, data),
                  x0 = (popt[0], popt[1]), # Give it the parameters of the fit that worked
                  method = "SLSQP",
                  bounds = bnds)

jac = result.get("jac")

plt.hist(data, bins = bins)
plt.plot(xpts, exp_fit(xpts, *popt), label = "Binned fit: {:.2f}exp(-{:.2f}x)".format(*popt))
plt.plot(xpts, exp_fit(xpts, *jac), label = "Unbinned fit: {:.2f}exp(-{:.2f}x)".format(*jac))
plt.text(s = result, x = 0.8, y = popt[0]*0.2)
plt.legend()
plt.show()

Tags: importdatabindefasnppltfit
2条回答

我删除了这样的方法:

# minimize the negative log-Likelihood
result = minimize(negloglik,  args = (exp_fit, data), x0 = (15, 0.5))

我得到的是:

^{pr2}$

以前的成功是错误的。也许方法应该改成其他方法?如果未指定,则默认值为BFGS、L-BFGS-B、SLSQP,这取决于问题是否具有约束或边界。在

所以,用iminuit和probfit而不是scipy重新编辑整个过程。语法有点奇怪(尤其是iminuit如何更改输入参数名以匹配合适的参数),但是一旦开始使用,它就很容易使用了。遗憾的是,关于这方面的社区文档很少。在

在这里,我做了无约束似然拟合:

import iminuit, probfit
import matplotlib.pyplot as plt
import numpy as np
import scipy.stats

np.random.seed(1)
data = np.random.exponential(0.5, 500)

def unbinned_exp_LLH(data, loc_init, scale_init, limit_loc, limit_scale):
    # Define function to fit
    def exp_func(x, loc, scale):
        return scipy.stats.expon.pdf(x, loc, scale)

    # Define initial parameters
    init_params = dict(loc = loc_init, scale = scale_init)

    # Create an unbinned likelihood object with function and data.
    unbin = probfit.UnbinnedLH(exp_func, data)

    # Minimizes the unbinned likelihood for the given function
    m = iminuit.Minuit(unbin,
                       **init_params,
                       limit_scale = limit_loc,
                       limit_loc = limit_scale,
                       pedantic=False,
                       print_level=0)
    m.migrad()
    params = m.values.values() # Get out fit values
    errs   = m.errors.values()
    return params, errs


params, errs = unbinned_exp_LLH(data, loc_init = 0, scale_init = 0.5, limit_loc = (-1, 1), limit_scale = (-1, 1))

loc, scale = params

# Plot
x_pts = np.linspace(0, 3, 100)
plt.plot(x_pts, scipy.stats.expon.pdf(x_pts, *params), label = "exp(-{1:.2f}x)".format(*params), color = "black")
plt.hist(data, color = "lightgrey", bins = 20, label = "generated data", normed = True)
plt.xlim(0, 3)
plt.legend()
plt.show()

相关问题 更多 >