Use weights in fit

2021-03-01 15:40:31 +01:00 · 2021-03-01 15:40:31 +01:00 · 69767da850
commit 69767da850
parent 3e174f22e5
1 changed files with 2 additions and 1 deletions
--- a/pyzebra/ccl_process.py
+++ b/pyzebra/ccl_process.py
@ -125,4 +125,5 @@ def fit_scan(scan, model_dict):
        else:
            model += _model

-    scan["fit"] = model.fit(y_fit, x=x_fit)
+    weights = [1 / np.sqrt(val) if val != 0 else 1 for val in y_fit]
+    scan["fit"] = model.fit(y_fit, x=x_fit, weights=weights)