create_run_from_scores

lightning_ir.base.validation_utils.create_run_from_scores(query_ids: Sequence[str], doc_ids: Sequence[Sequence[str]], scores: Tensor) → DataFrame[source]

Convience function to create a run DataFrame from query and document ids and scores.

Parameters:

  • query_ids (Sequence[str]) – Query ids

  • doc_ids (Sequence[Sequence[str]]) – Document ids

  • scores (torch.Tensor) – Scores

Returns:

DataFrame with query_id, q0, doc_id, score, rank, and system columns

Return type:

pd.DataFrame