Skip to content

utils.math

Math functions for the AtomicEmbeddings package.

cosine_distance(a, b)

Cosine distance of two vectors.

Source code in src/elementembeddings/utils/math.py 
25
26
27
28
29
30
def cosine_distance(
    a: Iterable[int | float],
    b: Iterable[int | float],
) -> float:
    """Cosine distance of two vectors."""
    return 1 - cosine_similarity(a, b)

cosine_similarity(a, b)

Cosine similarity of two vectors.

Source code in src/elementembeddings/utils/math.py 
17
18
19
20
21
22
def cosine_similarity(
    a: Iterable[int | float],
    b: Iterable[int | float],
) -> float:
    """Cosine similarity of two vectors."""
    return dot(a, b) / ((dot(a, a) ** 0.5) * (dot(b, b) ** 0.5))

dot(a, b)

Dot product of two vectors.

Source code in src/elementembeddings/utils/math.py 
12
13
14
def dot(a: Iterable[int | float], b: Iterable[int | float]) -> float:
    """Dot product of two vectors."""
    return float(sum(map(operator.mul, a, b)))