Skip to content

Commit

Permalink
Merge pull request #22 from LSSTDESC/tqz/debug_variance
Browse files Browse the repository at this point in the history
fix variance/sigma bug
  • Loading branch information
sschmidt23 authored Aug 29, 2024
2 parents 5860497 + 743abaa commit 0b62771
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/rail/estimation/algos/gpz.py
Original file line number Diff line number Diff line change
Expand Up @@ -161,7 +161,7 @@ def _process_chunk(self, start, end, data, first):
self.config.log_errors, self.config.replace_error_vals)

mu, totalV, modelV, noiseV, _ = self.model.predict(test_array)
ens = qp.Ensemble(qp.stats.norm, data=dict(loc=mu, scale=totalV))
ens = qp.Ensemble(qp.stats.norm, data=dict(loc=mu, scale=np.sqrt(totalV)))
zgrid = np.linspace(self.config.zmin, self.config.zmax, self.config.nzbins)
zmode = ens.mode(grid=zgrid)
ens.set_ancil(dict(zmode=zmode))
Expand Down

0 comments on commit 0b62771

Please sign in to comment.