vector_base 0.1.0

Vector base
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

use ndarray::arr1;

pub mod comand_line;
pub mod embedding;
pub mod vector_store;

pub async fn run() {
    let mut store = vector_store::store::VectorStore::new();

    let word = comand_line::enter::get_word().await.unwrap();
    let res = embedding::get_emb::get_vector(&word).await.unwrap();
    let vectore_container = vector_store::container::VectorContainer::new(res, word);
    store.add(vectore_container);

    let word = comand_line::enter::get_word().await.unwrap();
    let res = embedding::get_emb::get_vector(&word).await.unwrap();
    let vectore_container = vector_store::container::VectorContainer::new(res, word);
    store.add(vectore_container);

    let word = comand_line::enter::get_word().await.unwrap();
    let res = embedding::get_emb::get_vector(&word).await.unwrap();
    let vectore_container = vector_store::container::VectorContainer::new(res, word);
    store.add(vectore_container);

    let word = comand_line::enter::get_word().await.unwrap();
    let res = embedding::get_emb::get_vector(&word).await.unwrap();
    let vectore_container = vector_store::container::VectorContainer::new(res, word);
    store.add(vectore_container);
    println!("word to compare");

    let word = comand_line::enter::get_word().await.unwrap();
    let res = embedding::get_emb::get_vector(&word).await.unwrap();

    let vec_words = store.find_similers(&arr1(&res), 2);

    println!("{:?}", vec_words.get(0).unwrap().name);
    println!("{:?}", vec_words.get(1).unwrap().name);
}