plismbench.metrics.cosine_similarity module
Module for cosine similarity metric.
-
class plismbench.metrics.cosine_similarity.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.
