Merge pull request #55 from ucsd-hep-ex/mix_baseline

Mix baseline training and predictions
ucsd-hep-ex · Dec 4, 2024 · 8b61867 · 8b61867
2 parents 43b617a + 83ba1f3
commit 8b61867
Show file tree

Hide file tree

Showing 10 changed files with 2,276 additions and 76 deletions.
diff --git a/models/bdt_trained_on_hhh_qcd.json b/models/bdt_trained_on_hhh_qcd.json
diff --git a/notebooks/bdt/hhh_qcd_bdt.ipynb b/notebooks/bdt/hhh_qcd_bdt.ipynb
@@ -34,7 +34,7 @@
    },
    "outputs": [],
    "source": [
-    "fn_delphes = \"../../data/delphes/v2/merged_training_all.h5\"\n",
+    "fn_delphes = \"../../data/delphes/v2/hhh_training.h5"\"\n",
     "f2 = h5py.File(fn_delphes, \"r+\")"
    ]
   },
@@ -304,7 +304,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "6254969 1948467\n"
+      "553548 164601\n"
      ]
     }
    ],
@@ -562,7 +562,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "booster.save_model('bdt_trained_on_all.json')"
+    "booster.save_model('../../models/bdt_trained_on_hhh_qcd.json')"
    ]
   },
   {
@@ -600,7 +600,7 @@
     {
      "data": {
       "text/plain": [
-       "0.95553684"
+       "0.95626426"
       ]
      },
      "execution_count": 21,
@@ -621,7 +621,7 @@
     {
      "data": {
       "text/plain": [
-       "0.9346559"
+       "0.93498826"
       ]
      },
      "execution_count": 22,
@@ -642,7 +642,7 @@
     {
      "data": {
       "text/plain": [
-       "0.91041"
+       "0.911348"
       ]
      },
      "execution_count": 23,