FakeDataset¶
-
class
olympus.datasets.fake.FakeDataset(input_shape, target_shape, train_size=1024, valid_size=128, test_size=128, data_path=None)[source]¶ Bases:
olympus.datasets.dataset.AllDatasetGenerate random tensors as input data
Attributes: classesReturn the mapping between samples index and their class
- collate_fn
- dataset
input_shapeReturn the size of the samples
target_shapeReturn the size of the target
test_sizeSize of the test set
train_sizeSize of the training set
valid_sizeSize of the validation set
Methods
categories()Dataset tags so we can filter what we want depending on the task transforms()register_datapipe_as_function register_function