RankSample
- class lightning_ir.data.data.RankSample(query_id: str, query: str, doc_ids: Sequence[str], docs: Sequence[str], targets: torch.Tensor | None = None, qrels: List[Dict[str, Any]] | None = None)[source]
Bases:
object
- __init__(query_id: str, query: str, doc_ids: Sequence[str], docs: Sequence[str], targets: Tensor | None = None, qrels: List[Dict[str, Any]] | None = None) None
Methods
__init__
(query_id, query, doc_ids, docs[, ...])Attributes
qrels
targets
query_id
query
doc_ids
docs