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

Public Member Functions

def identifier (self)
 
def create_path (self)
 

Public Attributes

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

Static Public Attributes

int n_events = 10000
 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_positive"
 degree of MC truth-matching
 

Detailed Description

Run to record segment pairs encountered at the AxialSegmentPairCreator and retrain its mva method

Definition at line 19 of file trainRealisticAxialSegmentPairFilter.py.

Member Function Documentation

◆ create_path()

def create_path (   self)
Setup the recording path after the simulation

Reimplemented from ReadOrGenerateEventsRun.

Definition at line 36 of file trainRealisticAxialSegmentPairFilter.py.

36 def create_path(self):
37 """Setup the recording path after the simulation"""
38 path = super().create_path()
39
40 # In contrast to other training use only the first *half* loop for more aggressive training
41 path.add_module("TFCDC_WireHitPreparer",
42 flightTimeEstimation="outwards")
43
44 path.add_module('TFCDC_ClusterPreparer',
45 SuperClusterDegree=3,
46 SuperClusterExpandOverApogeeGap=True)
47
48 path.add_module("TFCDC_SegmentFinderFacetAutomaton")
49
50 # Also fix the segment orientation to outwards to make training additionally aggressive
51
52 if self.task == "train":
53 varSets = [
54 "realistic",
55 "filter(truth)",
56 ]
57 skim = "feasible"
58
59 elif self.task == "eval":
60 varSets = [
61 "filter(truth)",
62 "filter(realistic)",
63 "filter(feasible)",
64 "filter(simple)",
65 ]
66 skim = ""
67
68 elif self.task == "explore":
69 varSets = [
70 "basic",
71 "fitless",
72 "hit_gap",
73 "fit",
74 # "filter(simple)",
75 # "filter(feasible)",
76 # "filter(realistic)",
77 "filter(truth)",
78 ]
79 skim = "feasible"
80
81 else:
82 raise ValueError("Unknown task " + self.task)
83
84 path.add_module("TFCDC_TrackFinderSegmentPairAutomaton",
85 SegmentPairAxialBridging=True,
86 AxialSegmentPairFilter="unionrecording",
87 AxialSegmentPairFilterParameters={
88 "rootFileName": self.sample_file_name,
89 "varSets": varSets,
90 "skim": skim,
91 },
92 SegmentPairFilter='none',
93 SegmentPairRelationFilter='none')
94
95 return path
96
97

◆ identifier()

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

Definition at line 32 of file trainRealisticAxialSegmentPairFilter.py.

32 def identifier(self):
33 """Database identifier of the filter being trained"""
34 return "trackfindingcdc_RealisticAxialSegmentPairFilter.xml"
35

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 26 of file trainRealisticAxialSegmentPairFilter.py.

◆ generator_module

str generator_module = "generic"
static

use the generic event generator

Definition at line 24 of file trainRealisticAxialSegmentPairFilter.py.

◆ n_events

int n_events = 10000
static

number of events to generate

Definition at line 22 of file trainRealisticAxialSegmentPairFilter.py.

◆ task

task

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

Definition at line 52 of file trainRealisticAxialSegmentPairFilter.py.

◆ truth

str truth = "truth_positive"
static

degree of MC truth-matching

Definition at line 29 of file trainRealisticAxialSegmentPairFilter.py.


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