Skip to content

dataloader

dataloader

Module for abstract DataLoader

Classes

DataLoader

DataLoader()

Bases: ABC

Abstract implementation of a data loader which is used by the GenericDataset

Source code in niceml/data/dataloaders/dataloader.py
def __init__(self):
    self.data_description = None
Functions
initialize
initialize(data_description)

Initializes the DataLoader with a DataDescription

Source code in niceml/data/dataloaders/dataloader.py
def initialize(self, data_description: DataDescription):
    """Initializes the DataLoader with a DataDescription"""
    self.data_description: DataDescription = data_description
load_data abstractmethod
load_data(data_info)

Loads the data from a DataInfo object and puts it in a container class

Source code in niceml/data/dataloaders/dataloader.py
@abstractmethod
def load_data(self, data_info: DataInfo) -> Any:
    """Loads the data from a DataInfo object and puts it in a container class"""