kosmos.ml.datasets.wine_dataset

Classes

class WineDataset(*, min_max_scaler: bool = True)

Bases: kosmos.ml.datasets.dataset.SLDataset

Wine dataset for multiclass classification.

Notes

  • Number of instances: 178

  • Number of features: 13 numeric

  • Classes: 3 (slightly imbalanced;[59,71,48])

References

Initialize the dataset.

Parameters:

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


Properties

property class_names

class_0, class_1, class_2.

Type:

Return human-readable class labels

property input_dimension

Number of feature columns.

property output_dim

Number of distinct classes.