nip.scenario_base.data.NestedArrayDictDataset

nip.scenario_base.data.NestedArrayDictDataset#

class nip.scenario_base.data.NestedArrayDictDataset(hyper_params: HyperParameters, settings: ExperimentSettings, protocol_handler: ProtocolHandler, split: Literal['train', 'test', 'validation'] = 'train')[source]#

Base class for all datasets based on nested array dicts.

Parameters:

  • hyper_params (HyperParameters) – The parameters for the experiment.

  • settings (ExperimentSettings) – The settings for the experiment.

  • protocol_handler (ProtocolHandler) – The protocol handler for the experiment.

  • train (bool) – Whether to load the training or test set.

Methods Summary

__getitem__(index)

__init__(hyper_params, settings, ...[, split])

__len__()

Attributes

instance_keys

The keys specifying the input instance.

keys

The keys (field names) in the dataset.

Methods

abstract __getitem__(index: None | int | slice | str | Tensor | List[Any] | Tuple[Any, ...]) → NestedArrayDict | ndarray[Any, dtype[_ScalarType_co]][source]#
__init__(hyper_params: HyperParameters, settings: ExperimentSettings, protocol_handler: ProtocolHandler, split: Literal['train', 'test', 'validation'] = 'train')[source]#
abstract __len__() → int[source]#