SegmentTrackFilterTrainingRun Class Reference

Inheritance diagram for SegmentTrackFilterTrainingRun:

Collaboration diagram for SegmentTrackFilterTrainingRun:

Public Member Functions
def	identifier (self)
def	create_path (self)
def	postprocess (self)

Public Attributes
	task
	Process each event according to the user's desired task (train, eval, explore)
	truth
	modify degree of MC truth-matching

Static Public Attributes
int	n_events = 2000
	number of events to generate
str	generator_module = "generic"
	use the generic event generator
os	bkg_files = os.path.join(os.environ["VO_BELLE2_SW_DIR"], "bkg")
	overlay background hits from the events in these files
str	truth = "truth"
	degree of MC truth-matching

Detailed Description

Run to record segment track combinations encountered at the SegmentTrackSelector and retrain its mva method

Definition at line 22 of file trainSegmentTrackFilter.py.

Member Function Documentation

create_path()

def create_path

(

self

)

Setup the recording path after the simulation

Reimplemented from ReadOrGenerateEventsRun.

Definition at line 40 of file trainSegmentTrackFilter.py.

    def create_path(self):
        """Setup the recording path after the simulation"""
        path = super().create_path()
 
        add_cdc_track_finding(path)
 
        
        if self.task == "train":
            filterName = "recording"
 
        elif self.task == "eval":
            filterName = "eval"
            
            self.truth = "truth_accepted"
 
        elif self.task == "explore":
            # Change me.
            filterName = "recording"
 
        adjust_module(path, "TFCDC_SegmentTrackCombiner",
                      segmentTrackFilter=filterName,
                      segmentTrackFilterParameters={
                          "rootFileName": self.sample_file_name,
                      },
                      trackFilter="all",
                      trackFilterParameters={})
 
        return path
 

identifier()

def identifier

(

self

)

Database identifier of the filter being trained

Definition at line 36 of file trainSegmentTrackFilter.py.

    def identifier(self):
        """Database identifier of the filter being trained"""
        return "trackfindingcdc_SegmentTrackFilter.xml"
 

postprocess()

def postprocess

(

self

)

Post-process the results for MC-truth acceptance

Definition at line 69 of file trainSegmentTrackFilter.py.

    def postprocess(self):
        """Post-process the results for MC-truth acceptance"""
        if self.task == "eval":
            self.truth = "truth_accept"
        super().postprocess()
 
 

Member Data Documentation

bkg_files

os bkg_files = os.path.join(os.environ["VO_BELLE2_SW_DIR"], "bkg")

static

overlay background hits from the events in these files

Definition at line 30 of file trainSegmentTrackFilter.py.

generator_module

str generator_module = "generic"

static

use the generic event generator

Definition at line 28 of file trainSegmentTrackFilter.py.

n_events

int n_events = 2000

static

number of events to generate

Definition at line 26 of file trainSegmentTrackFilter.py.

task

task

Process each event according to the user's desired task (train, eval, explore)

Definition at line 47 of file trainSegmentTrackFilter.py.

truth [1/2]

str truth = "truth"

static

degree of MC truth-matching

Definition at line 33 of file trainSegmentTrackFilter.py.

truth [2/2]

truth

modify degree of MC truth-matching

Definition at line 53 of file trainSegmentTrackFilter.py.

---

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

• tracking/trackFindingCDC/train/trainSegmentTrackFilter.py