python 适合自定义功能模型

Posted 2021-05-09

## using scipy
from scipy.optimize import curve_fit

def model(x,a,b):
  return x*b+a
init_vals = [1, 0]  # for [a,b]
best_vals, covar = curve_fit(model, x, y, p0=init_vals)
print(best_vals)

## using lmfit
from lmfit import Model
gmodel = Model(model)
gmodel.param_names
# ['a', 'b']
gmodel.independent_vars
# ['x']
# fit
result = gmodel.fit(y, x=x, a=1,b=b)
print(result.fit_report())
# plot
plt.plot(x, y, 'bo')
plt.plot(x, result.init_fit, 'k--')
plt.plot(x,