holographic-memory 0.5.0

A high-performance Holographic Memory System (HMS) implementing Vector Symbolic Architectures (VSA).
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

// Copyright 2024-2026 WritersLogic Contributors
// SPDX-License-Identifier: Apache-2.0

pub(crate) mod inverted_list;
pub(crate) mod kmeans;
pub(crate) mod nystrom;
pub(crate) mod pq;
pub(crate) mod query;
pub(crate) mod training;

use anyhow::Result;
use serde::{Deserialize, Serialize};

use super::entangled::EntangledHVec;
use inverted_list::InvertedLists;
use kmeans::KMeansNystrom;
use nystrom::NystromProjector;
use pq::PQEncoder;

#[derive(Serialize, Deserialize)]
pub(crate) struct IVFIndex {
    pub(crate) projector: NystromProjector,
    pub(crate) kmeans: KMeansNystrom,
    pub(crate) pq: PQEncoder,
    #[serde(skip)]
    pub(crate) lists: Option<InvertedLists>,
    pub(crate) dim: usize,
    pub(crate) trained: bool,
}

impl IVFIndex {
    pub fn is_trained(&self) -> bool {
        self.trained
    }

    pub fn insert(&mut self, id: &str, vector: &EntangledHVec) -> Result<()> {
        if !self.trained {
            return Ok(());
        }

        let projected = self.projector.project(vector);
        let cluster = self.kmeans.assign(&projected);
        let pq_codes = self.pq.encode(vector);

        if let Some(ref lists) = self.lists {
            lists.append(cluster, id, &pq_codes)
        } else {
            Err(anyhow::anyhow!("Inverted lists database not connected"))
        }
    }
}