tirank.Dataloader.SCDataset

class tirank.Dataloader.SCDataset(*args: Any, **kwargs: Any)[source]

Bases: Dataset

PyTorch Dataset class for single-cell RNA-seq (gene pair) data.

Parameters:

df_Xb (pd.DataFrame or np.ndarray) – DataFrame of gene pair features (cells x gene pairs).