Belle II Software development
TrackFilterTrainingRun Class Reference
Inheritance diagram for TrackFilterTrainingRun:
ReadOrGenerateEventsRun MinimalRun EmptyRun

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 = 3000
 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 tracks encountered at the TrackRejecter and retrain its mva method

Definition at line 22 of file trainTrackFilter.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 trainTrackFilter.py.

40 def create_path(self):
41 """Setup the recording path after the simulation"""
42 path = super().create_path()
43
44 add_cdc_track_finding(path)
45
46
47 if self.task == "train":
48 filterName = "recording"
49
50 elif self.task == "eval":
51 filterName = "eval"
52
53 self.truth = "truth_accepted"
54
55 elif self.task == "explore":
56 # Change me.
57 filterName = "recording"
58
59 adjust_module(path, "TFCDC_SegmentTrackCombiner",
60 trackFilter=filterName,
61 trackFilterParameters={
62 "rootFileName": self.sample_file_name,
63 })
64
65 return path
66

◆ identifier()

def identifier (   self)
Database identifier of the filter being trained

Definition at line 36 of file trainTrackFilter.py.

36 def identifier(self):
37 """Database identifier of the filter being trained"""
38 return "trackfindingcdc_TrackFilter.xml"
39

◆ postprocess()

def postprocess (   self)
Post-process the results for MC-truth acceptance

Definition at line 67 of file trainTrackFilter.py.

67 def postprocess(self):
68 """Post-process the results for MC-truth acceptance"""
69 if self.task == "eval":
70 self.truth = "truth_accept"
71 super().postprocess()
72
73

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 trainTrackFilter.py.

◆ generator_module

str generator_module = "generic"
static

use the generic event generator

Definition at line 28 of file trainTrackFilter.py.

◆ n_events

int n_events = 3000
static

number of events to generate

Definition at line 26 of file trainTrackFilter.py.

◆ task

task

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

Definition at line 47 of file trainTrackFilter.py.

◆ truth [1/2]

str truth = "truth"
static

degree of MC truth-matching

Definition at line 33 of file trainTrackFilter.py.

◆ truth [2/2]

truth

Modify degree of MC truth-matching.

Definition at line 53 of file trainTrackFilter.py.


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