Skip to content
Snippets Groups Projects
Commit 05507ff3 authored by Ismail Ashraq's avatar Ismail Ashraq
Browse files

added option to set sparse_encoder

parent 7abb86d5
No related branches found
No related tags found
No related merge requests found
......@@ -16,11 +16,21 @@ class HybridRouteLayer:
score_threshold: float
def __init__(
self, encoder: BaseEncoder, routes: list[Route] = [], alpha: float = 0.3
self,
encoder: BaseEncoder,
sparse_encoder: BM25Encoder | None = None,
routes: list[Route] = [],
alpha: float = 0.3,
):
self.encoder = encoder
self.score_threshold = self.encoder.score_threshold
self.sparse_encoder = BM25Encoder()
if sparse_encoder is None:
logger.warning("No sparse_encoder provided. Using default BM25Encoder.")
self.sparse_encoder = BM25Encoder()
else:
self.sparse_encoder = sparse_encoder
self.alpha = alpha
# if routes list has been passed, we initialize index now
if routes:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment