Skip to content

Math

Math functions for the AtomicEmbeddings package.

cosine_distance(a, b)

Cosine distance of two vectors.

Source code in src/elementembeddings/utils/math.py 
20
21
22
23
24
25
def cosine_distance(
    a: List[Union[int, float]],
    b: List[Union[int, float]],
) -> Union[int, 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 
12
13
14
15
16
17
def cosine_similarity(
    a: List[Union[int, float]],
    b: List[Union[int, float]],
) -> Union[int, 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 
7
8
9
def dot(a: List[Union[int, float]], b: List[Union[int, float]]) -> Union[int, float]:
    """Dot product of two vectors."""
    return sum(map(operator.mul, a, b))