Inheritance diagram for CDCQEHarvestingValidationTask:

Public Member Functions
def	requires (self)

def	add_tracking_with_quality_estimation (self, path)

Static Public Attributes
b2luigi	training_target = b2luigi.Parameter()
	Feature/variable to use as truth label in the quality estimator MVA classifier.

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

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

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

Detailed Description

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

Definition at line 1581 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 CDC standalone tracking with CDC quality estimator to basf2 path.

Reimplemented from HarvestingValidationBaseTask.

Definition at line 1624 of file combined_quality_estimator_teacher.py.

    def add_tracking_with_quality_estimation(self, path):
        """
        Add modules for CDC standalone tracking with CDC quality estimator to basf2 path.
        """
        tracking.add_cdc_track_finding(
            path,
            output_reco_tracks="RecoTracks",
            add_mva_quality_indicator=True,
        )
        # change weightfile of quality estimator to the one produced by this training script
        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,
        )
        tracking.add_mc_matcher(path, components=["CDC"])
        tracking.add_track_fit_and_track_creator(path, components=["CDC"])
 
 

◆ requires()

def requires ( self )

Generate list of luigi Tasks that this Task depends on.

Reimplemented from HarvestingValidationBaseTask.

Definition at line 1596 of file combined_quality_estimator_teacher.py.

    def requires(self):
        """
        Generate list of luigi Tasks that this Task depends on.
        """
        yield self.teacher_task(
            n_events_training=self.n_events_training,
            experiment_number=self.experiment_number,
            process_type=self.process_type,
            training_target=self.training_target,
            exclude_variables=self.exclude_variables,
            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,
                )
 