renate.models.task_identification_strategies module#
- class renate.models.task_identification_strategies.TaskEstimator[source]#
Bases:
Module
,ABC
An ABC that all task estimator methods inherit.
They implement two methods
update_task_prototypes
andinfer_task
.
- class renate.models.task_identification_strategies.TaskPrototypes(task_id, clusters_per_task, embedding_size)[source]#
Bases:
TaskEstimator
Task identification method proposed in S-Prompts.
- Parameters: