11 lines
214 B
Python
11 lines
214 B
Python
from scipy.optimize import curve_fit
|
|
from numpy import tanh
|
|
|
|
|
|
def probe(x, I_0, T_e, A):
|
|
return I_0 * tanh(x / T_e) + A * x
|
|
|
|
|
|
def fit(x, y):
|
|
params = curve_fit(probe, x, y, [50, 10, 1])[0]
|
|
return params |