multiml.task.pytorch.datasets package

Submodules

• multiml.task.pytorch.datasets.numpy_dataset module
  • NumpyDataset
    • NumpyDataset.__init__()
    • NumpyDataset.__len__()
    • NumpyDataset.__getitem__()
    • NumpyDataset.get_size()
    • NumpyDataset.get_data()
• multiml.task.pytorch.datasets.storegate_dataset module
  • StoreGateDataset
    • StoreGateDataset.__init__()
    • StoreGateDataset.__len__()
    • StoreGateDataset.__getitem__()

Module contents

class multiml.task.pytorch.datasets.StoreGateDataset(storegate, phase, device=None, preload=None, input_var_names=None, true_var_names=None, callbacks=None)

Bases: Dataset

StoreGate dataset class.

__init__(storegate, phase, device=None, preload=None, input_var_names=None, true_var_names=None, callbacks=None)

__len__()

__getitem__(index)

class multiml.task.pytorch.datasets.NumpyDataset(inputs, targets, callbacks=None)

Bases: Dataset

StoreGate dataset class.

__init__(inputs, targets, callbacks=None)

__len__()

__getitem__(index)

get_size(inputs)

get_data(data, index)