Question

使用curvefit函数时遇到问题。这里我有一个代码，有两个函数可以使用。第一个是双曲线函数。第二个是相同的但有一个参数= 1。我的问题是使用curvefit拟合第一个函数的结果工作正常，但第二个函数没有。我有一个商业程序，分别为两者生成正确的解决方案。因此，有可能找到第二个功能的解决方案（如上所述，第一个功能的特定情况）是否有人可以让我知道我做错了什么？谢谢！

以下是要运行的代码：

def hypRegress(ptp,pir):

    xData = np.arange(len(ptp))
    yData = pir

    xData = np.array(xData, dtype=float)
    yData = np.array(yData, dtype= float)

    def funcHyp(x, qi, exp, di):
        return qi*(1+exp*di*x)**(-1/exp)

    def errfuncHyp(p):
        return funcHyp(xData, p[0], p[1], p[2]) - yData

    #print(xData.min(), xData.max())
    #print(yData.min(), yData.max())

    trialX = np.linspace(xData[0], xData[-1], 1000)



    # Fit an hyperbolic
    popt, pcov = optimize.curve_fit(funcHyp, xData, yData)
    print 'popt'
    #print(popt)
    yHYP = funcHyp(trialX, *popt)

    #optimization

    # initial values
    p1, success = optimize.leastsq(errfuncHyp, popt,maxfev=10000)
    print p1

    aaaa = funcHyp(trialX, *p1)

    plt.figure()
    plt.plot(xData, yData, 'r+', label='Data', marker='o')
    plt.plot(trialX, yHYP, 'r-',ls='--', label="Hyp Fit")
    plt.plot(trialX, aaaa, 'y', label = 'Optimized')
    plt.legend()
    plt.show(block=False)
    return p1


def harRegress(ptp,pir):

    xData = np.arange(len(ptp))
    yData = pir

    xData = np.array(xData, dtype=float)
    yData = np.array(yData, dtype=float)

    def funcHar(x, qi, di):
        return qi*(1+di*x)**(-1)

    def errfuncHar(p):
        return funcHar(xData, p[0], p[1]) - yData

    #print(xData.min(), xData.max())
    #print(yData.min(), yData.max())

    trialX = np.linspace(xData[0], xData[-1], 1000)



    # Fit an harmonic
    popt, pcov = optimize.curve_fit(funcHar, xData, yData)
    print 'popt'
    print(popt)
    yHAR = funcHar(trialX, *popt)

    #optimization

    # initial values
    p1, success = optimize.leastsq(errfuncHar, popt,maxfev=1000)
    print p1

    aaaa = funcHar(trialX, *p1)

    plt.figure()
    plt.plot(xData, yData, 'r+', label='Data', marker='o')
    plt.plot(trialX, yHAR, 'r-',ls='--', label="Har Fit")
    plt.plot(trialX, aaaa, 'y', label = 'Optimized')
    plt.legend()
    plt.show(block=False)
    return p1


ptp = ([0,1,2,3,4,5,6,7,8,9,10,11,12,13,14])
pir = ([150,85,90,50,45,60,60,40,40,30,28,30,38,30,26])

hypRegress(ptp,pir)
harRegress(ptp,pir)

input('pause')

Answer 1

这是一个经典问题。 curve_fit算法从对要优化的参数的初始猜测开始，如果没有提供，则只是所有参数。

这意味着，当你打电话

popt, pcov = optimize.curve_fit(funcHar, xData, yData)

拟合例程的第一次尝试是假设

funcHar(xData, qi=1, di=1)

如果您没有指定任何其他选项，那么拟合将会很差，参数估计值的大变化就会证明（检查pcov的对角线并将其与返回的实际值进行比较popt）。

在许多情况下，情况通过提供智能猜测来解决。在您的HAR模型中，我认为x==0周围的值与qi的大小相同。因此，您可以提供p0 = (pir[0], 1)的初始猜测，这将导致满意的解决方案。你也可以用

来调用它

popt, pcov = optimize.curve_fit(funcHar, ptp, pir, p0=(0, 1))

导致相同的结果。所以问题只是算法找到了局部最小值。

另一种方法是提供不同的factor，“确定初始步骤限制的参数”：

popt, pcov = optimize.curve_fit(funcHar, ptp, pir, p0=(1, 1), factor=1)

在这种情况下，即使使用（默认）初始猜测p0=(1,1)，它也会得到相同的结果。

记住：拟合是一门艺术，而不是一门科学。通常，通过分析您想要适合的模型，您可以提供良好的初始猜测。

我不能说商业计划中使用的算法。如果它是开源的（不太可能），你可以看看他们做了什么。

用curvefit拟合双曲和调和函数

1 个答案: