plismbench.metrics package#

Module for metrics in plismbench.

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

Bases: BasePlismMetric

Cosine similarity metric.

compute_metric(matrix_a, matrix_b)[source]#: Compute cosine similarity metric.

class plismbench.metrics.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.

Submodules#