tirank.Dataloader.BulkDataset

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

Bases: Dataset

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

Handles different analysis modes by returning the appropriate clinical labels (e.g., time and event for Cox, a single label for Classification).

Parameters:

df_Xa (pd.DataFrame) – DataFrame of gene pair features (samples x gene pairs).
df_cli (pd.DataFrame or pd.Series) – DataFrame/Series with clinical information.
mode (str, optional) – Analysis mode. One of ‘Cox’, ‘Classification’, or ‘Regression’. Defaults to ‘Cox’.