avalanche.benchmarks.utils.DataAttribute

class avalanche.benchmarks.utils.DataAttribute(data: IDataset[T_co], name: str, use_in_getitem: bool = False)[source]

Data attributes manage sample-wise information such as task or class labels.

It provides access to unique values (self.uniques) and their indices (self.val_to_idx). Both fields are initialized lazily.

Data attributes can be efficiently concatenated and subsampled.

__init__(data: IDataset[T_co], name: str, use_in_getitem: bool = False)[source]

Data Attribute.

Parameters:
  • data – a sequence of values, one for each sample.

  • name – a name that uniquely identifies the attribute. It is used by AvalancheDataset to dynamically add it to its attributes.

  • use_in_getitem – If True, AvalancheDataset will add the value at the end of each sample.

Methods

__init__(data, name[, use_in_getitem])

Data Attribute.

concat(other)

Concatenation operation.

index(value, [start, [stop]])

Raises ValueError if the value is not present.

subset(indices)

Subset operation.

Attributes

count

Dictionary of value -> count.

data

uniques

Set of unique values in the attribute.

val_to_idx

Dictionary mapping unique values to indices.