16from trackfindingcdc.run.training
import TrainingRunMixin
20 """Run to record track relations encountered at the TrackLinker after the feasible filter and retrain its mva method"""
25 generator_module =
"generic"
27 bkg_files = os.path.join(os.environ[
"VO_BELLE2_SW_DIR"],
"bkg")
30 truth =
"truth_positive"
34 """Database identifier of the filter being trained"""
35 return "trackfindingcdc_RealisticTrackRelationFilter.xml"
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")
43 path.add_module(
'TFCDC_ClusterPreparer',
45 SuperClusterExpandOverApogeeGap=
True)
47 path.add_module(
"TFCDC_SegmentFinderFacetAutomaton")
57 elif self.
task ==
"eval":
65 elif self.
task ==
"explore":
75 raise ValueError(
"Unknown task " + self.
task)
77 path.add_module(
"TFCDC_TrackFinderSegmentPairAutomaton",
78 TrackRelationFilter=
"unionrecording",
79 TrackRelationFilterParameters={
80 "rootFileName": self.sample_file_name,
90 run.configure_and_execute_from_commandline()
93if __name__ ==
"__main__":
95 logging.basicConfig(stream=sys.stdout, level=logging.INFO, format=
'%(levelname)s:%(message)s')
task
Process each event according to the user's desired task (train, eval, explore)