diff options
Diffstat (limited to 'utils/dc-closure-test')
| -rwxr-xr-x | utils/dc-closure-test | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/utils/dc-closure-test b/utils/dc-closure-test index 54d2629..45c0f6e 100755 --- a/utils/dc-closure-test +++ b/utils/dc-closure-test @@ -383,7 +383,7 @@ def fit(data, sacrifice, steps): #samples = metropolis_hastings(nll,xopt,stepsizes,100000) #print("nll(xopt) = %.2g" % nll(xopt)) - pos = np.empty((10, len(x0)),dtype=np.double) + pos = np.empty((100, len(x0)),dtype=np.double) for i in range(pos.shape[0]): pos[i] = xopt + np.random.randn(len(x0))*stepsizes pos[i,:6] = np.clip(pos[i,:6],EPSILON,1e9) @@ -395,7 +395,7 @@ def fit(data, sacrifice, steps): nwalkers, ndim = pos.shape - proposal = get_proposal_func(stepsizes*0.1,low,high) + proposal = get_proposal_func(stepsizes,low,high) sampler = emcee.EnsembleSampler(nwalkers, ndim, lambda x, grad, fill_value: -nll(x,grad,fill_value), moves=emcee.moves.MHMove(proposal),args=[None,np.inf]) with np.errstate(invalid='ignore'): sampler.run_mcmc(pos, steps) @@ -425,7 +425,8 @@ if __name__ == '__main__': import matplotlib.pyplot as plt ev = get_events(args.filenames,merge_fits=True) - ev_mc = get_events(args.mc, merge_fits=True) + ev = ev[ev.prompt] + ev = ev[ev.nhit_cal > 100] # figure out bins for high level variables ev = radius_cut(ev) @@ -441,7 +442,9 @@ if __name__ == '__main__': ev['muon'] = ((ev.dc & DC_MUON) != 0) & ~(ev.noise | ev.neck | ev.flasher | ev.breakdown) ev['signal'] = ~(ev.noise | ev.neck | ev.flasher | ev.breakdown | ev.muon) + ev_mc = get_events(args.mc, merge_fits=True) ev_mc = ev_mc[ev_mc.prompt] + ev_mc = ev_mc[ev_mc.nhit_cal > 100] # figure out bins for high level variables ev_mc = radius_cut(ev_mc) |