CAntoniadis commited on
Commit
ff800db
·
verified ·
1 Parent(s): 7cf833b

Update simple_search_engine/search_engine.py

Browse files
simple_search_engine/search_engine.py CHANGED
@@ -3,6 +3,7 @@ import json
3
  from typing import List, Dict
4
  from sklearn.feature_extraction.text import TfidfVectorizer
5
  from sklearn.metrics.pairwise import linear_kernel
 
6
 
7
  CORPUS_DIR = "../corpus" # Now points to the unified directory
8
 
 
3
  from typing import List, Dict
4
  from sklearn.feature_extraction.text import TfidfVectorizer
5
  from sklearn.metrics.pairwise import linear_kernel
6
+ from rank_bm25 import BM25Okapi
7
 
8
  CORPUS_DIR = "../corpus" # Now points to the unified directory
9