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

Definition at line 19 of file trainFeasibleSegmentPairFilter.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 trainFeasibleSegmentPairFilter.py.

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

◆ identifier()

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

Definition at line 33 of file trainFeasibleSegmentPairFilter.py.

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

◆ generator_module

str generator_module = "generic"
static

use the generic event generator

Definition at line 25 of file trainFeasibleSegmentPairFilter.py.

◆ n_events

int n_events = 10000
static

number of events to generate

Definition at line 23 of file trainFeasibleSegmentPairFilter.py.

◆ task

task

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

Definition at line 53 of file trainFeasibleSegmentPairFilter.py.

◆ truth

str truth = "truth_positive"
static

degree of MC truth-matching

Definition at line 30 of file trainFeasibleSegmentPairFilter.py.


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