From 345837c23a433818cf59c27ac31295a9bb20acf1 Mon Sep 17 00:00:00 2001 From: MatthieuR18 Date: Wed, 7 Aug 2024 10:25:32 +0200 Subject: [PATCH] MR: Small correction in lsq function --- ForMoSA/nested_sampling/nested_modif_spec.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/ForMoSA/nested_sampling/nested_modif_spec.py b/ForMoSA/nested_sampling/nested_modif_spec.py index a5157a2..c321834 100755 --- a/ForMoSA/nested_sampling/nested_modif_spec.py +++ b/ForMoSA/nested_sampling/nested_modif_spec.py @@ -59,17 +59,17 @@ def lsq_fct(flx_obs_spectro, err_obs_spectro, star_flx_obs, transm_obs, flx_mod_ if len(system_obs) > 0: A_matrix = np.zeros([np.size(flx_obs_spectro), 1 + len(star_flx_obs[0][0]) + len(system_obs[0][0])]) for j in range(len(system_obs[0][0])): - A_matrix[:,1+len(star_flx_obs[0][0])+j] = system_obs[0][:,j] * err_obs_spectro + A_matrix[:,1+len(star_flx_obs[0][0])+j] = system_obs[0][:,j] / err_obs_spectro else: A_matrix = np.zeros([np.size(flx_obs_spectro), 1 + len(star_flx_obs[0][0])]) for j in range(len(star_flx_obs[0][0])): - A_matrix[:,1+j] = star_flx_obs[0][:,j] * err_obs_spectro + A_matrix[:,1+j] = star_flx_obs[0][:,j] / err_obs_spectro - A_matrix[:,0] = flx_mod_spectro * err_obs_spectro + A_matrix[:,0] = flx_mod_spectro / err_obs_spectro # Least square - res = optimize.lsq_linear(A_matrix, flx_obs_spectro * err_obs_spectro, bounds=([0]*len(A_matrix[0]), [1]*len(A_matrix[0]))) + res = optimize.lsq_linear(A_matrix, flx_obs_spectro / err_obs_spectro, bounds=([0]*len(A_matrix[0]), [1]*len(A_matrix[0]))) cp = res.x[0] cs = np.array([])