Spaces:
Sleeping
Sleeping
File size: 488 Bytes
94cd2e4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
from abc import ABC, abstractmethod
from typing import Any, Dict, Type
class BaseRetriever(ABC):
@property
@abstractmethod
def index_class(self) -> Type[Any]:
pass
def get_term_weights(self, query: str, cid: str) -> Dict[str, float]:
raise NotImplementedError
@abstractmethod
def score(self, query: str, cid: str) -> float:
pass
@abstractmethod
def retrieve(self, query: str, topk: int = 10) -> Dict[str, float]:
pass
|