Belle II Software development
FeasibleTrackRelationFilterTrainingRun Class Reference
Inheritance diagram for FeasibleTrackRelationFilterTrainingRun:
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 and retrain its mva method

Definition at line 19 of file trainFeasibleTrackRelationFilter.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 trainFeasibleTrackRelationFilter.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 "feasible",
53 "filter(truth)",
54 ]
55
56 elif self.task == "eval":
57 varSets = [
58 "filter(feasible)",
59 "filter(truth)",
60 ]
61
62 elif self.task == "explore":
63 varSets = [
64 "basic",
65 "hit_gap",
66 "filter(truth)",
67 ]
68
69 else:
70 raise ValueError("Unknown task " + self.task)
71
72 path.add_module("TFCDC_TrackFinderSegmentPairAutomaton",
73 TrackRelationFilter="unionrecording",
74 TrackRelationFilterParameters={
75 "rootFileName": self.sample_file_name,
76 "varSets": varSets,
77 })
78
79 return path
80
81

◆ identifier()

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

Definition at line 33 of file trainFeasibleTrackRelationFilter.py.

33 def identifier(self):
34 """Database identifier of the filter being trained"""
35 return "trackfindingcdc_FeasibleTrackRelationFilter.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 trainFeasibleTrackRelationFilter.py.

◆ generator_module

str generator_module = "generic"
static

use the generic event generator

Definition at line 25 of file trainFeasibleTrackRelationFilter.py.

◆ n_events

int n_events = 10000
static

number of events to generate

Definition at line 23 of file trainFeasibleTrackRelationFilter.py.

◆ task

task

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

Definition at line 50 of file trainFeasibleTrackRelationFilter.py.

◆ truth

str truth = "truth_positive"
static

degree of MC truth-matching

Definition at line 30 of file trainFeasibleTrackRelationFilter.py.


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