hora 0.1.1

Hora Search Everywhere
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

#![allow(dead_code)]

use crate::core::node;

use serde::{Deserialize, Serialize};

#[derive(Debug, Serialize, Deserialize)]
pub struct HNSWParams<E: node::FloatElement> {
    pub max_item: usize,
    pub n_neighbor: usize,
    pub n_neighbor0: usize,
    pub max_level: usize,
    pub ef_build: usize,
    pub ef_search: usize,
    pub has_deletion: bool,
    pub e_type: E,
}

impl<E: node::FloatElement> HNSWParams<E> {
    pub fn max_item(mut self, new_max_item: usize) -> Self {
        self.max_item = new_max_item;
        self
    }

    pub fn n_neighbor(mut self, new_n_neighbor: usize) -> Self {
        self.n_neighbor = new_n_neighbor;
        self
    }

    pub fn n_neighbor0(mut self, new_n_neighbor0: usize) -> Self {
        self.n_neighbor0 = new_n_neighbor0;
        self
    }

    pub fn ef_build(mut self, new_ef_build: usize) -> Self {
        self.ef_build = new_ef_build;
        self
    }

    pub fn ef_search(mut self, new_ef_search: usize) -> Self {
        self.ef_search = new_ef_search;
        self
    }

    pub fn has_deletion(mut self, new_has_deletion: bool) -> Self {
        self.has_deletion = new_has_deletion;
        self
    }
}

impl<E: node::FloatElement> Default for HNSWParams<E> {
    fn default() -> Self {
        HNSWParams {
            max_item: 1000000,
            n_neighbor: 32,
            n_neighbor0: 64,
            max_level: 20,
            ef_build: 500,
            ef_search: 16,
            has_deletion: false,
            e_type: E::from_f32(0.0).unwrap(),
        }
    }
}