From 3dbe82d90fb9a7c957d015419eb596880dbd05ef Mon Sep 17 00:00:00 2001
From: Nikolai Hartmann <Nikolai.Hartmann@physik.uni-muenchen.de>
Date: Fri, 5 Oct 2018 11:24:10 +0200
Subject: [PATCH] write training weight in write_parametrized
---
scripts/write_parametrized.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/scripts/write_parametrized.py b/scripts/write_parametrized.py
index 3fcece7..a0a0ab8 100755
--- a/scripts/write_parametrized.py
+++ b/scripts/write_parametrized.py
@@ -46,6 +46,7 @@ if __name__ == "__main__":
params = tuple([float(i) for i in match.groups()])
if not params in count_dict:
count_dict[params] = 0
+ # TODO: might be better to use sum of weights
count_dict[params] += tree.GetEntries()
f.Close()
@@ -71,9 +72,11 @@ if __name__ == "__main__":
param_values = numbers[rnd_idx]
for param_idx, param_name in enumerate(param_names):
df[param_name] = param_values[:,param_idx]
+ df["training_weight"] = df["eventWeight"]*df["genWeight"]
else:
for param_name, param_value in zip(param_names, match_signal.groups()):
df[param_name] = float(param_value)
+ df["training_weight"] = df["eventWeight"]
if match_signal is None:
out_tree_name = tree_name
else:
--
GitLab