mmlearn.tasks.zero_shot_classification.ClassificationTaskSpec¶

class ClassificationTaskSpec(query_modality, top_k)[source]¶

Bases: object

Specification for a classification task.

Methods

Attributes

query_modality: str¶: The modality of the query input.

top_k: list[int]¶: The top-k values for which to compute the classification metrics like accuracy.