scglue.models.glue.GLUETrainer.format_data

GLUETrainer.format_data(data)[source]

Format data tensors

Note

The data dataset should contain data arrays for each domain in the same order as domain keys of the network.

Return type

Tuple[Mapping[str, Tensor], Mapping[str, Tensor], Tensor, Tensor, Tensor]