faiss-next 0.6.0

Rust bindings for Faiss (Facebook AI Similarity 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
56

use faiss_next_sys::FaissMetricType;

#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, Default)]
#[repr(u32)]
pub enum MetricType {
    InnerProduct = 0,
    #[default]
    L2 = 1,
    L1 = 2,
    Linf = 3,
    Lp = 4,
    Canberra = 20,
    BrayCurtis = 21,
    JensenShannon = 22,
}

impl MetricType {
    pub fn as_native(self) -> FaissMetricType {
        match self {
            MetricType::InnerProduct => FaissMetricType::METRIC_INNER_PRODUCT,
            MetricType::L2 => FaissMetricType::METRIC_L2,
            MetricType::L1 => FaissMetricType::METRIC_L1,
            MetricType::Linf => FaissMetricType::METRIC_Linf,
            MetricType::Lp => FaissMetricType::METRIC_Lp,
            MetricType::Canberra => FaissMetricType::METRIC_Canberra,
            MetricType::BrayCurtis => FaissMetricType::METRIC_BrayCurtis,
            MetricType::JensenShannon => FaissMetricType::METRIC_JensenShannon,
        }
    }

    pub fn from_native(mt: FaissMetricType) -> Self {
        match mt {
            FaissMetricType::METRIC_INNER_PRODUCT => MetricType::InnerProduct,
            FaissMetricType::METRIC_L2 => MetricType::L2,
            FaissMetricType::METRIC_L1 => MetricType::L1,
            FaissMetricType::METRIC_Linf => MetricType::Linf,
            FaissMetricType::METRIC_Lp => MetricType::Lp,
            FaissMetricType::METRIC_Canberra => MetricType::Canberra,
            FaissMetricType::METRIC_BrayCurtis => MetricType::BrayCurtis,
            FaissMetricType::METRIC_JensenShannon => MetricType::JensenShannon,
            _ => MetricType::L2,
        }
    }
}

impl From<FaissMetricType> for MetricType {
    fn from(mt: FaissMetricType) -> Self {
        Self::from_native(mt)
    }
}

impl From<MetricType> for FaissMetricType {
    fn from(mt: MetricType) -> Self {
        mt.as_native()
    }
}