edflow.data.processing.labels module

Summary

Classes:

ExtraLabelsDataset	A dataset with extra labels added.
LabelDataset	A label only dataset to avoid loading unnecessary data.

Reference

class edflow.data.processing.labels.LabelDataset(data)

Bases: edflow.data.dataset_mixin.DatasetMixin

A label only dataset to avoid loading unnecessary data.

__init__(data)

Parameters

data (DatasetMixin) – Some dataset where we are only interested in the labels.

get_example(i)

Return only labels of example.

class edflow.data.processing.labels.ExtraLabelsDataset(data, labeler)

Bases: edflow.data.dataset_mixin.DatasetMixin

A dataset with extra labels added.

__init__(data, labeler)

Parameters

• data (DatasetMixin) – Some Base dataset you want to add labels to

• labeler (Callable) – Must accept two arguments: a Dataset and an index i and return a dictionary of labels to add or overwrite. For all indices the keys in the returned dict must be the same and the type and shape of the values at those keys must be the same per key.

property labels

Add default behaviour for datasets defining an attribute data, which in turn is a dataset. This happens often when stacking several datasets on top of each other.

The default behaviour is to return self.data.labels if possible, and otherwise revert to the original behaviour.