Skip to main content

dataloaders

Classes concerning data loading and dataloaders.

Classes​

BitfountDataLoader​

class BitfountDataLoader(    dataset: _BaseBitfountDataset,    batch_size: Optional[int] = None,    shuffle: bool = False,):

A backend-agnostic data loader.

Arguments

  • dataset: The dataset for the dataloader.
  • batch_size: The batch size for the dataloader. Defaults to None.

Subclasses​

  • bitfount.backends.pytorch.data.dataloaders._BasePyTorchBitfountDataLoader
  • HuggingFaceIterableBitfountDataLoader
  • bitfount.data.huggingface.dataloaders._BaseHuggingFaceBitfountDataLoader

Methods​

get_x_dataframe​

def get_x_dataframe(    self,) ‑> Union[pandas.core.frame.DataFrame, Tuple[pandas.core.frame.DataFrame, pandas.core.frame.DataFrame]]:

Gets the x-dataframe of the data i.e. features.

For models incompatible with the iter approach.

get_y_dataframe​

def get_y_dataframe(self) ‑> pandas.core.frame.DataFrame:

Gets the y-dataframe of the data i.e. target.

For models incompatible with the iter approach.