Inheritance diagram for RecoTrackQETeacherTask:

Public Member Functions
	requires (self)

	make_db (self)

	weightfile_identifier_basename (self)

	get_weightfile_identifier (self, fast_bdt_option=None, recotrack_option=None)

	tree_name (self)

	random_seed (self)

Basf2PathTask	data_collection_task (self)

	output (self)

	process (self)

Public Attributes
	recotrack_option
	RecoTrack option, use string that is additive: deleteCDCQI0XY (= deletes CDCTracks with CDC-QI below 0.XY), useCDC (= uses trained CDC stored in datafiles/), useVXD (uses trained VXD stored in datafiles/), noVXD (= doesn't use the VXD MVA at all)

Static Public Attributes
	cdc_training_target = b2luigi.Parameter()
	Feature/variable to use as truth label for the CDC track quality estimator.

str	object_name = 'TrackQualityEstimatorMVA'
	DBObject name.

	n_events_training = b2luigi.IntParameter()
	Number of events to generate for the training data set.

	experiment_number = b2luigi.IntParameter()
	Experiment number of the conditions database, e.g.

	process_type
	Define which kind of process shall be used.

	training_target
	Feature/variable to use as truth label in the quality estimator MVA classifier.

	exclude_variables
	List of collected variables to not use in the training of the QE MVA classifier.

	fast_bdt_option
	Hyperparameter option of the FastBDT algorithm.

Detailed Description

Task to run basf2 mva teacher on collected data for the final, combined
track quality estimator

Definition at line 1445 of file combined_quality_estimator_teacher.py.

Member Function Documentation

◆ data_collection_task()

Basf2PathTask data_collection_task ( self )

inherited

Property defining the specific ``DataCollectionTask`` to require.  Must
implemented by the inheriting specific teacher task class.

Definition at line 1293 of file combined_quality_estimator_teacher.py.

    def data_collection_task(self) -> Basf2PathTask:
        """
        Property defining the specific ``DataCollectionTask`` to require.  Must
        implemented by the inheriting specific teacher task class.
        """
        raise NotImplementedError(
            "Teacher Task must define a data collection task to require "
        )
 

◆ get_weightfile_identifier()

get_weightfile_identifier	(	self,
		fast_bdt_option = None,
		recotrack_option = None )

inherited

Name of the xml weightfile that is created by the teacher task.
It is subsequently used as a local weightfile in the following validation tasks.

Definition at line 1251 of file combined_quality_estimator_teacher.py.

    def get_weightfile_identifier(self, fast_bdt_option=None, recotrack_option=None):
        """
        Name of the xml weightfile that is created by the teacher task.
        It is subsequently used as a local weightfile in the following validation tasks.
        """
        if fast_bdt_option is None:
            fast_bdt_option = self.fast_bdt_option
        if recotrack_option is None and hasattr(self, 'recotrack_option'):
            
            if isinstance(self.recotrack_option, str):
                recotrack_option = self.recotrack_option
            else:
                recotrack_option = self.recotrack_option._default
        else:
            recotrack_option = ''
        weightfile_details = create_fbdt_option_string(fast_bdt_option)
        weightfile_name = self.weightfile_identifier_basename + weightfile_details
        if recotrack_option != '':
            weightfile_name = weightfile_name + '_' + recotrack_option
        return weightfile_name + "_weights"
 

◆ make_db()

make_db ( self )

Creates the local Reco payload from weightfiles.

Definition at line 1497 of file combined_quality_estimator_teacher.py.

    def make_db(self):
        """
        Creates the local Reco payload from weightfiles.
        """
        recotrack_identifier = self.get_output_file_name(self.get_weightfile_identifier() + '.xml')
        with open(recotrack_identifier, 'r') as f:
            weight_file_content = f.read()
        recotrack_name = write_mva_weightfile_content_to_db(
            dbobj_name=self.object_name,
            content=weight_file_content,
            iovList=(0, 0, 0, -1)
        )
        return recotrack_name
 
 

◆ output()

output ( self )

inherited

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 1322 of file combined_quality_estimator_teacher.py.

    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.
        """
        yield self.add_to_output(self.get_weightfile_identifier() + '.xml')
 

◆ process()

process ( self )

inherited

Use basf2_mva teacher to create MVA weightfile from collected training
data variables.

This is the main process that is dispatched by the ``run`` method that
is inherited from ``Basf2Task``.

Definition at line 1329 of file combined_quality_estimator_teacher.py.

    def process(self):
        """
        Use basf2_mva teacher to create MVA weightfile from collected training
        data variables.
 
        This is the main process that is dispatched by the ``run`` method that
        is inherited from ``Basf2Task``.
        """
        if 'USEREC' in self.process_type:
            if 'USERECBB' in self.process_type:
                process = 'BBBAR'
            elif 'USERECEE' in self.process_type:
                process = 'BHABHA'
            records_files = ['datafiles/qe_records_N' + str(self.n_events_training) +
                             '_' + process + '_' + self.random_seed + '.root']
        else:
            if hasattr(self, 'recotrack_option') and isinstance(self, RecoTrackQETeacherTask):
                records_files = self.get_input_file_names(
                    self.data_collection_task.get_records_file_name(
                        self.data_collection_task,
                        n_events=self.n_events_training,
                        random_seed=self.process_type + '_' + self.random_seed,
                        recotrack_option=self.recotrack_option))
            else:
                records_files = self.get_input_file_names(
                    self.data_collection_task.get_records_file_name(
                        self.data_collection_task,
                        n_events=self.n_events_training,
                        random_seed=self.process_type + '_' + self.random_seed))
 
        weightfile_identifier = self.get_output_file_name(self.get_weightfile_identifier() + '.xml')
        print('The weightfile used is:', weightfile_identifier)
        my_basf2_mva_teacher(
            records_files=records_files,
            tree_name=self.tree_name,
            weightfile_identifier=weightfile_identifier,
            target_variable=self.training_target,
            exclude_variables=self.exclude_variables,
            fast_bdt_option=self.fast_bdt_option,
        )
        _ = self.make_db()
 
 

◆ random_seed()

random_seed ( self )

inherited

Property defining random seed to be used by the ``GenerateSimTask``.
Should differ from the random seed in the test data samples.  Must
implemented by the inheriting specific teacher task class.

Definition at line 1284 of file combined_quality_estimator_teacher.py.

    def random_seed(self):
        """
        Property defining random seed to be used by the ``GenerateSimTask``.
        Should differ from the random seed in the test data samples.  Must
        implemented by the inheriting specific teacher task class.
        """
        raise NotImplementedError("Teacher Task must define a static random seed")
 

◆ requires()

requires ( self )

Generate list of luigi Tasks that this Task depends on.

Reimplemented from TrackQETeacherBaseTask.

Definition at line 1474 of file combined_quality_estimator_teacher.py.

    def requires(self):
        """
        Generate list of luigi Tasks that this Task depends on.
        """
        if 'USEREC' in self.process_type:
            if 'USERECBB' in self.process_type:
                process = 'BBBAR'
            elif 'USERECEE' in self.process_type:
                process = 'BHABHA'
            yield CheckExistingFile(
                filename='datafiles/qe_records_N' + str(self.n_events_training) + '_' + process + '_' + self.random_seed + '.root',
                )
        else:
            yield self.data_collection_task(
                cdc_training_target=self.cdc_training_target,
                num_processes=MasterTask.num_processes,
                n_events=self.n_events_training,
                experiment_number=self.experiment_number,
                random_seed=self.process_type + '_' + self.random_seed,
                recotrack_option=self.recotrack_option,
                fast_bdt_option=self.fast_bdt_option,
                )
 

◆ tree_name()

tree_name ( self )

inherited

Property defining the name of the tree in the ROOT file from the
``data_collection_task`` that contains the recorded training data.  Must
implemented by the inheriting specific teacher task class.

Definition at line 1275 of file combined_quality_estimator_teacher.py.

    def tree_name(self):
        """
        Property defining the name of the tree in the ROOT file from the
        ``data_collection_task`` that contains the recorded training data.  Must
        implemented by the inheriting specific teacher task class.
        """
        raise NotImplementedError("Teacher Task must define a static tree_name")
 

◆ weightfile_identifier_basename()

weightfile_identifier_basename ( self )

inherited

Property defining the basename for the .xml and .root weightfiles that are created.
Has to be implemented by the inheriting teacher task class.

Definition at line 1242 of file combined_quality_estimator_teacher.py.

    def weightfile_identifier_basename(self):
        """
        Property defining the basename for the .xml and .root weightfiles that are created.
        Has to be implemented by the inheriting teacher task class.
        """
        raise NotImplementedError(
            "Teacher Task must define a static weightfile_identifier"
        )
 

Member Data Documentation

◆ cdc_training_target

cdc_training_target = b2luigi.Parameter()

static

Feature/variable to use as truth label for the CDC track quality estimator.

Definition at line 1470 of file combined_quality_estimator_teacher.py.

◆ exclude_variables

exclude_variables

staticinherited

Initial value:

=  b2luigi.ListParameter(
        
    )

List of collected variables to not use in the training of the QE MVA classifier.

In addition to variables containing the "truth" substring, which are excluded by default.

Definition at line 1229 of file combined_quality_estimator_teacher.py.

◆ experiment_number

experiment_number = b2luigi.IntParameter()

staticinherited

Experiment number of the conditions database, e.g.

defines simulation geometry

Definition at line 1212 of file combined_quality_estimator_teacher.py.

◆ fast_bdt_option

fast_bdt_option

staticinherited

Initial value:

=  b2luigi.ListParameter(
        
    )

Hyperparameter option of the FastBDT algorithm.

default are the FastBDT default values.

Definition at line 1235 of file combined_quality_estimator_teacher.py.

◆ n_events_training

n_events_training = b2luigi.IntParameter()

staticinherited

Number of events to generate for the training data set.

Definition at line 1210 of file combined_quality_estimator_teacher.py.

◆ object_name

str object_name = 'TrackQualityEstimatorMVA'

static

DBObject name.

Definition at line 1472 of file combined_quality_estimator_teacher.py.

◆ process_type

process_type

staticinherited

Initial value:

=  b2luigi.Parameter(
        
    )

Define which kind of process shall be used.

Decide between simulating BBBAR or BHABHA, MUMU, YY, DDBAR, UUBAR, SSBAR, CCBAR, reconstructing DATA or already simulated files (USESIMBB/EE) or running on existing reconstructed files (USERECBB/EE)

Definition at line 1216 of file combined_quality_estimator_teacher.py.

◆ recotrack_option

recotrack_option

inherited

RecoTrack option, use string that is additive: deleteCDCQI0XY (= deletes CDCTracks with CDC-QI below 0.XY), useCDC (= uses trained CDC stored in datafiles/), useVXD (uses trained VXD stored in datafiles/), noVXD (= doesn't use the VXD MVA at all)

Definition at line 1262 of file combined_quality_estimator_teacher.py.

◆ training_target

training_target

staticinherited

Initial value:

=  b2luigi.Parameter(
        
    )

Feature/variable to use as truth label in the quality estimator MVA classifier.

Definition at line 1222 of file combined_quality_estimator_teacher.py.

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

tracking/scripts/tracking/train/combined_quality_estimator_teacher.py

Public Member Functions

Public Attributes

Static Public Attributes

Detailed Description

Member Function Documentation

◆ data_collection_task()

◆ get_weightfile_identifier()

◆ make_db()

◆ output()

◆ process()

◆ random_seed()

◆ requires()

◆ tree_name()

◆ weightfile_identifier_basename()

Member Data Documentation

◆ cdc_training_target

◆ exclude_variables

◆ experiment_number

◆ fast_bdt_option

◆ n_events_training

◆ object_name

◆ process_type

◆ recotrack_option

◆ training_target