smoofit.model.MergedProcess

class smoofit.model.MergedProcess(processes, channels)[source]

Methods

__init__(processes, channels)

batch_pred(values)

Compute the yields of the processes using a batch of parameter values.

pred(values)

Compute the yields of the processes given the parameter values.

__init__(processes, channels)[source]
batch_pred(values)[source]

Compute the yields of the processes using a batch of parameter values.

Note

This method does not include the effect of the BB-lite parameters! use Model.pred() for that.

Parameters

values (DeviceArrayBase) – 2D array where the first axis is the batch dimension, i.e. every row specifies parameter values in the order expeced by the compiled model

Return type

Tuple[DeviceArrayBase, DeviceArrayBase]

Returns

a tuple of two 3D jnp.DeviceArray where the entries (i,j,:) in the first (second) array contain the predicted yields (sumw2), across all channels, of the process with index j given the parameter values in row i of values

pred(values)[source]

Compute the yields of the processes given the parameter values.

Note

This method does not include the effect of the BB-lite parameters! use Model.pred() for that.

Parameters

values (DeviceArrayBase) – 1D array with the parameter values in the order expected by the compiled model, e.g. as returned by Model.values_from_dict()

Return type

Tuple[DeviceArrayBase, DeviceArrayBase]

Returns

a tuple of two 2D jnp.DeviceArray where the entries (j,:) in the first (second) array contain the predicted yields (sumw2), across all channels, of the process with index j