Heimdall.models.HeimdallModel

class Heimdall.models.HeimdallModel(data, model_config)

Bases: Module

Attributes

dtype
training

Methods

encode_cell(cell_inputs)	Given the either single- or multiple-cells, use the cell encoder to embed the cell(s).
forward(inputs)	Define the computation performed at every call.

Parameters:

• data (CellRepresentation)

• model_config (DictConfig)