lisbet.training.tasks

Task configuration module.

Functions

configure_tasks(train_rec, dev_rec, ...)

Internal helper.

Classes

Task(task_id, head, out_dim, loss_function, ...)

class lisbet.training.tasks.Task(task_id, head, out_dim, loss_function, train_dataset, train_loss, train_score, dev_dataset=None, dev_loss=None, dev_score=None)

task_id: str

head: Module

out_dim: int

loss_function: Module

train_dataset: Dataset

train_loss: Metric

train_score: Metric

dev_dataset: Dataset | None = None

dev_loss: Metric | None = None

dev_score: Metric | None = None

__init__(task_id, head, out_dim, loss_function, train_dataset, train_loss, train_score, dev_dataset=None, dev_loss=None, dev_score=None)

lisbet.training.tasks.configure_tasks(train_rec, dev_rec, task_ids, window_size, window_offset, embedding_dim, hidden_dim, data_augmentation, run_seeds, device)

Internal helper. Configures all tasks.