From 1bc85cfea74d4fb7bee929b86cac1d962af8e80a Mon Sep 17 00:00:00 2001
From: Dmitry Kalinkin <dmitry.kalinkin@gmail.com>
Date: Wed, 18 Dec 2024 23:58:15 -0500
Subject: [PATCH] calo_pid: savefig inferred probability difference
---
benchmarks/calo_pid/calo_pid.org | 1 +
1 file changed, 1 insertion(+)
diff --git a/benchmarks/calo_pid/calo_pid.org b/benchmarks/calo_pid/calo_pid.org
index 965af140..c9ec2fa9 100644
--- a/benchmarks/calo_pid/calo_pid.org
+++ b/benchmarks/calo_pid/calo_pid.org
@@ -363,6 +363,7 @@ if "_EcalEndcapNParticleIDOutput_probability_tensor_floatData" in pi_train.field
eval_proba = ak.concatenate([pi_eval_proba, e_eval_proba])
plt.hist(clf.predict_proba(eval_x.to_numpy())[:,1] - eval_proba[:,1].to_numpy())
+ plt.savefig(output_dir / f"proba_diff.pdf", bbox_inches="tight")
plt.show()
else:
print("EcalEndcapNParticleIDOutput not present")
--
GitLab