copipe Python | scipyのcurve_fitに初期値を設定する。

概要

curve_fitの引数にp0=[第一パラメータの初期値, 第２パラメータの初期値, ...]を書くことで、フィッティングパラメータの初期値を設定することができる。

コード

from scipy.optimize import curve_fit

def funcA(x, a, b):
    y = a*x+b
    return y

xData = [1, 2, 4, 7, 13]
yData = [2, 6, 13, 45, 32]
opt, cov = curve_fit(funcA, xData, yData, p0=[1,2])
print(opt)

結果

[2.87339056 4.08369099]