diff options
Diffstat (limited to 'likelihood.py')
| -rw-r--r-- | likelihood.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/likelihood.py b/likelihood.py index 99d0543..2487442 100644 --- a/likelihood.py +++ b/likelihood.py @@ -7,7 +7,7 @@ from chroma.tools import profile_if_possible class Likelihood(object): "Class to evaluate likelihoods for detector events." - def __init__(self, sim, event=None, tbins=100, trange=(-0.5, 499.5), + def __init__(self, sim, event=None, tbins=100, trange=(-0.5e-9, 999.5e-9), qbins=10, qrange=(-0.5, 9.5)): """ Args: @@ -47,10 +47,10 @@ class Likelihood(object): will be propagated `nreps` times. """ - hitcount, pdfcount = sim.create_pdf(nevals, vertex_generator, - self.tbins, self.trange, - self.qbins, self.qrange, - nreps=nreps) + hitcount, pdfcount = self.sim.create_pdf(nevals, vertex_generator, + self.tbins, self.trange, + self.qbins, self.qrange, + nreps=nreps) # Normalize probabilities and put a floor to keep the log finite hit_prob = hitcount.astype(np.float32) / (nreps * nevals) |