pnpl.datasets.grouped_dataset.GroupedDataset

class pnpl.datasets.grouped_dataset.GroupedDataset(original_dataset, grouped_samples=10, drop_remaining=False, shuffle=False, average_grouped_samples=True)

__init__(original_dataset, grouped_samples=10, drop_remaining=False, shuffle=False, average_grouped_samples=True)

Groups n samples from the original dataset by label

Parameters: - original_dataset: The original dataset to group - grouped_samples: The number of samples to group over - drop_remaining: Whether to drop the last group if it is incomplete - shuffle: Whether to shuffle the samples - average_grouped_samples: Whether to average the grouped samples

Methods

__init__(original_dataset[, ...])

Groups n samples from the original dataset by label