Скачать или смотреть generate_request.ipynb

np.array(features_extrabkg[['pxj2', 'pyj2', 'pzj2']]))
Ejj_extrabkg = (np.sqrt(np.sum(np.array(
features_extrabkg[['pxj1', 'pyj1', 'pzj1', 'mj1']])**2, axis=1))
np.sqrt(np.sum(np.array(
features_extrabkg[['pxj2', 'pyj2', 'pzj2', 'mj2']])**2, axis=1)))
mjj_extrabkg = np.sqrt(Ejj_extrabkg**2-np.sum(pjj_extrabkg**2, axis=1))

if args.add_deltaR:
compute DeltaR JJ
deltaR_jj_bg = deltaR(features_bg)
deltaR_jj_sig = deltaR(features_sig)
deltaR_jj_extrabkg = deltaR(features_extrabkg)

dataset_bg = np.dstack((mjj_bg/1000, mjmin_bg/1000,
(mjmax_bg-mjmin_bg)/1000,
tau21min_bg, tau21max_bg,
deltaR_jj_bg,
np.zeros(len(mjj_bg))))[0]
dataset_sig = np.dstack((mjj_sig/1000, mjmin_sig/1000,
(mjmax_sig-mjmin_sig)/1000,
tau21min_sig, tau21max_sig,
deltaR_jj_sig,
np.ones(len(mjj_sig))))[0]
dataset_extrabkg = np.dstack((mjj_extrabkg/1000, mjmin_extrabkg/1000,
(mjmax_extrabkg-mjmin_extrabkg)/1000,
tau21min_extrabkg, tau21max_extrabkg,
deltaR_jj_extrabkg,
np.zeros(len(mjj_extrabkg))))[0]
else:
dataset_bg = np.dstack((mjj_bg/1000, mjmin_bg/1000,
(mjmax_bg-mjmin_bg)/1000,
tau21min_bg, tau21max_bg,
np.zeros(len(mjj_bg))))[0]
dataset_sig = np.dstack((mjj_sig/1000, mjmin_sig/1000,
(mjmax_sig-mjmin_sig)/1000,
tau21min_sig, tau21max_sig,
np.ones(len(mjj_sig))))[0]
dataset_extrabkg = np.dstack((mjj_extrabkg/1000, mjmin_extrabkg/1000,