Я использую lmfit
для подгонки четырехпараметрической логистической кривой к моим данным, с моим текущим кодом следующим образом:
import matplotlib, numpy as np, matplotlib.pyplot as plt
from lmfit import Model
def pl(x, a, b, c, d):
return (a - d) / (1+ (x / c) ** b) + d
x = [-4.522878745,
-5.22184875,
-5.920818754,
-6.619788758,
-7.318758763,
-8.017728767,
-8.716698771,
-9.415668776,
]
y = [12,
6,
37,
10297,
22635,
24279,
25666,
24365]
mod = Model(pl)
params = mod.make_params(a = 25000, b = 2, c = 1, d = 0)
result = mod.fit(y, params, x=x)
print(result.fit_report())
plt.plot(x, y, 'bo', markersize = 1.5)
plt.plot(x, result.best_fit, color = 'red', linewidth = 0.5)
plt.show()
Однако, это дает следующую ошибку:
Traceback (most recent call last):
File "C:\Users\George\OneDrive\Subjects\EE\Regression.py", line 29, in <module>
result = mod.fit(y, params, x=x)
File "C:\Users\George\AppData\Local\Programs\Python\Python37\lib\site-packages\lmfit\model.py", line 1007, in fit
output.fit(data=data, weights=weights)
File "C:\Users\George\AppData\Local\Programs\Python\Python37\lib\site-packages\lmfit\model.py", line 1355, in fit
_ret = self.minimize(method=self.method)
File "C:\Users\George\AppData\Local\Programs\Python\Python37\lib\site-packages\lmfit\minimizer.py", line 1949, in minimize
return function(**kwargs)
File "C:\Users\George\AppData\Local\Programs\Python\Python37\lib\site-packages\lmfit\minimizer.py", line 1492, in leastsq
lsout = scipy_leastsq(self.__residual, variables, **lskws)
File "C:\Users\George\AppData\Local\Programs\Python\Python37\lib\site-packages\scipy\optimize\minpack.py", line 396, in leastsq
gtol, maxfev, epsfcn, factor, diag)
File "C:\Users\George\AppData\Local\Programs\Python\Python37\lib\site-packages\lmfit\minimizer.py", line 538, in __residual
nan_policy=self.nan_policy)
File "C:\Users\George\AppData\Local\Programs\Python\Python37\lib\site-packages\lmfit\minimizer.py", line 2166, in _nan_policy
raise ValueError("The input contains nan values")
ValueError: The input contains nan values
Однако я заметил одну вещь: если я изменяю параметр c
на 0
, ошибка исчезает, но заменяется следующей ошибкой:
Warning (from warnings module):
File "C:\Users\George\OneDrive\Subjects\EE\Regression.py", line 5
return (a - d) / (1+ (x / c) ** b) + d
RuntimeWarning: divide by zero encountered in true_divide
Кроме того, 'Кривая '- это просто прямая линия:
Как мне решить эту проблему?