![]() |
Belle II Software
release-05-01-25
|
Public Member Functions | |
def | requires (self) |
def | weightfile_identifier_basename (self) |
def | get_weightfile_xml_identifier (self, fast_bdt_option=None, recotrack_option=None) |
def | tree_name (self) |
def | random_seed (self) |
Basf2PathTask | data_collection_task (self) |
def | output (self) |
def | process (self) |
Static Public Attributes | |
recotrack_option = b2luigi.Parameter(default='deleteCDCQI080') | |
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) | |
string | weightfile_identifier_basename = "recotrack_mva_qe" |
Name of the weightfile that is created. | |
string | 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. | |
string | random_seed = "train_rec" |
Random basf2 seed used to create the training data set. | |
data_collection_task = RecoTrackQEDataCollectionTask | |
Defines DataCollectionTask to require by tha base class to collect features for the MVA training. | |
cdc_training_target = b2luigi.Parameter() | |
Feature/variable to use as truth label for the CDC track quality estimator. | |
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. More... | |
process_type = b2luigi.Parameter(default="BBBAR") | |
Define which kind of process shall be used. More... | |
training_target = b2luigi.Parameter(default="truth") | |
Feature/variable to use as truth label in the quality estimator MVA classifier. | |
exclude_variables = b2luigi.ListParameter(hashed=True, default=[]) | |
List of collected variables to not use in the training of the QE MVA classifier. More... | |
fast_bdt_option = b2luigi.ListParameter(hashed=True, default=[200, 8, 3, 0.1]) | |
Hyperparameter option of the FastBDT algorithm. More... | |
Task to run basf2 mva teacher on collected data for the final, combined track quality estimator
Definition at line 1195 of file combined_quality_estimator_teacher.py.
|
inherited |
Property defining the specific ``DataCollectionTask`` to require. Must implemented by the inheriting specific teacher task class.
Definition at line 1087 of file combined_quality_estimator_teacher.py.
|
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 1051 of file combined_quality_estimator_teacher.py.
|
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 1116 of file combined_quality_estimator_teacher.py.
|
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 1123 of file combined_quality_estimator_teacher.py.
|
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 1078 of file combined_quality_estimator_teacher.py.
def requires | ( | self | ) |
Generate list of luigi Tasks that this Task depends on.
Reimplemented from TrackQETeacherBaseTask.
Definition at line 1218 of file combined_quality_estimator_teacher.py.
|
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 1069 of file combined_quality_estimator_teacher.py.
|
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 1042 of file combined_quality_estimator_teacher.py.
|
staticinherited |
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 1037 of file combined_quality_estimator_teacher.py.
|
staticinherited |
Experiment number of the conditions database, e.g.
defines simulation geometry
Definition at line 1028 of file combined_quality_estimator_teacher.py.
|
staticinherited |
Hyperparameter option of the FastBDT algorithm.
default are the FastBDT default values.
Definition at line 1039 of file combined_quality_estimator_teacher.py.
|
staticinherited |
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 1032 of file combined_quality_estimator_teacher.py.