Spaces:
Running
on
Zero
Running
on
Zero
Update main.py
Browse files
main.py
CHANGED
@@ -29,7 +29,7 @@ metrics = [faiss.METRIC_INNER_PRODUCT ,faiss.METRIC_L2]
|
|
29 |
normalization = [True, False]
|
30 |
vectors = numpy.stack(data["embedding"].tolist(), axis=0)
|
31 |
for metric in metrics:
|
32 |
-
for normal in normalization
|
33 |
index = faiss.IndexFlatL2(len(data["embedding"][0]))
|
34 |
index.metric_type = metric
|
35 |
if normal:
|
|
|
29 |
normalization = [True, False]
|
30 |
vectors = numpy.stack(data["embedding"].tolist(), axis=0)
|
31 |
for metric in metrics:
|
32 |
+
for normal in normalization:
|
33 |
index = faiss.IndexFlatL2(len(data["embedding"][0]))
|
34 |
index.metric_type = metric
|
35 |
if normal:
|