Inheritance diagram for RecoTrackQETeacherTask:

Collaboration diagram for RecoTrackQETeacherTask:

[legend]

Public Member Functions
def	requires (self)

Static Public Attributes
b2luigi	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)

str	weightfile_identifier_basename = "recotrack_mva_qe"
	Name of the weightfile that is created.

str	tree_name = "tree"
	Name of the TTree in the ROOT file from the `data_collection_task` that contains the training data for the MVA teacher.

str	random_seed = "train_rec"
	Random basf2 seed used to create the training data set.

RecoTrackQEDataCollectionTask	data_collection_task = RecoTrackQEDataCollectionTask
	Defines DataCollectionTask to require by the base class to collect features for the MVA training.

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

Detailed Description

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

Definition at line 1360 of file combined_quality_estimator_teacher.py.

Member Function Documentation

◆ requires()

def requires ( self )

Generate list of luigi Tasks that this Task depends on.

Reimplemented from TrackQETeacherBaseTask.

Definition at line 1387 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,
                )
 
 

Member Data Documentation

◆ cdc_training_target

b2luigi cdc_training_target = b2luigi.Parameter()

static

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

Definition at line 1385 of file combined_quality_estimator_teacher.py.

◆ data_collection_task

RecoTrackQEDataCollectionTask data_collection_task = RecoTrackQEDataCollectionTask

static

Defines DataCollectionTask to require by the base class to collect features for the MVA training.

Definition at line 1383 of file combined_quality_estimator_teacher.py.

◆ random_seed

str random_seed = "train_rec"

static

Random basf2 seed used to create the training data set.

Definition at line 1380 of file combined_quality_estimator_teacher.py.

◆ recotrack_option

b2luigi recotrack_option

static

Initial value:

=  b2luigi.Parameter(
        
    )

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