plismbench.metrics.base module#

Module for base metric object.

class plismbench.metrics.base.BasePlismMetric(device: str, use_mixed_precision: bool = True)[source]#

Bases: object

Base class for metrics.

device#

Device to use for computation.

Type:

str: Literal[“cpu”, “gpu”]

abstract compute_metric(matrix_a: ndarray, matrix_b: ndarray)[source]#

Compute metric between feature matrices A and B.