seismic 0.2.1

Seismic is designed for effective and efficient KNN retrieval over learned sparse embeddings.
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

from seismic import PySeismicIndex
from time import time
import argparse

def main():
    parser = argparse.ArgumentParser(description="Parser for searching with a pre-built Seismic Index.")
    parser.add_argument("--index-path", help="Path to the index file")
    parser.add_argument("--queries-path", help="Path to the queries in the inner Seismic format")
    
    
    args = parser.parse_args()
    
    
    

    index_path = args.index_path

    print(f"Loading the index from {index_path}..")
    index = PySeismicIndex.load(index_path)
    print("Index loaded")
    query_path = args.queries_path
    

    k = 10

    query_cut = 6

    heap_factor = 0.7
    num_threads = 1

    start = time()
    results = index.batch_search(query_path, k, query_cut, heap_factor, num_threads)
    end = time()

    n_queries = 6980
    print("Number of queries ", n_queries)

    print( ((end - start) / 6980) * 10**6)

#print((results[0] ) )


if __name__ == "__main__":
    main()