scvi.dataloaders.SemiSupervisedDataSplitter.predict_dataloader

SemiSupervisedDataSplitter.predict_dataloader()

Implement one or multiple PyTorch DataLoaders for prediction.

It’s recommended that all data downloads and preparation happen in prepare_data().

Note

Lightning adds the correct sampler for distributed and arbitrary hardware There is no need to set it yourself.

Return type

DataLoader | List[DataLoader]Union[DataLoader, List[DataLoader]]

Returns

Single or multiple PyTorch DataLoaders.

Note

In the case where you return multiple prediction dataloaders, the predict() will have an argument dataloader_idx which matches the order here.