pyprobound.table.score

score(module, batch, fun='forward', max_split=None, **kwargs)

Scores a batch using a chosen function, automatically managing devices.

Parameters:

• module (Transform) – The Transform used for scoring.

• batch (CountBatch) – The CountBatch containing the sequences and counts to be scored.

• fun (str) – The name of the function taken from the module for scoring.

• max_split (int | None) – Maximum number of sequences scored at a time.

• kwargs (Any) – Any keyword arguments passed to the function.

Return type:

tuple[Tensor, Tensor]

Returns:

A tuple of the observed counts and predicted scores, both on CPU.