forked from glouppe/recnn
-
Notifications
You must be signed in to change notification settings - Fork 0
/
plots.py
52 lines (42 loc) · 1.19 KB
/
plots.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
from ROOT import TFile, TH2F, TGraph, TLegend
import numpy as np
f = TFile('/data/gtouquet/testdir_23Sept/ROCs.root')
tree = f.Get('NN_ROC')
stdtree = f.Get('std_ROC')
basegraph = TH2F('basegraph','#tau_h ID ROC curve',10,0.,0.1,10,0.,1.)
basegraph.SetStats(0)
basegraph.GetXaxis().SetTitle('Jet fake rate')
basegraph.GetXaxis().SetTitleSize(.045)
basegraph.GetYaxis().SetTitle('#tau ID efficiency')
basegraph.GetYaxis().SetTitleSize(.045)
tpr = np.zeros(stdtree.GetEntries())
fpr = np.zeros(stdtree.GetEntries())
i = 0
for event in stdtree:
tpr[i] = event.tpr
fpr[i] = event.fpr
#print tpr[i],fpr[i]
i+=1
graph_std = TGraph(10000,fpr,tpr)
graph_std.SetMarkerColor(2)
graph_std.SetLineColor(2)
graph_std.SetLineWidth(3)
tpr = np.zeros(tree.GetEntries())
fpr = np.zeros(tree.GetEntries())
i = 0
for event in tree:
tpr[i] = event.tpr
fpr[i] = event.fpr
#print tpr[i],fpr[i]
i+=1
graph_NN = TGraph(10000,fpr,tpr)
graph_NN.SetMarkerColor(4)
graph_NN.SetLineColor(4)
graph_NN.SetLineWidth(3)
leg = TLegend(0.6,0.1,0.9,0.3)
leg.AddEntry(graph_std,"Standard ID","l")
leg.AddEntry(graph_NN,"RecNN ID","l")
basegraph.Draw()
leg.Draw("same")
graph_NN.Draw("same")
graph_std.Draw("same")