kosmos.ml.datasets.dataset

Classes

class SLDataset(x: collections.abc.Collection, y: collections.abc.Collection, *, min_max_scaler: bool = True)

Bases: torch.utils.data.Dataset[tuple[torch.Tensor, torch.Tensor]], abc.ABC

Dataset for supervised learning classification.

Initialize the dataset.

Parameters:

• x (Collection) – Feature values.

• y (Collection) – Target values.

• min_max_scaler (bool) – Whether to apply min-max scaling to the features.

Properties

property input_dimension

Number of feature columns.

property output_dim

Number of distinct classes.

property class_names

Class names.