Skip to content

Commit

Permalink
plot fits
Browse files Browse the repository at this point in the history
  • Loading branch information
@rkansal47
rkansal47 committed Jul 18, 2024
1 parent 50b803a commit b5510bd
Show file tree
Hide file tree
Showing 2 changed files with 50 additions and 19 deletions.
36 changes: 23 additions & 13 deletions src/HHbbVV/postprocessing/PlotFits.ipynb
View file
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,7 @@
"k2v0sig = False\n",
"unblinded = True\n",
"\n",
"plot_dir = MAIN_DIR / f\"plots/PostFit/24Jun5\"\n",
"plot_dir = MAIN_DIR / f\"plots/PostFit/24Jul15Fig8Prelim\"\n",
"# plot_dir = (\n",
"# MAIN_DIR\n",
"# / \"plots/PostFit/24Apr9ggFScan/nTF1/ggf_txbb_MP_ggf_bdt_0.9965_vbf_txbb_HP_vbf_bdt_0.999_lepton_veto_Hbb\"\n",
Expand Down Expand Up @@ -217,16 +217,7 @@
"metadata": {},
"outputs": [],
"source": [
"plot_dir"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"ylims = {\"passggf\": 40, \"passvbf\": 7, \"fail\": 6e5}\n",
"ylims = {\"passggf\": 40, \"passvbf\": 9, \"fail\": 6e5}\n",
"title_label = \" Asimov Dataset\" if asimov else \"\"\n",
"sig_scale_dict = {\n",
" \"HHbbVV\": 100,\n",
Expand All @@ -237,6 +228,8 @@
"sig_scale_dict = {key: val for key, val in sig_scale_dict.items() if key in sig_keys}\n",
"\n",
"for shape, shape_label in shapes.items():\n",
" if shape != \"postfit\":\n",
" continue\n",
" for region, region_label in selection_regions.items():\n",
" pass_region = region.startswith(\"pass\")\n",
" for i, shape_var in enumerate(shape_vars):\n",
Expand All @@ -246,11 +239,14 @@
" \"bg_keys\": bg_keys,\n",
" \"bg_err\": bgerrs[shape][region],\n",
" \"sig_scale_dict\": sig_scale_dict if pass_region else None,\n",
" \"show\": False,\n",
" \"show\": True,\n",
" \"year\": \"all\",\n",
" \"ylim\": ylims[region],\n",
" \"title\": f\"{shape_label} {region_label} Region{title_label}\",\n",
" \"name\": f\"{plot_dir}/{shape}_{region}_{shape_var.var}.pdf\",\n",
" \"ncol\": 2 if region == \"passvbf\" else 1,\n",
" \"ratio_ylims\": [0, 5] if region == \"passvbf\" else [0, 2],\n",
" \"cmslabel\": \"Preliminary\",\n",
" }\n",
"\n",
" plotting.ratioHistPlot(**plot_params, data_err=True)\n",
Expand All @@ -259,6 +255,20 @@
" # break"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": []
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"## QCD Transfer Factor"
]
},
{
"cell_type": "code",
"execution_count": null,
Expand Down Expand Up @@ -331,7 +341,7 @@
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.9.15"
"version": "undefined.undefined.undefined"
},
"orig_nbformat": 4,
"vscode": {
Expand Down
33 changes: 27 additions & 6 deletions src/HHbbVV/postprocessing/plotting.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -44,6 +44,8 @@
"qqHH_CV_1_C2V_1_kl_1_HHbbVV": "VBF HHbbVV",
"qqHH_CV_1_C2V_0_kl_1_HHbbVV": r"VBF HHbbVV ($\kappa_{2V} = 0$)",
"qqHH_CV_1_C2V_2_kl_1_HHbbVV": r"VBF HHbbVV ($\kappa_{2V} = 2$)",
"ST": r"Single-$t$",
"TT": r"$t\bar{t}$",
}

colours = {
Expand Down Expand Up @@ -137,7 +139,7 @@ def _process_samples(sig_keys, bg_keys, bg_colours, sig_scale_dict, bg_order, sy
# set up samples, colours and labels
bg_keys = [key for key in bg_order if key in bg_keys]
bg_colours = [colours[bg_colours[sample]] for sample in bg_keys]
bg_labels = deepcopy(bg_keys)
bg_labels = [sample_label_map.get(bg_key, bg_key) for bg_key in bg_keys]

if sig_scale_dict is None:
sig_scale_dict = OrderedDict([(sig_key, 1.0) for sig_key in sig_keys])
Expand All @@ -149,7 +151,7 @@ def _process_samples(sig_keys, bg_keys, bg_colours, sig_scale_dict, bg_order, sy
label = sample_label_map.get(sig_key, sig_key)

if sig_scale != 1:
if sig_scale <= 100:
if sig_scale <= 10000:
label = f"{label} $\\times$ {sig_scale:.0f}"
else:
label = f"{label} $\\times$ {sig_scale:.1e}"
Expand Down Expand Up @@ -251,6 +253,8 @@ def ratioHistPlot(
significance_dir: str = "right",
plot_ratio: bool = True,
axrax: tuple = None,
ncol: int = None,
cmslabel: str = None,
):
"""
Makes and saves a histogram plot, with backgrounds stacked, signal separate (and optionally
Expand Down Expand Up @@ -289,6 +293,7 @@ def ratioHistPlot(
plot_significance (bool): plot Asimov significance below ratio plot
significance_dir (str): "Direction" for significance. i.e. a > cut ("right"), a < cut ("left"), or per-bin ("bin").
axrax (Tuple): optionally input ax and rax instead of creating new ones
ncol (int): # of legend columns. By default, it is 2 for log-plots and 1 for non-log-plots.
"""

if ratio_ylims is None:
Expand All @@ -297,6 +302,8 @@ def ratioHistPlot(
bg_colours = BG_COLOURS
if sig_colours is None:
sig_colours = SIG_COLOURS
if ncol is None:
ncol = 2 if log else 1

# copy hists and bg_keys so input objects are not changed
hists, bg_keys = deepcopy(hists), deepcopy(bg_keys)
Expand Down Expand Up @@ -460,7 +467,7 @@ def ratioHistPlot(
# two column legend
ax.legend(fontsize=16, ncol=2)
else:
ax.legend(fontsize=16)
ax.legend(fontsize=16, ncol=ncol)

y_lowlim = 0 if not log else 1e-5
if ylim is not None:
Expand All @@ -472,10 +479,20 @@ def ratioHistPlot(
if plot_ratio:
if plot_data:
bg_tot = sum([pre_divide_hists[sample, :] for sample in bg_keys])
yerr = ratio_uncertainty(
pre_divide_hists[data_key, :].values(), bg_tot.values(), "poisson"
# new: plotting data errors (black lines) and background errors (shaded) separately
yerr = np.nan_to_num(
np.abs(
poisson_interval(pre_divide_hists[data_key, ...])
- pre_divide_hists[data_key, ...]
)
/ (bg_tot.values() + 1e-5)
)

# old version: using Garwood ratio intervals
# yerr = ratio_uncertainty(
# pre_divide_hists[data_key, :].values(), bg_tot.values(), "poisson"
# )

hep.histplot(
pre_divide_hists[data_key, :] / (bg_tot.values() + 1e-5),
yerr=yerr,
Expand All @@ -500,6 +517,10 @@ def ratioHistPlot(
rax.set_xlabel(hists.axes[1].label)

rax.set_ylabel("Data/MC")
# rax.set_yscale("log")
# formatter = mticker.ScalarFormatter(useOffset=False)
# formatter.set_scientific(False)
# rax.yaxis.set_major_formatter(formatter)
rax.set_ylim(ratio_ylims)
rax.grid()

Expand Down Expand Up @@ -539,7 +560,7 @@ def ratioHistPlot(
if title is not None:
ax.set_title(title, y=1.08)

add_cms_label(ax, year)
add_cms_label(ax, year, label=cmslabel)

if axrax is None:
if len(name):
Expand Down

0 comments on commit b5510bd

Please sign in to comment.