ccm commited on
Commit
8b90459
·
verified ·
1 Parent(s): dc47567

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -0
main.py CHANGED
@@ -18,6 +18,7 @@ data.reset_index(inplace=True)
18
 
19
  dimensionality = len(data['embedding'][0])
20
  index = faiss.IndexFlatL2(dimensionality)
 
21
 
22
  vectors = numpy.stack(data['embedding'].to_list(), axis=0)
23
 
 
18
 
19
  dimensionality = len(data['embedding'][0])
20
  index = faiss.IndexFlatL2(dimensionality)
21
+ index.metric_type = faiss.METRIC_INNER_PRODUCT
22
 
23
  vectors = numpy.stack(data['embedding'].to_list(), axis=0)
24