File size: 338 Bytes
b06298d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
# %%
from datasets import load_dataset
from src.retrievers.faiss_retriever import FaissRetriever
data = load_dataset("GroNLP/ik-nlp-22_slp", "paragraphs")
# # %%
# x = data["test"][:3]
# # %%
# for y in x:
# print(y)
# # %%
# x.num_rows
# # %%
retriever = FaissRetriever(data)
scores, result = retriever.retrieve("hello world")
|