def Init():
from ROOT import gStyle
from ROOT import gROOT
gROOT.ProcessLine("gErrorIgnoreLevel = 2000;")
gStyle.SetCanvasColor(0);
gStyle.SetCanvasBorderMode(0);
gStyle.SetCanvasBorderSize(3);
gStyle.SetPadLeftMargin(0.125);
gStyle.SetPadBottomMargin(0.12);
gStyle.SetPadColor(0);
gStyle.SetPadBorderMode(0);
gStyle.SetFrameBorderMode(0);
gStyle.SetFrameBorderSize(0);
gStyle.SetFrameFillColor(0);
gStyle.SetOptStat(0);
gStyle.SetLabelOffset(0.005,"X");
gStyle.SetLabelSize(0.03,"X");
gStyle.SetLabelFont(42,"X");
gStyle.SetTitleOffset(.85,"X");
gStyle.SetTitleSize(0.04,"X");
gStyle.SetTitleFont(42,"X");
gStyle.SetLabelOffset(0.005,"Y");
gStyle.SetLabelSize(0.03,"Y");
gStyle.SetLabelFont(42,"Y");
gStyle.SetTitleOffset(.98,"Y");
gStyle.SetTitleSize(0.04,"Y");
gStyle.SetTitleFont(42,"Y");
gStyle.SetStatColor(0);
gStyle.SetStatBorderSize(0);
gStyle.SetTextFont(2);
gStyle.SetTextSize(.05);
gStyle.SetLegendBorderSize(1);
gStyle.SetHistLineWidth(2);
gStyle.SetTitleFillColor(0);
gStyle.SetTitleFontSize(0.06);
gStyle.SetTitleBorderSize(0);
gStyle.SetTitleAlign(13);
gStyle.SetTextAlign(22);
#RJ
gStyle.SetPadTickX(1);
gStyle.SetPadTickY(1);
gStyle.SetPadTopMargin(0.05);
gStyle.SetPadBottomMargin(0.13);
gStyle.SetPadLeftMargin(0.16);
gStyle.SetPadRightMargin(0.03);
def Styling():
# Center title
gStyle.SetTitleAlign(22)
gStyle.SetTitleX(.5)
gStyle.SetTitleY(.95)
gStyle.SetTitleBorderSize(0)
# Remove stats box
gStyle.SetOptStat(0)
# Set background color to white
gStyle.SetFillColor(10)
gStyle.SetFrameFillColor(10)
gStyle.SetCanvasColor(10)
gStyle.SetPadColor(10)
gStyle.SetTitleFillColor(0)
gStyle.SetStatColor(10)
# No colored frames around plots
gStyle.SetFrameBorderMode(0)
gStyle.SetCanvasBorderMode(0)
gStyle.SetPadBorderMode(0)
# Set the default line color for a fit function to be red
gStyle.SetFuncColor(2)
# Marker settings
gStyle.SetMarkerStyle(20)
# No border on legends
gStyle.SetLegendBorderSize(0)
# Disabled for violating NOvA style guidelines
# Scientific notation on axes
TGaxis.SetMaxDigits(3)
# Axis titles
gStyle.SetTitleSize(.055, "xyz")
gStyle.SetTitleOffset(2.2, "xyz")
gStyle.SetTitleOffset(2.2, "y")
gStyle.SetTitleSize(.05, "")
gStyle.SetTitleOffset(2.2, "")
# Axis labels (numbering)
gStyle.SetLabelSize(.05, "xyz")
gStyle.SetLabelOffset(.005, "xyz")
# Thicker lines
gStyle.SetHistLineWidth(2)
gStyle.SetFrameLineWidth(2)
gStyle.SetFuncWidth(2)
# Set the number of tick marks
gStyle.SetNdivisions(506, "xyz")
# Set tick marks on all sides
gStyle.SetPadTickX(1)
gStyle.SetPadTickY(1)
# Fonts
kNovaFont = 42
gStyle.SetStatFont(kNovaFont)
gStyle.SetLabelFont(kNovaFont, "xyz")
gStyle.SetTitleFont(kNovaFont, "xyz")
gStyle.SetTitleFont(kNovaFont, "")
# Apply same settings to titles
gStyle.SetTextFont(kNovaFont)
gStyle.SetLegendFont(kNovaFont)
def looks():
gROOT.ProcessLine('TGaxis::SetMaxDigits(3)')
gStyle.SetFrameBorderMode(0)
gStyle.SetFrameFillColor(0)
gStyle.SetCanvasBorderMode(0)
gStyle.SetCanvasColor(0)
gStyle.SetPadBorderMode(0)
gStyle.SetPadColor(0)
gStyle.SetStatColor(0)
gStyle.SetTitleFont(42, '')
gStyle.SetLabelFont(42, 'x')
gStyle.SetTitleFont(42, 'x')
gStyle.SetLabelFont(42, 'y')
gStyle.SetTitleFont(42, 'y')
gStyle.SetLabelFont(42, 'z')
gStyle.SetTitleFont(42, 'z')
gStyle.SetLabelSize(0.048, 'x')
gStyle.SetTitleSize(0.048, 'x')
gStyle.SetLabelSize(0.048, 'y')
gStyle.SetTitleSize(0.048, 'y')
gStyle.SetLabelSize(0.048, 'z')
gStyle.SetTitleSize(0.048, 'z')
gStyle.SetTitleSize(0.048, '')
gStyle.SetTitleAlign(23)
gStyle.SetTitleX(0.5)
gStyle.SetTitleBorderSize(0)
gStyle.SetTitleFillColor(0)
gStyle.SetTitleStyle(0)
gStyle.SetOptStat('emr')
gStyle.SetOptStat(0)
gStyle.SetStatBorderSize(0)
gStyle.SetStatFont(42)
gStyle.SetStatFontSize(0.048)
gStyle.SetStatY(0.9)
gStyle.SetStatX(0.86)
gStyle.SetHistLineColor(ROOT.kBlue + 2)
gStyle.SetLegendBorderSize(0)
gStyle.SetLegendFillColor(0)
gStyle.SetFuncWidth(2)
gStyle.SetFuncColor(2)
gStyle.SetPadTopMargin(0.08)
gStyle.SetPadBottomMargin(0.12)
gStyle.SetPadLeftMargin(0.12)
gStyle.SetPadRightMargin(0.06)
gStyle.SetCanvasDefX(400)
gStyle.SetCanvasDefY(20)
gStyle.SetCanvasDefH(420)
gStyle.SetCanvasDefW(610)
gStyle.SetFrameBorderMode(0)
gStyle.SetFrameLineWidth(2)
gStyle.SetHistLineWidth(2)
gStyle.SetTitleOffset(1.16, 'y')
gStyle.SetTitleOffset(1.20, 'x')
def __init__(self):
'''
Set up PlotStyle, set gStyle for things we always want no matter what.
'''
# CMS-approved everything
tdrstyle.setTDRStyle()
### Differences from TDR standard:
# Big canvas (can always shrink later)
gStyle.SetCanvasDefH(1200)
gStyle.SetCanvasDefW(1200)
# Tick marks on all sides
gStyle.SetPadTickX(1)
gStyle.SetPadTickY(1)
# Everything has white backgrounds
gStyle.SetLegendFillColor(0)
# Colors that don't suck
gStyle.SetPalette(1)
# Make axis title and labels just a little smaller and (for Y) closer to the axis
gStyle.SetTitleSize(0.044, "XYZ")
gStyle.SetLabelSize(0.033, "XYZ")
gStyle.SetTitleYOffset(1.15)
gStyle.SetTitleXOffset(0.95)
gStyle.SetPadLeftMargin(0.1)
gStyle.SetPadRightMargin(0.025)
gStyle.SetPadBottomMargin(0.095)
gStyle.SetTitleAlign(12)
# Apply changes
gROOT.ForceStyle()
# Force exponentials when axes are over 3 digits
TGaxis.SetMaxDigits(3)
TGaxis.SetExponentOffset(-0.060, 0.008, "y")
TGaxis.SetExponentOffset(
-0.055, -0.062,
"x") # will overlap with title unless title is centered
# help='Which ML method? DNN, BDT, MLP, or kNN')
(options, args) = parser.parse_args()
argv = []
folder = "projections_" + str(options.n) + "_" + str(options.layers)
#ML = str(options.ML)
eff = options.eff
gStyle.SetPadLeftMargin(0.1)
gStyle.SetPadBottomMargin(0.1)
gStyle.SetPadRightMargin(0.18)
gStyle.SetPadTopMargin(0.1)
gStyle.SetTitleX(0.3)
gStyle.SetTitleAlign(23)
planes = [
"Hit_plane0", "Hit_plane1", "Hit_plane2", "Hit_plane3", "Hit_plane4",
"Hit_plane5", "Hit_plane6", "Hit_plane7"
]
names = [("Hit_plane0", "X0"), ("Hit_plane1", "X1"), ("Hit_plane2", "U0"),
("Hit_plane3", "V0"), ("Hit_plane4", "U1"), ("Hit_plane5", "V1"),
("Hit_plane6", "X2"), ("Hit_plane7", "X3")]
samples = [("sig", "Signal"), ("bkg", "Background")]
MLlist = ["DNN", "BDT", "MLP"]
#MLlist = ["DNN"]
for ML in MLlist: