space 0.18.0

A library providing abstractions for spatial datastructures and search
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

use space::{Knn, KnnFromBatch, LinearKnn, Metric, Neighbor};

#[derive(Default)]
struct Hamming;

impl Metric<u8> for Hamming {
    type Unit = u8;

    fn distance(&self, &a: &u8, &b: &u8) -> Self::Unit {
        (a ^ b).count_ones() as u8
    }
}

#[test]
fn test_linear_knn() {
    let data = vec![
        (0b1010_1010, 12),
        (0b1111_1111, 13),
        (0b0000_0000, 14),
        (0b1111_0000, 16),
        (0b0000_1111, 10),
    ];

    let search: LinearKnn<Hamming, _> = KnnFromBatch::from_batch(data.iter());

    assert_eq!(
        &search.knn(&0b0101_0000, 3),
        &[
            (
                Neighbor {
                    index: 2,
                    distance: 2
                },
                &data[2].0,
                &data[2].1
            ),
            (
                Neighbor {
                    index: 3,
                    distance: 2
                },
                &data[3].0,
                &data[3].1
            ),
            (
                Neighbor {
                    index: 0,
                    distance: 6
                },
                &data[0].0,
                &data[0].1
            )
        ]
    );
}