python.physvalPostProcessingTools.ROCCurveComputation Class Reference

Collaboration diagram for python.physvalPostProcessingTools.ROCCurveComputation:

Public Member Functions
	__init__ (self, signal_path, background_path, output)
	__call__ (self, root_file)
	compute_roc (self, sig, bkg)

Static Public Member Functions
	from_yaml (fragment)

Public Attributes
	signal_path = signal_path
	background_path = background_path
	output = output

Detailed Description

Definition at line 285 of file physvalPostProcessingTools.py.

Constructor & Destructor Documentation

__init__()

python.physvalPostProcessingTools.ROCCurveComputation.__init__	(		self,
			signal_path,
			background_path,
			output )

Definition at line 286 of file physvalPostProcessingTools.py.

    def __init__(self, signal_path, background_path, output):
        self.signal_path = signal_path
        self.background_path = background_path
        self.output = output
 

Member Function Documentation

__call__()

python.physvalPostProcessingTools.ROCCurveComputation.__call__	(		self,
			root_file )

Definition at line 291 of file physvalPostProcessingTools.py.

    def __call__(self, root_file):
        sig = find_histo(root_file, self.signal_path)
        bkg = find_histo(root_file, self.background_path)
 
        if not sig or not bkg: 
            if crash_on_error:
                raise RuntimeError(f"Missing histogram: {self.signal_path} or {self.background_path}")
            else:
                logging.error(f"Missing histogram: {self.signal_path} or {self.background_path}")
                return
 
        graph = self.compute_roc(sig, bkg)
        if not graph:
            logging.error("ROC graph is empty or could not be constructed.")
            return
 
        path_parts = self.output.strip("/").split("/")
        *dir_path, obj_name = path_parts
        directory_path = "/".join(dir_path)
 
        graph.SetName(obj_name)
        graph.SetTitle(f"{obj_name};Signal Efficiency;Background Rejection")
 
        ensure_directory_exists(root_file, directory_path)
        root_file.cd(directory_path)
        graph.Write()
        logging.info(f"Saved ROC curve '{obj_name}' in '{directory_path}'.")
 

compute_roc()

python.physvalPostProcessingTools.ROCCurveComputation.compute_roc	(		self,
			sig,
			bkg )

Definition at line 319 of file physvalPostProcessingTools.py.

    def compute_roc(self, sig, bkg):
        n_bins = sig.GetNbinsX()
        total_sig = sig.Integral()
        total_bkg = bkg.Integral()
 
        x_vals, y_vals = [], []
        if total_sig == 0 or total_bkg == 0:
            logging.error("Zero total signal or background counts; cannot compute ROC.")
            return None
 
        for cut_bin in range(1, n_bins + 1): # Loops over histogram bins, each corresponding to a cut value on the tagger output.
            sig_pass = sig.Integral(cut_bin, n_bins)
            bkg_pass = bkg.Integral(cut_bin, n_bins)
 
            eff = sig_pass / total_sig # Signal efficiency eff = (number of b-jets passing cut) / (total b-jets).
 
            if bkg_pass <= 0:
                # logging.info(f"Skipping cut bin {cut_bin} due to zero or negative background pass count: {bkg_pass}. Any other options?")
                continue
 
            rej = total_bkg / bkg_pass # Background rejection rej = (total background jets) / (number of background jets passing cut).
 
            x_vals.append(eff)
            y_vals.append(rej)
 
        if not x_vals:
            return None
 
        graph = ROOT.TGraphErrors(len(x_vals))
        for i, (x, y) in enumerate(zip(x_vals, y_vals)):
            graph.SetPoint(i, x, y)
 
        return graph
 

from_yaml()

python.physvalPostProcessingTools.ROCCurveComputation.from_yaml ( fragment )

static

Definition at line 354 of file physvalPostProcessingTools.py.

    def from_yaml(fragment):
        return ROCCurveComputation(
            signal_path=fragment["signal"],
            background_path=fragment["background"],
            output=fragment["output"]
        )

Member Data Documentation

background_path

python.physvalPostProcessingTools.ROCCurveComputation.background_path = background_path

Definition at line 288 of file physvalPostProcessingTools.py.

output

python.physvalPostProcessingTools.ROCCurveComputation.output = output

Definition at line 289 of file physvalPostProcessingTools.py.

signal_path

python.physvalPostProcessingTools.ROCCurveComputation.signal_path = signal_path

Definition at line 287 of file physvalPostProcessingTools.py.

---

The documentation for this class was generated from the following file:

• physvalPostProcessingTools.py