holisticai.datasets.DatasetDict

class holisticai.datasets.DatasetDict(**datasets)

A dictionary-like class that represents a collection of datasets. Usually, the keys are train, validation, and test.

Parameters:

datasetsdict

A dictionary containing the datasets, where the keys are the names of the datasets and the values are the datasets themselves.

Methods:

__getitem__(key)

Returns a Dataset object corresponding to the given key.