victor-db 0.1.1

A browser-optimized vector database
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85

pub(crate) fn cosine(v1: &[f32], v2: &[f32]) -> Result<f32, String> {
    if v1.len() != v2.len() {
        return Err(format!(
            "Vector lengths do not match: {} != {}",
            v1.len(),
            v2.len()
        ));
    }

    let mut dot_product = 0.0;
    let mut v1_norm = 0.0;
    let mut v2_norm = 0.0;

    for i in 0..v1.len() {
        dot_product += v1[i] * v2[i];
        v1_norm += v1[i] * v1[i];
        v2_norm += v2[i] * v2[i];
    }

    v1_norm = v1_norm.sqrt();
    v2_norm = v2_norm.sqrt();

    Ok(dot_product / (v1_norm * v2_norm))
}

pub(crate) fn euclidean(v1: &[f32], v2: &[f32]) -> Result<f32, String> {
    if v1.len() != v2.len() {
        return Err(format!(
            "Vector lengths do not match: {} != {}",
            v1.len(),
            v2.len()
        ));
    }

    let mut sum_of_squares = 0.0;

    for i in 0..v1.len() {
        let difference = v1[i] - v2[i];
        sum_of_squares += difference * difference;
    }

    Ok(sum_of_squares.sqrt())
}

#[test]
fn cosine_test() {
    let v1 = vec![1.0, 2.0, 3.0];
    let v2 = vec![3.0, 2.0, 1.0];
    let result = cosine(&v1, &v2).unwrap();
    let expected = 0.7142857;
    assert!(
        (result - expected).abs() < 0.001,
        "result ({}) != expected ({})",
        result,
        expected
    );
}

#[test]
fn cosine_test_same() {
    let v1 = vec![1.0, 2.0, 3.0];
    let v2 = vec![1.0, 2.0, 3.0];
    let result = cosine(&v1, &v2).unwrap();
    let expected = 1.0;
    assert!(
        (result - expected).abs() < 0.001,
        "result ({}) != expected ({})",
        result,
        expected
    );
}

#[test]
fn cosine_test_opposite() {
    let v1 = vec![1.0, 2.0, 3.0];
    let v2 = vec![-1.0, -2.0, -3.0];
    let result = cosine(&v1, &v2).unwrap();
    let expected = -1.0;
    assert!(
        (result - expected).abs() < 0.001,
        "result ({}) != expected ({})",
        result,
        expected
    );
}