Inheritance diagram for RecoTrackQEHarvestingValidationTask:

Public Member Functions
def	requires (self)

def	add_tracking_with_quality_estimation (self, path)

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

str	validation_output_file_name = "reco_qe_harvesting_validation.root"
	Name of the "harvested" ROOT output file with variables that can be used for validation.

str	reco_output_file_name = "reco_qe_reconstruction.root"
	Name of the output of the RootOutput module with reconstructed events.

RecoTrackQETeacherTask	teacher_task = RecoTrackQETeacherTask
	Teacher task to require to provide a quality estimator weightfile for `add_tracking_with_quality_estimation`

Detailed Description

Run track reconstruction and write out (="harvest") a root file with variables
useful for validation of the MVA track Quality Estimator.

Definition at line 1648 of file combined_quality_estimator_teacher.py.

Member Function Documentation

◆ add_tracking_with_quality_estimation()

def add_tracking_with_quality_estimation	(	self,
		path
	)

Add modules for reco tracking with all track quality estimators to basf2 path.

Reimplemented from HarvestingValidationBaseTask.

Definition at line 1706 of file combined_quality_estimator_teacher.py.

    def add_tracking_with_quality_estimation(self, path):
        """
        Add modules for reco tracking with all track quality estimators to basf2 path.
        """
 
        # add tracking reconstruction with quality estimator modules added
        tracking.add_tracking_reconstruction(
            path,
            add_cdcTrack_QI=True,
            add_vxdTrack_QI=True,
            add_recoTrack_QI=True,
            skipGeometryAdding=True,
            skipHitPreparerAdding=False,
        )
 
        # Replace the weightfiles of all quality estimator modules by those
        # produced in the training by b2luigi
        cdc_qe_mva_filter_parameters = {
            "identifier": self.get_input_file_names(
                CDCQETeacherTask.get_weightfile_xml_identifier(
                    CDCQETeacherTask,
                    fast_bdt_option=self.fast_bdt_option))[0]}
        basf2.set_module_parameters(
            path,
            name="TFCDC_TrackQualityEstimator",
            filterParameters=cdc_qe_mva_filter_parameters,
        )
        basf2.set_module_parameters(
            path,
            name="VXDQualityEstimatorMVA",
            WeightFileIdentifier=self.get_input_file_names(
                VXDQETeacherTask.get_weightfile_xml_identifier(VXDQETeacherTask, fast_bdt_option=self.fast_bdt_option)
            )[0],
        )
        basf2.set_module_parameters(
            path,
            name="TrackQualityEstimatorMVA",
            WeightFileIdentifier=self.get_input_file_names(
                RecoTrackQETeacherTask.get_weightfile_xml_identifier(RecoTrackQETeacherTask, fast_bdt_option=self.fast_bdt_option)
            )[0],
        )
 
 

◆ requires()

def requires ( self )

Generate list of luigi Tasks that this Task depends on.

Reimplemented from HarvestingValidationBaseTask.

Definition at line 1662 of file combined_quality_estimator_teacher.py.

    def requires(self):
        """
        Generate list of luigi Tasks that this Task depends on.
        """
        yield CDCQETeacherTask(
            n_events_training=self.n_events_training,
            experiment_number=self.experiment_number,
            process_type=self.process_type,
            training_target=self.cdc_training_target,
            exclude_variables=MasterTask.exclude_variables_cdc,
            fast_bdt_option=self.fast_bdt_option,
            )
        yield VXDQETeacherTask(
            n_events_training=self.n_events_training,
            experiment_number=self.experiment_number,
            process_type=self.process_type,
            exclude_variables=MasterTask.exclude_variables_vxd,
            fast_bdt_option=self.fast_bdt_option,
            )
 
        yield self.teacher_task(
            n_events_training=self.n_events_training,
            experiment_number=self.experiment_number,
            process_type=self.process_type,
            exclude_variables=self.exclude_variables,
            cdc_training_target=self.cdc_training_target,
            fast_bdt_option=self.fast_bdt_option,
            )
        if 'USE' in self.process_type:  # USESIM and USEREC
            if 'BB' in self.process_type:
                process = 'BBBAR'
            elif 'EE' in self.process_type:
                process = 'BHABHA'
            yield CheckExistingFile(
                filename='datafiles/generated_mc_N' + str(self.n_events_testing) + '_' + process + '_test.root'
                )
        else:
            yield SplitNMergeSimTask(
                bkgfiles_dir=MasterTask.bkgfiles_by_exp[self.experiment_number],
                random_seed=self.process_type + '_test',
                n_events=self.n_events_testing,
                experiment_number=self.experiment_number,
                )
 