SplitNMergeSimTask Class Reference

Inheritance diagram for SplitNMergeSimTask:

Public Member Functions
	output_file_name (self, n_events=None, random_seed=None)
	Name of the ROOT output file with generated and simulated events.
	output (self)
	requires (self)
	process (self)

Static Public Attributes
	n_events = b2luigi.IntParameter()
	Number of events to generate.
	experiment_number = b2luigi.IntParameter()
	Experiment number of the conditions database, e.g.
	random_seed = b2luigi.Parameter()
	Random basf2 seed.
	bkgfiles_dir
	Directory with overlay background root files.
str	queue = 'l'
	specify queue.

Detailed Description

Generate simulated Monte Carlo with background overlay.

Make sure to use different ``random_seed`` parameters for the training data
format the classifier trainings and for the test data for the respective
evaluation/validation tasks.

Definition at line 558 of file combined_quality_estimator_teacher.py.

Member Function Documentation

output()

output

(

self

)

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

    def output(self):
        """
        Generate list of output files that the task should produce.
        The task is considered finished if and only if the outputs all exist.
        """
        yield self.add_to_output(self.output_file_name())
 

output_file_name()

output_file_name	(		self,
			n_events = None,
			random_seed = None )

Name of the ROOT output file with generated and simulated events.

Create output file name depending on number of events and production
mode that is specified in the random_seed string.

Definition at line 584 of file combined_quality_estimator_teacher.py.

    def output_file_name(self, n_events=None, random_seed=None):
        """
        Create output file name depending on number of events and production
        mode that is specified in the random_seed string.
        """
        if n_events is None:
            n_events = self.n_events
        if random_seed is None:
            random_seed = self.random_seed
        return "generated_mc_N" + str(n_events) + "_" + random_seed + ".root"
 

process()

process

(

self

)

When all GenerateSimTasks finished, merge the output.

Definition at line 626 of file combined_quality_estimator_teacher.py.

    def process(self):
        """
        When all GenerateSimTasks finished, merge the output.
        """
        file_list = self.get_all_input_file_names()
        file_list = flat(file_list)
        print("Merge the following files:\n")
        print(file_list)
        cmd = ["b2file-merge", "-f"]
        args = cmd + [self.get_output_file_name(self.output_file_name())] + file_list
        print(f"args to merge: {args}")
        subprocess.check_call(args)
        print("Finished merging. Removing the input files to save space.")
        cmd2 = ["rm", "-f"]
        args = cmd2 + file_list
        print(f"args for deleting: {args}")
        subprocess.check_call(args)
 
 

requires()

requires

(

self

)

Generate list of luigi Tasks that this Task depends on.

Definition at line 602 of file combined_quality_estimator_teacher.py.

    def requires(self):
        """
        Generate list of luigi Tasks that this Task depends on.
        """
        n_events_per_task = MasterTask.n_events_per_task
        quotient, remainder = divmod(self.n_events, n_events_per_task)
        for i in range(quotient):
            yield GenerateSimTask(
                bkgfiles_dir=self.bkgfiles_dir,
                num_processes=MasterTask.num_processes,
                random_seed=self.random_seed + '_' + str(i).zfill(3),
                n_events=n_events_per_task,
                experiment_number=self.experiment_number,
                )
        if remainder > 0:
            yield GenerateSimTask(
                bkgfiles_dir=self.bkgfiles_dir,
                num_processes=MasterTask.num_processes,
                random_seed=self.random_seed + '_' + str(quotient).zfill(3),
                n_events=remainder,
                experiment_number=self.experiment_number,
                )
 

Member Data Documentation

bkgfiles_dir

bkgfiles_dir

static

Initial value:

=  b2luigi.Parameter(
        
 
 
    )

Directory with overlay background root files.

Definition at line 575 of file combined_quality_estimator_teacher.py.

experiment_number

experiment_number = b2luigi.IntParameter()

static

Experiment number of the conditions database, e.g.

defines simulation geometry

Definition at line 570 of file combined_quality_estimator_teacher.py.

n_events

n_events = b2luigi.IntParameter()

static

Number of events to generate.

Definition at line 568 of file combined_quality_estimator_teacher.py.

queue

str queue = 'l'

static

specify queue.

E.g. choose between 'l' (long), 's' (short) or 'sx' (short, extra ram)

Definition at line 581 of file combined_quality_estimator_teacher.py.

random_seed

random_seed = b2luigi.Parameter()

static

Random basf2 seed.

It is further used to read of the production process to preserve clearness in the b2luigi output.

Definition at line 573 of file combined_quality_estimator_teacher.py.

---

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

• tracking/scripts/tracking/train/combined_quality_estimator_teacher.py