Belle II Software development
RealisticTrackRelationFilterTrainingRun Class Reference
Inheritance diagram for RealisticTrackRelationFilterTrainingRun:
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 track relations encountered at the TrackLinker after the feasible filter and retrain its mva method

Definition at line 19 of file trainRealisticTrackRelationFilter.py.

Member Function Documentation

◆ create_path()

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

Reimplemented from ReadOrGenerateEventsRun.

Definition at line 37 of file trainRealisticTrackRelationFilter.py.

37 def create_path(self):
38 """Setup the recording path after the simulation"""
39 path = super().create_path()
40 path.add_module("TFCDC_WireHitPreparer",
41 flightTimeEstimation="outwards")
42
43 path.add_module('TFCDC_ClusterPreparer',
44 SuperClusterDegree=3,
45 SuperClusterExpandOverApogeeGap=True)
46
47 path.add_module("TFCDC_SegmentFinderFacetAutomaton")
48
49
50 if self.task == "train":
51 varSets = [
52 "realistic",
53 "filter(truth)",
54 ]
55 skim = "feasible"
56
57 elif self.task == "eval":
58 varSets = [
59 "filter(feasible)",
60 "filter(realistic)",
61 "filter(truth)",
62 ]
63 skim = ""
64
65 elif self.task == "explore":
66 varSets = [
67 "realistic",
68 "hit_gap",
69 "fit",
70 "filter(truth)",
71 ]
72 skim = "feasible"
73
74 else:
75 raise ValueError("Unknown task " + self.task)
76
77 path.add_module("TFCDC_TrackFinderSegmentPairAutomaton",
78 TrackRelationFilter="unionrecording",
79 TrackRelationFilterParameters={
80 "rootFileName": self.sample_file_name,
81 "varSets": varSets,
82 "skim": skim,
83 })
84
85 return path
86
87

◆ identifier()

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

Definition at line 33 of file trainRealisticTrackRelationFilter.py.

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

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 27 of file trainRealisticTrackRelationFilter.py.

◆ generator_module

str generator_module = "generic"
static

use the generic event generator

Definition at line 25 of file trainRealisticTrackRelationFilter.py.

◆ n_events

int n_events = 10000
static

number of events to generate

Definition at line 23 of file trainRealisticTrackRelationFilter.py.

◆ task

task

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

Definition at line 50 of file trainRealisticTrackRelationFilter.py.

◆ truth

str truth = "truth_positive"
static

degree of MC truth-matching

Definition at line 30 of file trainRealisticTrackRelationFilter.py.


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