refactor: imports of scipy.stats in fits simplified.

This commit is contained in:
Fabian Joswig 2022-06-15 14:14:01 +01:00
parent d600793d0e
commit 0150ef4c0b

View file

@ -8,7 +8,6 @@ import scipy.stats
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from matplotlib import gridspec from matplotlib import gridspec
from scipy.odr import ODR, Model, RealData from scipy.odr import ODR, Model, RealData
from scipy.stats import chi2
import iminuit import iminuit
from autograd import jacobian from autograd import jacobian
from autograd import elementwise_grad as egrad from autograd import elementwise_grad as egrad
@ -303,7 +302,7 @@ def total_least_squares(x, y, func, silent=False, **kwargs):
output.odr_chisquare = odr_chisquare(np.concatenate((out.beta, out.xplus.ravel()))) output.odr_chisquare = odr_chisquare(np.concatenate((out.beta, out.xplus.ravel())))
output.dof = x.shape[-1] - n_parms output.dof = x.shape[-1] - n_parms
output.p_value = 1 - chi2.cdf(output.odr_chisquare, output.dof) output.p_value = 1 - scipy.stats.chi2.cdf(output.odr_chisquare, output.dof)
return output return output
@ -594,7 +593,7 @@ def _standard_fit(x, y, func, silent=False, **kwargs):
output.chisquare = chisquare output.chisquare = chisquare
output.dof = x.shape[-1] - n_parms output.dof = x.shape[-1] - n_parms
output.p_value = 1 - chi2.cdf(output.chisquare, output.dof) output.p_value = 1 - scipy.stats.chi2.cdf(output.chisquare, output.dof)
if kwargs.get('resplot') is True: if kwargs.get('resplot') is True:
residual_plot(x, y, func, result) residual_plot(x, y, func, result)