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.