torch_mist.utils.data.csv

Module Contents

Classes

CSVDataset

class torch_mist.utils.data.csv.CSVDataset(filepath: str, remove_nan_rows: bool = True, rename_columns: Dict[str, str] | None = None, **kwargs)

Bases: torch.utils.data.Dataset

__len__()

__getitem__(index)