tirank.Dataloader.STDataset

class tirank.Dataloader.STDataset(*args: Any, **kwargs: Any)

Bases: Dataset

PyTorch Dataset class for Spatial Transcriptomics (gene pair) data.

Parameters:

df_Xc (pd.DataFrame or np.ndarray) – DataFrame of gene pair features (spots x gene pairs).