Skip to content

declearn.dataset.DataSpecs

Dataclass to wrap a dataset's metadata.

Source code in declearn/dataset/_base.py 
34
35
36
37
38
39
40
41
42
43
@dataclasses.dataclass
class DataSpecs:
    """Dataclass to wrap a dataset's metadata."""

    n_samples: int
    features_shape: Optional[
        Union[Tuple[Optional[int], ...], List[Optional[int]]]
    ] = None
    classes: Optional[Set[Any]] = None
    data_type: Optional[str] = None