mmlearn.tasks.zero_shot_retrieval.RetrievalTaskSpec

class RetrievalTaskSpec(query_modality, target_modality, top_k)[source]

Bases: object

Specification for a retrieval task.

Methods

Attributes

query_modality: str

The query modality.

target_modality: str

The target modality.

top_k: list[int]

The top-k values for which to compute the retrieval recall metrics.