mlcolvar.data.DictDataset

class mlcolvar.data.DictDataset(dictionary: dict = None, feature_names=None, create_ref_idx: bool = False, **kwargs)

Bases: Dataset

Define a torch dataset from a dictionary of lists/array/tensors and names.

E.g. { ‘data’torch.Tensor([1,2,3,4]),

‘labels’ : [0,0,1,1], ‘weights’ : np.asarray([0.5,1.5,1.5,0.5]) }

__init__(dictionary: dict = None, feature_names=None, create_ref_idx: bool = False, **kwargs)

Create a Dataset from a dictionary or from a list of kwargs.

Parameters:

• dictionary (dict) – Dictionary with names and tensors

• feature_names (array-like) – List or numpy array with feature names

Methods

__init__([dictionary, feature_names, ...])	Create a Dataset from a dictionary or from a list of kwargs.
get_stats()	Compute statistics ('mean','Std','Min','Max') of the dataset.

property feature_names

Feature names.

get_stats()

Compute statistics (‘mean’,’Std’,’Min’,’Max’) of the dataset.

Returns:

dictionary of dictionaries with statistics

Return type:

stats

Attributes

feature_names	Feature names.
keys