Heimdall.task.SingleInstanceTask

Heimdall.task.SingleInstanceTask#

class Heimdall.task.SingleInstanceTask(data, task_type, metrics, shuffle, head_config, loss_config, interaction_type=None, top_k=None, label_obsm_name=None, label_col_name=None, reducer_config=None, splits=None, train_split=(None,), track_metric=None)[source]#

Bases: Task

Attributes

data

task_type

metrics

shuffle

head_config

loss_config

Methods

setup_labels()

Parameters:

  • data (CellRepresentation)

  • task_type (str)

  • metrics (list[str])

  • shuffle (bool)

  • head_config (DictConfig)

  • loss_config (DictConfig)

  • interaction_type (str | None)

  • top_k (list[int] | None)

  • label_obsm_name (str | None)

  • label_col_name (str | None)

  • reducer_config (DictConfig | None)

  • splits (DictConfig | None)

  • train_split (float | None)

  • track_metric (str | None)