使用类的 __call__ 方法作为 Numpy curve_fit 的输入

Question

我想使用一个类的__call__ 方法作为 Numpy curve_fit 函数的输入，因为我的函数和数据准备过程相当复杂（将分析模型数据拟合到某些测量值）。通过定义一个函数它工作得很好，但我不能让它与类一起工作。

要重现我的问题，您可以运行：

import numpy as np
from scipy.optimize import curve_fit

#WORKS:
#def goal(x,a1,a2,a3,a4,a5):
#    y=a1*x**4*np.sin(x)+a2*x**3+a3*x**2+a4*x+a5
#    return y

# DOES NOT WORK:
class func():
    def __call__(self,x,a1,a2,a3,a4,a5):
        y=a1*x**4*np.sin(x)+a2*x**3+a3*x**2+a4*x+a5
        return y    

goal=func()

#data prepraration ***********
xdata=np.linspace(0,50,100)
ydata=goal(xdata,-2.1,-3.5,6.6,-1,2)
# ****************************

popt, pcov = curve_fit(goal, xdata, ydata)
print 'optimial parameters',popt
print 'The estimated covariance of optimial parameters',pcov

我得到的错误是：

Traceback (most recent call last):
  File "D:\...some path...\test_minimizacija.py", line 35, in <module>
    popt, pcov = curve_fit(goal, xdata, ydata)
  File "C:\Python26\lib\site-packages\scipy\optimize\minpack.py", line 412, in curve_fit
    args, varargs, varkw, defaults = inspect.getargspec(f)
  File "C:\Python26\lib\inspect.py", line 803, in getargspec
    raise TypeError('arg is not a Python function')
TypeError: arg is not a Python function

我怎样才能做到这一点？

Answer 1

简单（虽然不漂亮），只需将其更改为：

popt, pcov = curve_fit(goal.__call__, xdata, ydata)

有趣的是，numpy 会强制您将函数对象传递给 curve_fit，而不是任意可调用的 ...

快速检查curve_fit 的来源，似乎另一种解决方法可能是：

popt,pcov = curve_fit(goal, xdata, ydata, p0=[1]*5)

这里，p0 是拟合参数的初始猜测。问题似乎是scipy 检查函数的参数，以便如果您实际上没有提供参数作为初始猜测，它知道要使用多少个参数。在这里，由于我们有 5 个参数，我最初的猜测是长度为 5 的所有参数的列表。（scipy 默认使用参数，如果你不提供猜测）。