plismbench.metrics.retrieval module
Module for retrieval metrics.
-
class plismbench.metrics.retrieval.TopkAccuracy(device: str, use_mixed_precision: bool = True, k: list[int] | None = None)[source]
Bases: BasePlismMetric
Top-k accuracy.
-
compute_metric(matrix_a, matrix_b)[source]
Compute top-k accuracy metric.
