update bdt shape plots

rkansal47 · Jan 30, 2024 · 94648fb · 94648fb
1 parent 936e138
commit 94648fb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/HHbbVV/postprocessing/PostProcess.ipynb b/src/HHbbVV/postprocessing/PostProcess.ipynb
@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 1,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -79,7 +79,7 @@
     "# bdt_data_dir = \"/eos/uscms/store/user/cmantill/bbVV/skimmer/Jun10/bdt_data/\"\n",
     "year = \"2018\"\n",
     "\n",
-    "date = \"24Jan13\"\n",
+    "date = \"24Jan18\"\n",
     "plot_dir = f\"../../../plots/PostProcessing/{date}/\"\n",
     "templates_dir = f\"templates/{date}\"\n",
     "_ = os.system(f\"mkdir -p {plot_dir}\")\n",

diff --git a/src/HHbbVV/postprocessing/TrainBDT.py b/src/HHbbVV/postprocessing/TrainBDT.py
@@ -538,6 +538,7 @@ def evaluate_model(
         utils.ShapeVar("BDTScore", "BDT Score", [20, 0.4, 1]),
         utils.ShapeVar("BDTScore", "BDT Score", [20, 0.8, 1]),
         utils.ShapeVar("BDTScore", "BDT Score", [20, 0.9, 1]),
+        utils.ShapeVar("BDTScore", "BDT Score", [20, 0.98, 1]),
     ]
 
     save_model_dir = f"{model_dir}/hists/"