由于需要利用L-M算法进行光谱拟合,所以下载了mpfit.py文件使用,但是不是很清楚用法;我尝试拟合最简单的线性函数,但是结果完全不正确;初始参数未经任何改变就被输出;代码如下,不太清楚错误在哪里;希望各位帮忙。
import mpfit as mt
import numpy as np
import random
import types
x=np.arange(0,10,0.1)
y=[]
for i in range(np.size(x)):
y.append(5*x[i]+3+random.randrange(0,1))
def myfunct(p, fjac=None,x=None, y=None, err=None):
model=p[0]*x+p[1]
status=0
return([status,(y-model)/(err)])
p0=[10,9]
fa = {'x':x, 'y':y,'err':0.1}
m=mt.mpfit(myfunct,p0,functkw=fa)
print ('status = ', m.status)
if (m.status <= 0):
print ('error message = ', m.errmsg)
print ('parameters = ', m.params)
输出:
status = 4
parameters = [10. 9.]