Question

我想对某些数据使用函数，但遇到了问题。我尝试从scipy使用lmfit或curve_fit。下面我描述这个问题。

这是我的数据：

dataOT = pd.read_csv("KIC3239945e.csv", sep=';') 
t=dataOT['time']
y=dataOT['flux']

此外，这是要拟合到数据的模型函数：

def model(t, Rp, Rs, a, orb_inclination, Rin, Rout, tau):  
    gps=Rp/Rs
    gis=Rin/Rs
    gos=Rout/Rs
    Agps=A(t, gps, Rp, Rs, a, orb_inclination, Rin, Rout)
    Agos=A(t, gos, Rp, Rs, a, orb_inclination, Rin, Rout)
    Agis=A(t, gis, Rp, Rs, a, orb_inclination, Rin, Rout)
    return (np.pi*(1-u1/3-u2/6)-Agps-(1-np.exp(-tau))*(Agos-Agis))/(np.pi*(1-u1/3-u2/6))

其中u1，u2是已知数，要拟合的参数为：Rp，Rs，a，orb_inclination，Rin，Rout，tau，它们包含在Agps，Agos，Agis中。这是函数A的定义：

def A(t, gamma, Rp, Rs, a, orb_inclination, Rin, Rout):  
Xp,Zp= planetary_position(t, a, orb_inclination)
return np.where(rho(Xp,Zp,Rs)<1-gamma,
                np.pi*gamma**2*(1-u1-u2*(2-rho(Xp,Zp,Rs)**2-gamma**2/2)+(u1+2*u2)*W11(Xp,Zp,gamma,Rs) ) , 
                np.where(np.logical_and(  (1-gamma<=rho(Xp,Zp,Rs)),  (rho(Xp,Zp,Rs)<=1+gamma)  ), 
                (1-u1-3*u2/2)*(gamma**2*np.arccos(zeta(Xp,Zp,gamma,Rs)/gamma)+np.arcsin(zo(Xp,Zp,gamma,Rs))-rho(Xp,Zp,Rs)*zo(Xp,Zp,gamma,Rs))+(u2/2)*rho(Xp,Zp,Rs)*((rho(Xp,Zp,Rs)+2*zeta(Xp,Zp,gamma,Rs))*gamma**2*np.arccos(zeta(Xp,Zp,gamma,Rs)/gamma)-zo(Xp,Zp,gamma,Rs)*(rho(Xp,Zp,Rs)*zeta(Xp,Zp,gamma,Rs)+2*gamma**2))  +(u1+2*u2)*W3(Xp,Zp,gamma,Rs)    , 0))

第一次尝试：curve_fit

from scipy.optimize import curve_fit
p0=[4.5*10**9, 4.3*10**10, 1.4*10**13, 1.2, 4.5*10**9, 13.5*10**9, 1]
popt, pcov = curve_fit(model, t, y, p0, bounds=((0, 0, 0, 0, 0, 0 ,0 ),(np.inf, np.inf, np.inf,np.inf, np.inf, np.inf ,np.inf )), maxfev=6000)
print(popt)

第二次尝试：lmfit

   from lmfit import Parameters, Minimizer, report_fit, Model
gmodel=Model(model)

def residual(p,t, y):
    Rp=p['Rp']
    Rs=p['Rs']
    a=p['a']
    orb_inclination=p['orb_inclination']
    Rin=p['Rin']
    Rout=p['Rout']
    tau=p['tau']
    tmp = model(t, Rp, Rs, a, orb_inclination, Rin, Rout, tau)-y
    return tmp

p = Parameters()

p.add('Rp' ,  value=0.000394786,     min= 0,max=1)
p.add('Rs' ,  value=0.003221125,    min= 0,max=1)
p.add('a',   value=1.86,            min= 0,max= 1)
p.add('orb_inclination',  value=1,   min= 0,max=4)
p.add('Rin',  value=0.000394786,    min= 0,max=1)
p.add('Rout',  value=0.000394786,    min= 0,max=1)
p.add('tau',  value=0,                 min= 0,max=2)

mini = Minimizer(residual,params=p,fcn_args=(t,y))

out = mini.minimize(method='leastsq')

print(report_fit(out))

所有情况均以最合适的参数返回初始猜测。我应该怎么做才能使其正常工作？

注意：假设参数已知，则模型具有预期的行为（Figure 1），因此我认为模型定义明确，问题与此无关

任何帮助将不胜感激。预先谢谢你！

Answer 1

我有两个想法，我想第一个可能是您的罪魁祸首。

在我看来，使用nan_policy ='omit'可以在非常特殊的情况下工作。如果在尝试拟合时收到错误消息“ nan_policy ='omit'可能不起作用” ，那可能就行不通了。您可以对NaN值进行简单检查，以确认该函数为您的间隔输出NaN值。
变量的界限是巨大。尝试提高间隔的最小值。

Answer 2

没有真实的数据和完整的示例，很难猜测可能出了什么问题。因此，这将包括一些有关如何解决该问题的建议。

首先：由于您正在进行曲线拟合，并且具有模型功能，因此建议您使用lmfit.Model从第二版开始。但是，我建议显式地制作一组参数，例如：

from lmfit import Model
def A(t, gamma, Rp, Rs, a, orb_inclination, Rin, Rout):  
    Alpha = np.zeros(len(t))
    Xp, Zp = planetary_position(t, a, orb_inclination)
    values_rho = rho(Xp, Zp, Rs)
    v_W11 = W11(Xp,Zp, gamma, Rs)
    v_W11 = pd.Series(v_W11)
    v_zeta = zeta(Xp,Zp, gamma,Rs)
    v_zo = zo(Xp, Zp, gamma,Rs)
    v_W3 = W3(Xp,Zp, gamma,Rs)
    for i in range(len(values_rho)):
        Alpha[i]=np.where(values_rho[i]<1-gamma, np.pi*gamma**2*(1-u1-u2*(2-values_rho[i]**2-gamma**2/2)+(u1+2*u2)*v_W11[i] ) ,  np.where(((1-gamma<=values_rho[i]) and (values_rho[i]<=1+gamma)),  (1-u1-3*u2/2)*(gamma**2*np.arccos(v_zeta[i]/gamma)+np.arcsin(v_zo[i])-values_rho[i]*v_zo[i])+(u2/2)*values_rho[i]*((values_rho[i]+2*v_zeta[i])*gamma**2*np.arccos(v_zeta[i]/gamma) -v_zo[i]*(values_rho[i]*v_zeta[i]+2*gamma**2))+(u1+2*u2)*v_W3[i]    , 0)) 
    return Alpha

model = Model(model)
params = model.make_params(Rp=4.5*10**9, Rs=4.3*10**10, 
                           a=1.4*10**13, orb_inclination=1.2, 
                           Rin=4.5*10**9, Rout=13.5*10**9, tau=1)
result = gmodel.fit(y, params, t=t)
print(result.fit_report())

就其本身而言，这不能解决问题，但清晰度很重要。但是，您可以自己调用该模型函数或执行

  gmodel.eval(params, t=t)

并查看其对于任何一组参数值的实际计算结果。

第二：您应该对在拟合问题中涉及多个数量级的变量保持谨慎。让变量更像1阶（或者恰好在1.e-6和1.e6阶之间），然后视情况乘以1e9或1e12的因子-或仅以值更接近1的单位工作。拟合的全都是双精度浮点，并且参数的相对值很重要。

第三：是的，您的模型函数。可读性计数。编写一个难以理解的函数对任何人都没有帮助。包括你。我保证您不知道这是做什么的。例如，您也许可以避免循环，而仅使用numpy的ufunc-ness，但这是不可能的。而且要明确一点，因为您是用这种方式编写的，所以无法确定。像u1和u2到底应该是什么？确实，此功能不存在，您编写了一个完整的烂摊子，然后出了点问题。

因此：编写模型函数，就像您希望明年阅读它一样，然后使用合理的输入值测试它的计算结果。在这种情况下，试穿也应该起作用。

Answer 3

我通过减少参数数量解决了这个问题。另外，另一个问题是其中一个参数根本没有影响拟合。

lmfit和scipy curve_fit将初始猜测作为最适合的参数返回

3 个答案: