diff options
Diffstat (limited to 'utils/dm-search')
| -rwxr-xr-x | utils/dm-search | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/utils/dm-search b/utils/dm-search index c48d970..5e3d7b5 100755 --- a/utils/dm-search +++ b/utils/dm-search @@ -728,6 +728,9 @@ if __name__ == '__main__': # Set the random seed so we get reproducible results here np.random.seed(0) + data_mc_with_weights = pd.merge(data_mc,weights[weights.universe == 0],how='left',on=['run','evn']) + data_atm_mc_with_weights = pd.merge(data_atm_mc,weights[weights.universe == 0],how='left',on=['run','evn']) + discoveries = 0 for i in range(args.test): @@ -746,8 +749,8 @@ if __name__ == '__main__': n_muon_atm = np.random.poisson(N_muon_atm) # Sample data from Monte Carlo - data = pd.concat((data_mc.sample(n=n,replace=True), muon.sample(n=n_muon,replace=True))) - data_atm = pd.concat((data_atm_mc.sample(n=n_atm,replace=True), muon_atm.sample(n=n_muon_atm,replace=True))) + data = pd.concat((data_mc_with_weights.sample(n=n,replace=True,weights='weight'), muon.sample(n=n_muon,replace=True))) + data_atm = pd.concat((data_atm_mc_with_weights.sample(n=n_atm,replace=True,weights='weight'), muon_atm.sample(n=n_muon_atm,replace=True))) # Smear the energies by the additional energy resolution data.loc[data.id1 == 20,'energy1'] *= (1+xtrue[1]+np.random.randn(np.count_nonzero(data.id1 == 20))*xtrue[2]) |