scglue.models.scglue.SCGLUETrainer.format_data

SCGLUETrainer.format_data(data)[source]

Format data tensors

Note

The data dataset should contain data arrays for each domain, followed by alternative input arrays for each domain, in the same order as domain keys of the network.

Return type

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