Collaboration diagram for python.physvalPostProcessingTools.ROCCurveComputation:

Public Member Functions
def	__init__ (self, signal_path, background_path, output)

def	__call__ (self, root_file)

def	compute_roc (self, sig, bkg)

Static Public Member Functions
def	from_yaml (fragment)

Public Attributes
	signal_path

	background_path

	output

Detailed Description

Definition at line 212 of file physvalPostProcessingTools.py.

Constructor & Destructor Documentation

◆ init()

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

Definition at line 213 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()

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

Definition at line 218 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()

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

Definition at line 246 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()

def python.physvalPostProcessingTools.ROCCurveComputation.from_yaml ( fragment )

static

Definition at line 281 of file physvalPostProcessingTools.py.

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