我正在尝试使用symfit包执行全局拟合,遵循symfit 文档。
import numpy as np
import symfit as sf
import matplotlib.pyplot as plt
%matplotlib inline # for ipynb
# Generate example data
t = np.arange(0.0, 600.1, 30)
k = 0.005
C1_0, C2_0 = 1.0, 2.0
C1 = C1_0 * np.exp(-k*t)
C2 = C2_0 * np.exp(-k*t)
# Construct model
x_1, x_2, y_1, y_2 = sf.variables('x_1, x_2, y_1, y_2')
kg = sf.Parameter(value=0.01, min=0.0, max=0.1)
a_1, a_2 = sf.parameters('a_1, a_2')
globalmodel = sf.Model({
y_1: a_1 * np.e**(- kg * x_1),
y_2: a_2 * np.e**(- kg * x_2),
})
# Do fit
globalfit = sf.Fit(globalmodel, x_1=t, x_2=t, y_1=C1, y_2=C2)
globalfit_result = globalfit.execute()
print(globalfit_result)
### EDITED START
while globalfit_result.r_squared < 0.99:
kg = sf.Parameter(value=globalfit_result.params['kg'])
a_1 = sf.Parameter(value=globalfit_result.params['a_1'])
a_2 = sf.Parameter(value=globalfit_result.params['a_2'])
globalmodel = sf.Model({
y_1: a_1 * np.e**(- kg * x_1),
y_2: a_2 * np.e**(- kg * x_2),
})
globalfit = sf.Fit(globalmodel, x_1=t, x_2=t, y_1=C1, y_2=C2)
globalfit_result = globalfit.execute()
### EDITED END
y_r = globalmodel(x_1=t, x_2=t, **globalfit_result.params)
# Plot fit
plt.plot(t,C1,'ro')
plt.plot(t,C2,'b+')
plt.plot(t,y_r[0],'r-')
plt.plot(t,y_r[1],'b-')
plt.show()
在这个例子中,我希望“globalmodel”中的“kg”参数被优化为 0.005。但是,“kg”的值约为 9.6e-3,与初始值(10.0e-3)太接近。我想我做了一些愚蠢的事情,但我无法弄清楚。
欢迎任何意见和建议!
已编辑
我添加了(一个非常丑陋的)while 循环以获得最佳匹配。我不确定为什么会这样,但它似乎有效。