iceid

Electron ID input data classes and functions.

iceid.common

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

Loads the root files.

Parameters:

root_path – paths to root files (list)

Returns:

columnar data Y: class labels W: event weights ids: columnar variable string (list) info: trigger and pre-selection acceptance x efficiency information (dict)

Return type:

X

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

Transform data into different datatypes.

Parameters:

• data – jagged arrays

• args – arguments dictionary

Returns:

dictionary with different data representations

iceid.graphio

get_node_features(p4vec, p4track, X, ids, num_nodes, num_node_features, coord)

parse_graph_data(X, ids, features, graph_param, Y=None, weights=None, entry_start=None, entry_stop=None, EPS=1e-12, null_value=-999.0)

Jagged array data into pytorch-geometric style Data format array.

Parameters:

• X – Jagged array of variables

• ids – Variable names as an array of strings

• features – List of active global feature strings

• graph_param – Graph construction parameters dict

• Y – Target class array (if any, typically MC only)

• weights – (Re-)weighting array (if any, typically MC only)

Returns:

List of pytorch-geometric Data objects

parse_tensor_data(X, ids, image_vars, args)

Parameters:

• X – Jagged array of variables

• ids – Variable names as an array of strings

• args – Arguments

Returns:

Tensor of pytorch-geometric Data objects