diff options
| author | tlatorre <tlatorre@uchicago.edu> | 2020-12-08 12:54:12 -0600 |
|---|---|---|
| committer | tlatorre <tlatorre@uchicago.edu> | 2020-12-08 12:54:12 -0600 |
| commit | 32be8ba6ec566e40285a7f6b556219bea4d4d6c0 (patch) | |
| tree | f35b740c97048b486df179a2dde52a69019f53f4 | |
| parent | ff4904530a099a2adb50830f53d82fb9bd220542 (diff) | |
| download | sddm-32be8ba6ec566e40285a7f6b556219bea4d4d6c0.tar.gz sddm-32be8ba6ec566e40285a7f6b556219bea4d4d6c0.tar.bz2 sddm-32be8ba6ec566e40285a7f6b556219bea4d4d6c0.zip | |
use weighted MC when running dm-search --test
| -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]) |