icebrem

icebrem input data classes and functions.

icebrem.common

load_root_file(root_path, ids=None, entry_start=0, entry_stop=None, maxevents=None, args=None)

Loads the root files

Parameters:

root_path – path to root files

Returns:

jagged columnar data Y: class labels W: event weights ids: columnar variables string (list) info: trigger, MC xs, pre-selection acceptance x efficiency information (dict)

Return type:

X

process_root(X, args, ids=None, isMC=None, return_mask=False, class_id=None, **kwargs)

Apply selections

splitfactor(x, y, w, ids, args)

Transform data into different datatypes.

Parameters:

• data – jagged arrays

• args – arguments dictionary

Returns:

dictionary with different data representations