ONNXRuntime/test/plotDDX.py

0001 import ROOT
0002 from DataFormats.FWLite import Handle, Events
0003 import numpy as np
0004 import pickle
0005
0006 events_c = Events('output_test_DDX.root')
0007
0008 handleJ  = Handle("std::vector<pat::Jet>")
0009 labelJ = ("selectedUpdatedPatJets","","PATtest")
0010
0011 h_probQ_ddb = ROOT.TH1F('h_probQ_ddb', ';prob Q;', 40, 0., 1.)
0012 h_probH_ddb = ROOT.TH1F('h_probH_ddb', ';prob H;', 40, 0., 1.)
0013
0014 h_probQ_ddc = ROOT.TH1F('h_probQ_ddc', ';prob Q;', 40, 0., 1.)
0015 h_probH_ddc = ROOT.TH1F('h_probH_ddc', ';prob H;', 40, 0., 1.)
0016
0017 info = {}
0018 info['pt'] = []
0019 info['eta'] = []
0020 info['mass'] = []
0021 info['BvL'] = []
0022 info['CvL'] = []
0023 info['CvB'] = []
0024
0025
0026 for iev, event in enumerate(events_c):
0027     event.getByLabel (labelJ, handleJ)
0028     jets = handleJ.product()
0029     print(iev)
0030     for jet in jets:
0031         #if jet.pt() < 300 or jet.pt() > 2000: continue
0032         #if jet.mass() < 40 or jet.mass() > 200: continue
0033
0034         print(jet.pt(), jet.mass(), jet.eta())
0035         print("DDB", jet.bDiscriminator("pfDeepDoubleBvLJetTags:probQCD"), jet.bDiscriminator("pfDeepDoubleBvLJetTags:probHbb"))
0036         # print("DDB", jet.bDiscriminator("pfMassIndependentDeepDoubleBvLJetTags:probQCD"), jet.bDiscriminator("pfMassIndependentDeepDoubleBvLJetTags:probHbb"))
0037         print("DDCvL", jet.bDiscriminator("pfDeepDoubleCvLJetTags:probQCD"), jet.bDiscriminator("pfDeepDoubleCvLJetTags:probHcc"))
0038         # print("DDCvL", jet.bDiscriminator("pfMassIndependentDeepDoubleCvLJetTags:probQCD"), jet.bDiscriminator("pfMassIndependentDeepDoubleCvLJetTags:probHcc"))
0039         print("DDCvB", jet.bDiscriminator("pfDeepDoubleCvBJetTags:probHbb"), jet.bDiscriminator("pfDeepDoubleCvBJetTags:probHcc") )
0040         # print("DDCvB", jet.bDiscriminator("pfMassIndependentDeepDoubleCvBJetTags:probHbb"), jet.bDiscriminator("pfMassIndependentDeepDoubleCvBJetTags:probHcc"))
0041         h_probQ_ddb.Fill(jet.bDiscriminator("pfDeepDoubleBvLJetTags:probQCD"))
0042         h_probH_ddb.Fill(jet.bDiscriminator("pfDeepDoubleBvLJetTags:probHbb"))
0043         h_probQ_ddc.Fill(jet.bDiscriminator("pfDeepDoubleCvLJetTags:probQCD"))
0044         h_probH_ddc.Fill(jet.bDiscriminator("pfDeepDoubleCvLJetTags:probHcc"))
0045
0046         info['mass'].append(jet.mass())
0047         info['pt'].append(jet.pt())
0048         info['eta'].append(jet.eta())
0049         info['BvL'].append(jet.bDiscriminator("pfDeepDoubleBvLJetTags:probHbb"))
0050         info['CvL'].append(jet.bDiscriminator("pfDeepDoubleCvLJetTags:probHcc"))
0051         info['CvB'].append(jet.bDiscriminator("pfDeepDoubleCvBJetTags:probHcc"))
0052
0053 with open('outputs.pkl', 'wb') as handle:
0054     pickle.dump(info, handle)
0055
0056 c1a = ROOT.TCanvas()
0057 h_probH_ddb.Draw("HISTO")
0058 h_probH_ddb.SetLineColor(632)
0059 h_probH_ddb.SetLineStyle(10)
0060 h_probQ_ddb.Draw("SAME")
0061 c1a.Draw()
0062 c1a.SaveAs("ProbQ_vc_vb.png")
0063
0064 c1b = ROOT.TCanvas()
0065 h_probH_ddc.Draw("HISTO")
0066 h_probH_ddc.SetLineColor(632)
0067 h_probH_ddc.SetLineStyle(10)
0068 h_probQ_ddc.Draw("SAME")
0069 c1b.Draw()
0070 c1b.SaveAs("ProbH_vc_vb.png")