kosmos.ml.datasets.iris_dataset

Classes

class IrisDataset(*, min_max_scaler: bool = True)

Bases: kosmos.ml.datasets.dataset.SLDataset

Iris dataset for multiclass classification.

Notes

  • Number of instances: 150 (50 per class)

  • Number of features: 4 numeric

  • Classes: 3 (balanced; ~33.3% each)

References

Initialize the dataset.

Parameters:

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


Properties

property class_names

0 -> setosa, 1 -> versicolor, 2 -> virginica.

Type:

Return human-readable class labels

property input_dimension

Number of feature columns.

property output_dim

Number of distinct classes.