zebra/pyzebra
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use weights in fit

Browse Source
This commit is contained in:
usov_i 2021-03-01 15:40:31 +01:00
parent 3e174f22e5
commit 69767da850
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pyzebra/ccl_process.py
View File

@ -125,4 +125,5 @@ def fit_scan(scan, model_dict):
else: else:
model += _model 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)