Belle II Software development
RecoTrackQETeacherTask Class Reference
Inheritance diagram for RecoTrackQETeacherTask:
TrackQETeacherBaseTask

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.

1387 def requires(self):
1388 """
1389 Generate list of luigi Tasks that this Task depends on.
1390 """
1391 if 'USEREC' in self.process_type:
1392 if 'USERECBB' in self.process_type:
1393 process = 'BBBAR'
1394 elif 'USERECEE' in self.process_type:
1395 process = 'BHABHA'
1396 yield CheckExistingFile(
1397 filename='datafiles/qe_records_N' + str(self.n_events_training) + '_' + process + '_' + self.random_seed + '.root',
1398 )
1399 else:
1400 yield self.data_collection_task(
1401 cdc_training_target=self.cdc_training_target,
1402 num_processes=MasterTask.num_processes,
1403 n_events=self.n_events_training,
1404 experiment_number=self.experiment_number,
1405 random_seed=self.process_type + '_' + self.random_seed,
1406 recotrack_option=self.recotrack_option,
1407 fast_bdt_option=self.fast_bdt_option,
1408 )
1409
1410

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

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.

◆ tree_name

str tree_name = "tree"
static

Name of the TTree in the ROOT file from the data_collection_task that contains the training data for the MVA teacher.

Definition at line 1378 of file combined_quality_estimator_teacher.py.

◆ weightfile_identifier_basename

str weightfile_identifier_basename = "recotrack_mva_qe"
static

Name of the weightfile that is created.

Definition at line 1375 of file combined_quality_estimator_teacher.py.


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