使用類的__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默認使用，如果你還沒有提供猜測）。