SummaryTask Class Reference

Inheritance diagram for SummaryTask:

Public Member Functions
	output (self)
	requires (self)
	run (self)
	remove_output (self)

Static Public Attributes
	n_events_training
	Number of events to generate for the training data set.
	n_events_testing
	Number of events to generate for the test data set.
	n_events_per_task
	Number of events per task.
	num_processes
	Number of basf2 processes to use in Basf2PathTasks.
	bkgfiles_by_exp = b2luigi.get_setting("bkgfiles_by_exp")
	Dictionary with experiment numbers as keys and background directory paths as values.
dict	bkgfiles_by_exp = {int(key): val for (key, val) in bkgfiles_by_exp.items()}
	Transform dictionary keys (exp.
str	batch_system = 'local'
	Use local resources.
str	output_file_name = 'summary.json'
	Output file name.

Detailed Description

Task that collects and summarizes the main figure-of-merits from all the
(validation and optimisation) child taks.

Definition at line 1104 of file combined_to_pxd_ckf_mva_training.py.

Member Function Documentation

output()

output

(

self

)

Output method.

Definition at line 1144 of file combined_to_pxd_ckf_mva_training.py.

    def output(self):
        """
        Output method.
        """
        yield self.add_to_output(self.output_file_name)
 

remove_output()

remove_output

(

self

)

Default function from base b2luigi.Task class.

Definition at line 1222 of file combined_to_pxd_ckf_mva_training.py.

    def remove_output(self):
        """
        Default function from base b2luigi.Task class.
        """
        self._remove_output()
 
 

requires()

requires

(

self

)

Generate list of tasks that needs to be done for luigi to finish running
this steering file.

Definition at line 1150 of file combined_to_pxd_ckf_mva_training.py.

    def requires(self):
        """
        Generate list of tasks that needs to be done for luigi to finish running
        this steering file.
        """
 
        fast_bdt_options = [
            [50, 8, 3, 0.1],
            [100, 8, 3, 0.1],
            [200, 8, 3, 0.1],
        ]
 
        experiment_numbers = b2luigi.get_setting("experiment_numbers")
 
        # iterate over all possible combinations of parameters from the above defined parameter lists
        for experiment_number, fast_bdt_option_state_filter, fast_bdt_option_result_filter in itertools.product(
                experiment_numbers, fast_bdt_options, fast_bdt_options
        ):
 
            state_filter_cuts = [0.01, 0.02, 0.03, 0.05, 0.1, 0.2]
            n_best_states_list = [3, 5, 10]
            result_filter_cuts = [0.05, 0.1, 0.2]
            n_best_results_list = [2, 3, 5]
            for state_filter_cut, n_best_states, result_filter_cut, n_best_results in \
                    itertools.product(state_filter_cuts, n_best_states_list, result_filter_cuts, n_best_results_list):
                yield self.clone(
                    ValidationAndOptimisationTask,
                    experiment_number=experiment_number,
                    n_events_training=self.n_events_training,
                    n_events_testing=self.n_events_testing,
                    state_filter_cut=state_filter_cut,
                    use_n_best_states=n_best_states,
                    result_filter_cut=result_filter_cut,
                    use_n_best_results=n_best_results,
                    fast_bdt_option_state_filter=fast_bdt_option_state_filter,
                    fast_bdt_option_result_filter=fast_bdt_option_result_filter,
                )
 

run()

run

(

self

)

Run method.

Definition at line 1188 of file combined_to_pxd_ckf_mva_training.py.

    def run(self):
        """
        Run method.
        """
        import ROOT  # noqa
 
        # These are the "TNtuple" names to check for
        ntuple_names = (
            'MCSideTrackingValidationModule_overview_figures_of_merit',
            'PRSideTrackingValidationModule_overview_figures_of_merit',
            'PRSideTrackingValidationModule_subdetector_figures_of_merit'
        )
 
        # Collect the information in a dictionary...
        output_dict = {}
        all_files = self.get_all_input_file_names()
        for idx, single_file in enumerate(all_files):
            with ROOT.TFile.Open(single_file, 'READ') as f:
                branch_data = {}
                for ntuple_name in ntuple_names:
                    ntuple = f.Get(ntuple_name)
                    for i in range(min(1, ntuple.GetEntries())):  # Here we expect only 1 entry
                        ntuple.GetEntry(i)
                        for branch in ntuple.GetListOfBranches():
                            name = branch.GetName()
                            value = getattr(ntuple, name)
                            branch_data[name] = value
                branch_data['file_path'] = single_file
                output_dict[f'{idx}'] = branch_data
 
        # ... and store the information in a JSON file
        with open(self.get_output_file_name(self.output_file_name), 'w') as f:
            json.dump(output_dict, f, indent=4)
 

Member Data Documentation

batch_system

str batch_system = 'local'

static

Use local resources.

Definition at line 1140 of file combined_to_pxd_ckf_mva_training.py.

bkgfiles_by_exp [1/2]

bkgfiles_by_exp = b2luigi.get_setting("bkgfiles_by_exp")

static

Dictionary with experiment numbers as keys and background directory paths as values.

Definition at line 1135 of file combined_to_pxd_ckf_mva_training.py.

bkgfiles_by_exp [2/2]

dict bkgfiles_by_exp = {int(key): val for (key, val) in bkgfiles_by_exp.items()}

static

Transform dictionary keys (exp.

numbers) from strings to int

Definition at line 1137 of file combined_to_pxd_ckf_mva_training.py.

n_events_per_task

n_events_per_task

static

Initial value:

=  b2luigi.get_setting(
        
 
 
    )

Number of events per task.

Used to split up the simulation tasks.

Definition at line 1122 of file combined_to_pxd_ckf_mva_training.py.

n_events_testing

n_events_testing

static

Initial value:

=  b2luigi.get_setting(
        
 
 
    )

Number of events to generate for the test data set.

Definition at line 1116 of file combined_to_pxd_ckf_mva_training.py.

n_events_training

n_events_training

static

Initial value:

=  b2luigi.get_setting(
        
 
 
    )

Number of events to generate for the training data set.

Definition at line 1110 of file combined_to_pxd_ckf_mva_training.py.

num_processes

num_processes

static

Initial value:

=  b2luigi.get_setting(
        
 
 
    )

Number of basf2 processes to use in Basf2PathTasks.

Definition at line 1128 of file combined_to_pxd_ckf_mva_training.py.

output_file_name

output_file_name = 'summary.json'

static

Output file name.

Definition at line 1142 of file combined_to_pxd_ckf_mva_training.py.

---

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

• tracking/scripts/ckf_training/to_pxd/combined_to_pxd_ckf_mva_training.py