inference_core 0.1.0

Feakin is a architecture design and visual collaboration tool. This is the parser for Feakin.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

use crate::embedding::Embedding;
use crate::similarity::Similarity;

pub struct CosineSimilarity;

impl Similarity for CosineSimilarity {
    fn similarity_score(&self, vector_a: &Embedding, vector_b: &Embedding) -> f32 {
        if vector_a.len() != vector_b.len() {
            panic!(
                "Length of vector a ({}) must be equal to the length of vector b ({})",
                vector_a.len(),
                vector_b.len()
            );
        }

        let dot_product: f32 = vector_a
            .iter()
            .zip(vector_b.iter())  // Use vector_b.iter() directly
            .map(|(a, b)| a * b)
            .sum();

        let norm_a: f32 = vector_a.iter().map(|x| x * x).sum();
        let norm_b: f32 = vector_b.iter().map(|x| x * x).sum();

        dot_product / (f32::sqrt(norm_a) * f32::sqrt(norm_b))
    }
}

impl CosineSimilarity {
    pub fn between(embedding: &Embedding, reference_embedding: &Embedding) -> f32 {
        CosineSimilarity.similarity_score(embedding, reference_embedding)
    }
}