Belle II Software  release-05-02-19
CDCQEDataCollectionTask Class Reference
Inheritance diagram for CDCQEDataCollectionTask:
Collaboration diagram for CDCQEDataCollectionTask:

Public Member Functions

def get_records_file_name (self, n_events=None, random_seed=None)
 Filename of the recorded/collected data for the final QE MVA training.
 
def get_input_files (self, n_events=None, random_seed=None)
 
def requires (self)
 
def output (self)
 
def create_path (self)
 

Static Public Attributes

 n_events = b2luigi.IntParameter()
 Number of events to generate.
 
 experiment_number = b2luigi.IntParameter()
 Experiment number of the conditions database, e.g. More...
 
 random_seed = b2luigi.Parameter()
 Random basf2 seed used by the GenerateSimTask. More...
 
string queue = 'l'
 

Detailed Description

Collect variables/features from CDC tracking and write them to a ROOT file.

These variables are to be used as labelled training data for the MVA
classifier which is the CDC track quality estimator

Definition at line 686 of file combined_quality_estimator_teacher.py.

Member Function Documentation

◆ create_path()

def create_path (   self)
Create basf2 path with CDC standalone tracking and CDC QE with recording filter for MVA feature collection.

Definition at line 761 of file combined_quality_estimator_teacher.py.

761  def create_path(self):
762  """
763  Create basf2 path with CDC standalone tracking and CDC QE with recording filter for MVA feature collection.
764  """
765  path = basf2.create_path()
766  inputFileNames = self.get_input_files()
767  path.add_module(
768  "RootInput",
769  inputFileNames=inputFileNames,
770  )
771  path.add_module("Gearbox")
772  tracking.add_geometry_modules(path)
773  if 'DATA' in self.random_seed:
774  filter_choice = "recording_data"
775  from rawdata import add_unpackers
776  add_unpackers(path, components=['CDC'])
777  else:
778  filter_choice = "recording"
779  # tracking.add_hit_preparation_modules(path) # only needed for SVD and
780  # PXD hit preparation. Does not change the CDC output.
781  tracking.add_cdc_track_finding(path, with_ca=False, add_mva_quality_indicator=True)
782 
783  basf2.set_module_parameters(
784  path,
785  name="TFCDC_TrackQualityEstimator",
786  filter=filter_choice,
787  filterParameters={
788  "rootFileName": self.get_output_file_name(self.get_records_file_name())
789  },
790  )
791  return path
792 
793 

◆ output()

def output (   self)
Generate list of output files that the task should produce.
The task is considered finished if and only if the outputs all exist.

Definition at line 754 of file combined_quality_estimator_teacher.py.

◆ requires()

def requires (   self)
Generate list of luigi Tasks that this Task depends on.

Definition at line 737 of file combined_quality_estimator_teacher.py.

Member Data Documentation

◆ experiment_number

experiment_number = b2luigi.IntParameter()
static

Experiment number of the conditions database, e.g.

defines simulation geometry

Definition at line 696 of file combined_quality_estimator_teacher.py.

◆ random_seed

random_seed = b2luigi.Parameter()
static

Random basf2 seed used by the GenerateSimTask.

It is further used to read of the production process to preserve clearness in the b2luigi output.

Definition at line 699 of file combined_quality_estimator_teacher.py.


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