View source on GitHub |
Utility function to compute cosine similarity between two embedding.
mp.tasks.components.utils.cosine_similarity.cosine_similarity(
u: mp.tasks.components.containers.Embedding
,
v: mp.tasks.components.containers.Embedding
) -> float
May return an InvalidArgumentError if e.g. the feature vectors are of different types (quantized vs. float), have different sizes, or have an L2-norm of 0.
Args | |
---|---|
u
|
An embedding. |
v
|
An embedding. |
Returns | |
---|---|
Cosine similarity value. |